def make_server(processor):
"""
Creates a TCppServer given a processor. This is the function used
internally, but it may be of interest separately as well.
"""
server = TCppServer(processor)
server.setPort(0)
server.setNumCPUWorkerThreads(1)
server.setNumIOWorkerThreads(1)
server.setNewSimpleThreadManager(1, 5, False, 50)
return server
def make_server(processor):
"""
Creates a TCppServer given a processor. This is the function used
internally, but it may be of interest separately as well.
"""
server = TCppServer(processor)
server.setPort(0)
server.setNumCPUWorkerThreads(1)
server.setNumIOWorkerThreads(1)
server.setNewSimpleThreadManager(count=1,
pendingTaskCountMax=5,
enableTaskStats=False,
maxQueueLen=50)
return server
def initTask(self):
"""Overridden to bind sockets, etc"""
super(CPPServerTask, self).initTask()
self._stopped = False
self.server = TCppServer(self.processor)
# TODO: There is no API to set host yet.
# self.server.setHost(self.host)
self.server.setPort(self.port)
self.server.setNWorkerThreads(self.num_threads)
self.server.setup()
self.bound_addrs = [self.server.getAddress()]
for addrinfo in self.bound_addrs:
self.bound_host, self.bound_port = addrinfo[0:2]
self.logger.info(
"%s Server Started on %s", self.name,
self._fmt_hostport(self.bound_host, self.bound_port))
def setUp(self):
super(TestServer, self).setUp()
self.noop_event = threading.Event()
self.serverEventHandler = TestServerEventHandler()
self.server = TCppServer(self.getProcessor())
self.server.setServerEventHandler(self.serverEventHandler)
self.addCleanup(self.stopServer)
# Let the kernel choose a port.
self.server.setPort(0)
self.server_thread = threading.Thread(target=self.server.serve)
self.server_thread.start()
for t in range(30):
addr = self.server.getAddress()
if addr:
break
time.sleep(0.1)
self.assertTrue(addr)
self.server_addr = addr
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
from __future__ import unicode_literals
from fb303.ContextFacebookBase import ContextFacebookBase
from thrift.Thrift import TProcessorEventHandler
from thrift.server.TCppServer import TCppServer
from test.reverse import ReverseService, ttypes
class ReverseEventHandler(TProcessorEventHandler):
def getHandlerContext(self, fn_name, server_context):
return server_context
class ReverseHandler(ContextFacebookBase,
ReverseService.ContextIface):
def __init__(self):
ContextFacebookBase.__init__(self, "reverse")
def reverse(self, handler_ctx, req):
print("client_principal is {}".format(handler_ctx.getClientPrincipal()))
return ttypes.Response(str="".join(reversed(req.str)))
processor = ReverseService.ContextProcessor(ReverseHandler())
processor.setEventHandler(ReverseEventHandler())
server = TCppServer(processor)
server.setPort(9999)
server.serve()
