mirror of
https://github.com/valitydev/thrift.git
synced 2024-11-07 02:45:22 +00:00
117 lines
3.1 KiB
Python
Executable File
117 lines
3.1 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
#
|
|
# Licensed to the Apache Software Foundation (ASF) under one
|
|
# or more contributor license agreements. See the NOTICE file
|
|
# distributed with this work for additional information
|
|
# regarding copyright ownership. The ASF licenses this file
|
|
# to you under the Apache License, Version 2.0 (the
|
|
# "License"); you may not use this file except in compliance
|
|
# with the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing,
|
|
# software distributed under the License is distributed on an
|
|
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
# KIND, either express or implied. See the License for the
|
|
# specific language governing permissions and limitations
|
|
# under the License.
|
|
#
|
|
|
|
import sys
|
|
import glob
|
|
sys.path.append('gen-py.tornado')
|
|
sys.path.insert(0, glob.glob('../../lib/py/build/lib.*')[0])
|
|
|
|
import logging
|
|
|
|
from tutorial import Calculator
|
|
from tutorial.ttypes import Operation, Work, InvalidOperation
|
|
|
|
from thrift import TTornado
|
|
from thrift.transport import TSocket
|
|
from thrift.transport import TTransport
|
|
from thrift.protocol import TBinaryProtocol
|
|
|
|
from tornado import gen
|
|
from tornado import ioloop
|
|
|
|
|
|
@gen.engine
|
|
def communicate(callback=None):
|
|
# create client
|
|
transport = TTornado.TTornadoStreamTransport('localhost', 9090)
|
|
pfactory = TBinaryProtocol.TBinaryProtocolFactory()
|
|
client = Calculator.Client(transport, pfactory)
|
|
|
|
# open the transport, bail on error
|
|
try:
|
|
yield gen.Task(transport.open)
|
|
except TTransport.TTransportException as ex:
|
|
logging.error(ex)
|
|
if callback:
|
|
callback()
|
|
return
|
|
|
|
# ping
|
|
yield gen.Task(client.ping)
|
|
print "ping()"
|
|
|
|
# add
|
|
sum_ = yield gen.Task(client.add, 1, 1)
|
|
print "1 + 1 = {}".format(sum_)
|
|
|
|
# make a oneway call without a callback (schedule the write and continue
|
|
# without blocking)
|
|
client.zip()
|
|
print "zip() without callback"
|
|
|
|
# make a oneway call with a callback (we'll wait for the stream write to
|
|
# complete before continuing)
|
|
yield gen.Task(client.zip)
|
|
print "zip() with callback"
|
|
|
|
# calculate 1/0
|
|
work = Work()
|
|
work.op = Operation.DIVIDE
|
|
work.num1 = 1
|
|
work.num2 = 0
|
|
|
|
try:
|
|
quotient = yield gen.Task(client.calculate, 1, work)
|
|
print "Whoa? You know how to divide by zero?"
|
|
except InvalidOperation as io:
|
|
print "InvalidOperation: {}".format(io)
|
|
|
|
# calculate 15-10
|
|
work.op = Operation.SUBTRACT
|
|
work.num1 = 15
|
|
work.num2 = 10
|
|
|
|
diff = yield gen.Task(client.calculate, 1, work)
|
|
print "15 - 10 = {}".format(diff)
|
|
|
|
# getStruct
|
|
log = yield gen.Task(client.getStruct, 1)
|
|
print "Check log: {}".format(log.value)
|
|
|
|
# close the transport
|
|
client._transport.close()
|
|
|
|
if callback:
|
|
callback()
|
|
|
|
|
|
def main():
|
|
# create an ioloop, do the above, then stop
|
|
io_loop = ioloop.IOLoop.instance()
|
|
def this_joint():
|
|
communicate(callback=io_loop.stop)
|
|
io_loop.add_callback(this_joint)
|
|
io_loop.start()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|