Add a "native" target to run the kernel NFS server.
[bluesky.git] / kvstore / socket_pool.h
1 #ifndef _SOCKET_POOL_H_
2 #define _SOCKET_POOL_H_ 1
3
4 #include <iostream>
5 #include <set>
6 #include <queue>
7 #include <boost/shared_ptr.hpp>
8 #include <boost/thread.hpp>
9
10 #include "util.h"
11
12 using namespace std;
13 using namespace boost;
14
15
16 class SocketPool
17 {
18 public:
19     SocketPool(int max_streams,
20                asio::io_service &io_svc);
21     void setEndpoint(const tcp::endpoint &endpoint);
22     void cancelAndClear();
23     shared_ptr<tcp::socket> getSocket();
24     void putSocket(shared_ptr<tcp::socket> socket);
25 private:
26     int _issued;
27     int _max_sockets;
28     mutex _sockets_lock;
29     condition_variable _sockets_non_empty;
30     asio::io_service &_io_service;
31     tcp::endpoint _endpoint;
32     queue<shared_ptr<tcp::socket> > _queue;
33     set<shared_ptr<tcp::socket> > _set;
34 };
35
36 class SocketCheckout
37 {
38 public:
39     SocketCheckout(SocketPool *pool);
40     ~SocketCheckout();
41
42     tcp::socket& operator*();
43     tcp::socket* operator->();
44
45     shared_ptr<tcp::socket>& socket();
46
47 private:
48     shared_ptr<tcp::socket> _socket;
49     SocketPool *_pool;
50 };
51
52 #endif