git.sesse.net Git - cubemap/blob - serverpool.cpp

   1 #include "serverpool.h"
   2
   3 using namespace std;
   4
   5 ServerPool::ServerPool(int size)
   6         : servers(new Server[size]),
   7           num_servers(size),
   8           clients_added(0)
   9 {
  10 }
  11
  12 ServerPool::~ServerPool()
  13 {
  14         delete[] servers;
  15 }
  16
  17 void ServerPool::add_client(int sock)
  18 {
  19         servers[clients_added++ % num_servers].add_client_deferred(sock);
  20 }
  21
  22 void ServerPool::add_client_from_serialized(const ClientProto &client)
  23 {
  24         servers[clients_added++ % num_servers].add_client_from_serialized(client);
  25 }
  26
  27 void ServerPool::add_stream(const std::string &stream_id)
  28 {
  29         for (int i = 0; i < num_servers; ++i) {
  30                 servers[i].add_stream(stream_id);
  31         }
  32 }
  33
  34 void ServerPool::add_stream_from_serialized(const StreamProto &stream)
  35 {
  36         for (int i = 0; i < num_servers; ++i) {
  37                 servers[i].add_stream_from_serialized(stream);
  38         }
  39 }
  40
  41 void ServerPool::set_header(const std::string &stream_id, const std::string &header)
  42 {
  43         for (int i = 0; i < num_servers; ++i) {
  44                 servers[i].set_header(stream_id, header);
  45         }
  46 }
  47
  48 void ServerPool::add_data(const std::string &stream_id, const char *data, size_t bytes)
  49 {
  50         for (int i = 0; i < num_servers; ++i) {
  51                 servers[i].add_data_deferred(stream_id, data, bytes);
  52         }
  53 }
  54
  55 void ServerPool::run()
  56 {
  57         for (int i = 0; i < num_servers; ++i) {
  58                 servers[i].run();
  59         }
  60 }
  61
  62 void ServerPool::stop()
  63 {
  64         for (int i = 0; i < num_servers; ++i) {
  65                 servers[i].stop();
  66         }
  67 }
  68
  69 vector<ClientStats> ServerPool::get_client_stats() const
  70 {
  71         vector<ClientStats> ret;
  72         for (int i = 0; i < num_servers; ++i) {
  73                 vector<ClientStats> stats = servers[i].get_client_stats();
  74                 ret.insert(ret.end(), stats.begin(), stats.end());
  75         }
  76         return ret;
  77 }
  78
  79 void ServerPool::set_mark_pool(const std::string &stream_id, MarkPool *mark_pool)
  80 {
  81         for (int i = 0; i < num_servers; ++i) {
  82                 servers[i].set_mark_pool(stream_id, mark_pool);
  83         }
  84 }