Add support for setting fwmarks per-connection. Useful for shaping individual streams...
[cubemap] / 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 vector<ClientStats> ServerPool::get_client_stats() const
63 {
64         vector<ClientStats> ret;
65         for (int i = 0; i < num_servers; ++i) {
66                 vector<ClientStats> stats = servers[i].get_client_stats();
67                 ret.insert(ret.end(), stats.begin(), stats.end());
68         }
69         return ret;
70 }
71         
72 void ServerPool::set_mark_pool(const std::string &stream_id, MarkPool *mark_pool)
73 {
74         for (int i = 0; i < num_servers; ++i) {
75                 servers[i].set_mark_pool(stream_id, mark_pool);
76         }       
77 }