1#!/usr/bin/env python
2
3#
4# Licensed to the Apache Software Foundation (ASF) under one
5# or more contributor license agreements. See the NOTICE file
6# distributed with this work for additional information
7# regarding copyright ownership. The ASF licenses this file
8# to you under the Apache License, Version 2.0 (the
9# "License"); you may not use this file except in compliance
10# with the License. You may obtain a copy of the License at
11#
12#   http://www.apache.org/licenses/LICENSE-2.0
13#
14# Unless required by applicable law or agreed to in writing,
15# software distributed under the License is distributed on an
16# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
17# KIND, either express or implied. See the License for the
18# specific language governing permissions and limitations
19# under the License.
20#
21
22import glob
23import sys
24sys.path.append('gen-py')
25sys.path.insert(0, glob.glob('../../lib/py/build/lib*')[0])
26
27from tutorial import Calculator
28from tutorial.ttypes import InvalidOperation, Operation
29
30from shared.ttypes import SharedStruct
31
32from thrift.transport import TSocket
33from thrift.transport import TTransport
34from thrift.protocol import TBinaryProtocol
35from thrift.server import TServer
36
37
38class CalculatorHandler:
39    def __init__(self):
40        self.log = {}
41
42    def ping(self):
43        print('ping()')
44
45    def add(self, n1, n2):
46        print('add(%d,%d)' % (n1, n2))
47        return n1 + n2
48
49    def calculate(self, logid, work):
50        print('calculate(%d, %r)' % (logid, work))
51
52        if work.op == Operation.ADD:
53            val = work.num1 + work.num2
54        elif work.op == Operation.SUBTRACT:
55            val = work.num1 - work.num2
56        elif work.op == Operation.MULTIPLY:
57            val = work.num1 * work.num2
58        elif work.op == Operation.DIVIDE:
59            if work.num2 == 0:
60                raise InvalidOperation(work.op, 'Cannot divide by 0')
61            val = work.num1 / work.num2
62        else:
63            raise InvalidOperation(work.op, 'Invalid operation')
64
65        log = SharedStruct()
66        log.key = logid
67        log.value = '%d' % (val)
68        self.log[logid] = log
69
70        return val
71
72    def getStruct(self, key):
73        print('getStruct(%d)' % (key))
74        return self.log[key]
75
76    def zip(self):
77        print('zip()')
78
79
80if __name__ == '__main__':
81    handler = CalculatorHandler()
82    processor = Calculator.Processor(handler)
83    transport = TSocket.TServerSocket(host='127.0.0.1', port=9090)
84    tfactory = TTransport.TBufferedTransportFactory()
85    pfactory = TBinaryProtocol.TBinaryProtocolFactory()
86
87    server = TServer.TSimpleServer(processor, transport, tfactory, pfactory)
88
89    # You could do one of these for a multithreaded server
90    # server = TServer.TThreadedServer(
91    #     processor, transport, tfactory, pfactory)
92    # server = TServer.TThreadPoolServer(
93    #     processor, transport, tfactory, pfactory)
94
95    print('Starting the server...')
96    server.serve()
97    print('done.')
98