]>
Commit | Line | Data |
---|---|---|
f67539c2 TL |
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 | ||
22 | import glob | |
23 | import sys | |
24 | sys.path.append('gen-py') | |
25 | sys.path.insert(0, glob.glob('../../lib/py/build/lib*')[0]) | |
26 | ||
27 | from tutorial import Calculator | |
28 | from tutorial.ttypes import InvalidOperation, Operation | |
29 | ||
30 | from shared.ttypes import SharedStruct | |
31 | ||
32 | from thrift.transport import TSocket | |
33 | from thrift.transport import TTransport | |
34 | from thrift.protocol import TBinaryProtocol | |
35 | from thrift.server import TServer | |
36 | ||
37 | ||
38 | class 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 | x = InvalidOperation() | |
61 | x.whatOp = work.op | |
62 | x.why = 'Cannot divide by 0' | |
63 | raise x | |
64 | val = work.num1 / work.num2 | |
65 | else: | |
66 | x = InvalidOperation() | |
67 | x.whatOp = work.op | |
68 | x.why = 'Invalid operation' | |
69 | raise x | |
70 | ||
71 | log = SharedStruct() | |
72 | log.key = logid | |
73 | log.value = '%d' % (val) | |
74 | self.log[logid] = log | |
75 | ||
76 | return val | |
77 | ||
78 | def getStruct(self, key): | |
79 | print('getStruct(%d)' % (key)) | |
80 | return self.log[key] | |
81 | ||
82 | def zip(self): | |
83 | print('zip()') | |
84 | ||
85 | ||
86 | if __name__ == '__main__': | |
87 | handler = CalculatorHandler() | |
88 | processor = Calculator.Processor(handler) | |
89 | transport = TSocket.TServerSocket(host='127.0.0.1', port=9090) | |
90 | tfactory = TTransport.TBufferedTransportFactory() | |
91 | pfactory = TBinaryProtocol.TBinaryProtocolFactory() | |
92 | ||
93 | server = TServer.TSimpleServer(processor, transport, tfactory, pfactory) | |
94 | ||
95 | # You could do one of these for a multithreaded server | |
96 | # server = TServer.TThreadedServer( | |
97 | # processor, transport, tfactory, pfactory) | |
98 | # server = TServer.TThreadPoolServer( | |
99 | # processor, transport, tfactory, pfactory) | |
100 | ||
101 | print('Starting the server...') | |
102 | server.serve() | |
103 | print('done.') |