blob: aabd68619bdc08a9d60ea03c6e2906d7edfaf17a [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,
Marc Slemko16698852006-08-04 03:16:10 +000022 shared_ptr<TServerTransport> serverTransport,
Mark Sleed788b2e2006-09-07 01:26:35 +000023 shared_ptr<TTransportFactory> transportFactory,
Mark Slee4af6ed72006-10-25 19:02:49 +000024 shared_ptr<TProtocolFactory> protocolFactory,
25 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,
28 shared_ptr<TServerTransport> serverTransport,
29 shared_ptr<TTransportFactory> inputTransportFactory,
30 shared_ptr<TTransportFactory> outputTransportFactory,
31 shared_ptr<TProtocolFactory> inputProtocolFactory,
32 shared_ptr<TProtocolFactory> outputProtocolFactory,
33 shared_ptr<ThreadManager> threadManager);
34
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_