from thrift.transport import TSocket
from thrift.transport import TTransport
from thrift.protocol import TBinaryProtocol
from thrift.server import TServer
import sys

sys.path.append("./gen-py/")

from DemoService import DemoService
import json

class DemoServer:

    def __init__(self):
        self.log = {}

    def ping(self, param):
        return "echo:" + param

    def post(self, jstr):
        try:
            jdict = json.loads(jstr)
        except Exception as e :
            return json.dumps({"res":False})
        
        jdict["res"] = True
        return json.dumps(jdict);

if __name__ == '__main__':
    handler = DemoServer()
    processor = DemoService.Processor(handler)

    # 监听端口
    transport = TSocket.TServerSocket(host="0.0.0.0", port=9999)

    # 选择传输层
    tfactory = TTransport.TBufferedTransportFactory()

    # 选择传输协议
    pfactory = TBinaryProtocol.TBinaryProtocolFactory()

    # 创建服务端
    server = TServer.TThreadPoolServer(processor, transport, tfactory, pfactory)

    # 设置连接线程池数量
    server.setNumThreads(5)

    # 启动服务
    server.serve()