blob: e7e7148447b71bfa31f0273b7d53250704d7371c [file] [log] [blame]
Mark Sleed788b2e2006-09-07 01:26:35 +00001#ifndef _THRIFT_SERVER_TTHREADPOOLSERVER_H_
2#define _THRIFT_SERVER_TTHREADPOOLSERVER_H_ 1
Marc Slemko35452342006-08-03 19:01:37 +00003
Marc Slemko16698852006-08-04 03:16:10 +00004#include <concurrency/ThreadManager.h>
5#include <server/TServer.h>
6#include <transport/TServerTransport.h>
7
8#include <boost/shared_ptr.hpp>
Marc Slemko35452342006-08-03 19:01:37 +00009
10namespace facebook { namespace thrift { namespace server {
Marc Slemko16698852006-08-04 03:16:10 +000011
12using namespace facebook::thrift::concurrency;
13using namespace facebook::thrift::transport;
14using namespace boost;
15
16class TThreadPoolServer : public TServer {
17public:
18
19 class Task;
20
21 TThreadPoolServer(shared_ptr<TProcessor> processor,
Martin Kraemer50919172007-01-26 01:41:03 +000022 shared_ptr<TServerTransport> serverTransport,
23 shared_ptr<TTransportFactory> transportFactory,
Mark Slee4af6ed72006-10-25 19:02:49 +000024 shared_ptr<TProtocolFactory> protocolFactory,
Martin Kraemer50919172007-01-26 01:41:03 +000025 shared_ptr<ThreadManager> threadManager);
Marc Slemko16698852006-08-04 03:16:10 +000026
Aditya Agarwal9abb0d62007-01-24 22:53:54 +000027 TThreadPoolServer(shared_ptr<TProcessor> processor,
Martin Kraemer50919172007-01-26 01:41:03 +000028 shared_ptr<TServerTransport> serverTransport,
Aditya Agarwal9abb0d62007-01-24 22:53:54 +000029 shared_ptr<TTransportFactory> inputTransportFactory,
30 shared_ptr<TTransportFactory> outputTransportFactory,
31 shared_ptr<TProtocolFactory> inputProtocolFactory,
32 shared_ptr<TProtocolFactory> outputProtocolFactory,
Martin Kraemer50919172007-01-26 01:41:03 +000033 shared_ptr<ThreadManager> threadManager);
Aditya Agarwal9abb0d62007-01-24 22:53:54 +000034
Marc Slemko16698852006-08-04 03:16:10 +000035 virtual ~TThreadPoolServer();
36
Mark Slee794993d2006-09-20 01:56:10 +000037 virtual void serve();
Marc Slemko16698852006-08-04 03:16:10 +000038
39protected:
40
Marc Slemko16698852006-08-04 03:16:10 +000041 shared_ptr<ThreadManager> threadManager_;
42
43};
44
Marc Slemko35452342006-08-03 19:01:37 +000045}}} // facebook::thrift::server
46
Mark Sleed788b2e2006-09-07 01:26:35 +000047#endif // #ifndef _THRIFT_SERVER_TTHREADPOOLSERVER_H_