def process_get200(self, args, handler_ctx):
result = get200_result()
try:
result.success = self._handler.get200(handler_ctx)
except:
ex = sys.exc_info()[1]
self._event_handler.handlerError(handler_ctx, 'get200', ex)
result = Thrift.TApplicationException(message=str(ex))
return result
def process_query(self, args, handler_ctx):
result = query_result()
try:
self._handler.query(args.s, args.i)
except:
ex = sys.exc_info()[1]
self._event_handler.handlerError(handler_ctx, 'query', ex)
result = Thrift.TApplicationException(message=str(ex))
return result
def process_doBland(self, args, handler_ctx):
result = doBland_result()
try:
self._handler.doBland(handler_ctx)
except:
ex = sys.exc_info()[1]
self._event_handler.handlerError(handler_ctx, 'doBland', ex)
result = Thrift.TApplicationException(message=str(ex))
return result
def process_execute(self, args, handler_ctx):
result = execute_result()
try:
result.success = self._handler.execute(handler_ctx, args.sessionId, args.stmt)
except:
ex = sys.exc_info()[1]
self._event_handler.handlerError(handler_ctx, 'execute', ex)
result = Thrift.TApplicationException(message=str(ex))
return result
def process_authenticate(self, args, handler_ctx):
result = authenticate_result()
try:
result.success = self._handler.authenticate(handler_ctx, args.username, args.password)
except:
ex = sys.exc_info()[1]
self._event_handler.handlerError(handler_ctx, 'authenticate', ex)
result = Thrift.TApplicationException(message=str(ex))
return result
def process_import(self, args, handler_ctx):
result = import_result()
try:
result.success = self._handler.import_PY_RESERVED_KEYWORD()
except:
ex = sys.exc_info()[1]
self._event_handler.handlerError(handler_ctx, 'import', ex)
result = Thrift.TApplicationException(message=repr(ex))
return result
def process_has_arg_docs(self, args, handler_ctx):
result = has_arg_docs_result()
try:
self._handler.has_arg_docs(handler_ctx, args.s, args.i)
except:
ex = sys.exc_info()[1]
self._event_handler.handlerError(handler_ctx, 'has_arg_docs', ex)
result = Thrift.TApplicationException(message=str(ex))
return result
def then_isPrime(self, args, handler_ctx):
result = isPrime_result()
try:
result.success = self._handler.future_isPrime(args.num).result()
except:
ex = sys.exc_info()[1]
self._event_handler.handlerError(handler_ctx, 'isPrime', ex)
result = Thrift.TApplicationException(message=repr(ex))
return result
def process_init(self, args, handler_ctx):
result = init_result()
try:
result.success = self._handler.init(handler_ctx, args.int1)
except:
ex = sys.exc_info()[1]
self._event_handler.handlerError(handler_ctx, 'init', ex)
result = Thrift.TApplicationException(message=repr(ex))
return result
def then_sleep(self, args, handler_ctx):
result = sleep_result()
try:
self._handler.future_sleep().result()
except:
ex = sys.exc_info()[1]
self._event_handler.handlerError(handler_ctx, 'sleep', ex)
result = Thrift.TApplicationException(message=repr(ex))
return result
def then_getResult(self, args, handler_ctx):
result = getResult_result()
try:
result.success = self._handler.future_getResult().result()
except:
ex = sys.exc_info()[1]
self._event_handler.handlerError(handler_ctx, 'getResult', ex)
result = Thrift.TApplicationException(message=repr(ex))
return result
def process_getDataByKey1(self, args, handler_ctx):
result = getDataByKey1_result()
try:
result.success = self._handler.getDataByKey1(handler_ctx, args.key)
except:
ex = sys.exc_info()[1]
self._event_handler.handlerError(handler_ctx, 'getDataByKey1', ex)
result = Thrift.TApplicationException(message=repr(ex))
return result
def process_init(self, args, handler_ctx):
result = init_result()
try:
result.success = self._handler.init(handler_ctx, args.int1, args.int2, args.int3, args.int4, args.int5, args.int6, args.int7, args.int8, args.int9, args.int10, args.int11, args.int12, args.int13, args.int14, args.int15, args.int16)
except:
ex = sys.exc_info()[1]
self._event_handler.handlerError(handler_ctx, 'init', ex)
result = Thrift.TApplicationException(message=str(ex))
return result
def process_remove(self, args, handler_ctx):
result = remove_result()
try:
result.success = self._handler.remove(handler_ctx, args.req)
except:
ex = sys.exc_info()[1]
self._event_handler.handlerError(handler_ctx, 'remove', ex)
result = Thrift.TApplicationException(message=str(ex))
return result
def process_pong(self, args, handler_ctx):
result = pong_result()
try:
self._handler.pong(handler_ctx)
except:
ex = sys.exc_info()[1]
self._event_handler.handlerError(handler_ctx, 'pong', ex)
result = Thrift.TApplicationException(message=repr(ex))
return result
def process_func(self, args, handler_ctx):
result = func_result()
try:
result.success = self._handler.func(args.arg1, args.arg2)
except:
ex = sys.exc_info()[1]
self._event_handler.handlerError(handler_ctx, 'func', ex)
result = Thrift.TApplicationException(message=repr(ex))
return result
def process_echo(self, args, handler_ctx):
result = echo_result()
try:
result.success = self._handler.echo(handler_ctx, args.input)
except:
ex = sys.exc_info()[1]
self._event_handler.handlerError(handler_ctx, 'echo', ex)
result = Thrift.TApplicationException(message=repr(ex))
return result
def process_binary_keyed_map(self, args, handler_ctx):
result = binary_keyed_map_result()
try:
result.success = self._handler.binary_keyed_map(args.r)
except:
ex = sys.exc_info()[1]
self._event_handler.handlerError(handler_ctx, 'binary_keyed_map',
ex)
result = Thrift.TApplicationException(message=repr(ex))
return result
def process_global(self, args, handler_ctx):
result = global_result()
try:
result.success = self._handler.global_PY_RESERVED_KEYWORD(
handler_ctx, args.raise_PY_RESERVED_KEYWORD)
except:
ex = sys.exc_info()[1]
self._event_handler.handlerError(handler_ctx, 'global', ex)
result = Thrift.TApplicationException(message=repr(ex))
return result
def process_add(self, args, handler_ctx):
result = add_result()
try:
result.success = self._handler.add(handler_ctx, args.num1,
args.num2)
except:
ex = sys.exc_info()[1]
self._event_handler.handlerError(handler_ctx, 'add', ex)
result = Thrift.TApplicationException(message=str(ex))
return result
def process_count_activity_multi(self, args, handler_ctx):
result = count_activity_multi_result()
try:
result.success = self._handler.count_activity_multi(handler_ctx, args.context_ids)
except InvalidContextIDException as exc0:
self._event_handler.handlerException(handler_ctx, 'count_activity_multi', exc0)
result.invalid_context_id = exc0
except:
ex = sys.exc_info()[1]
self._event_handler.handlerError(handler_ctx, 'count_activity_multi', ex)
result = Thrift.TApplicationException(message=str(ex))
return result
def process_doRaise(self, args, handler_ctx):
result = doRaise_result()
try:
self._handler.doRaise(handler_ctx)
except Banal as exc0:
self._event_handler.handlerException(handler_ctx, 'doRaise', exc0)
result.b = exc0
except Fiery as exc1:
self._event_handler.handlerException(handler_ctx, 'doRaise', exc1)
result.f = exc1
except:
ex = sys.exc_info()[1]
self._event_handler.handlerError(handler_ctx, 'doRaise', ex)
result = Thrift.TApplicationException(message=str(ex))
return result
def process_get500(self, args, handler_ctx):
result = get500_result()
try:
result.success = self._handler.get500(handler_ctx)
except Fiery as exc0:
self._event_handler.handlerException(handler_ctx, 'get500', exc0)
result.f = exc0
except Banal as exc1:
self._event_handler.handlerException(handler_ctx, 'get500', exc1)
result.b = exc1
except:
ex = sys.exc_info()[1]
self._event_handler.handlerError(handler_ctx, 'get500', ex)
result = Thrift.TApplicationException(message=str(ex))
return result
def execute(self, param):
print("server: execute failure")
raise Thrift.TApplicationException(
type=Thrift.TApplicationException.INTERNAL_ERROR,
message="for execute",
)
def add(self, a, b):
print("server: add failure")
raise Thrift.TApplicationException(
type=Thrift.TApplicationException.INTERNAL_ERROR,
message="for add",
)
def get_time(self, country_1, country_2):
if country_1 not in countries or country_2 not in countries[country_1]:
raise Thrift.TApplicationException(
TApplicationException.INVALID_MESSAGE_TYPE, "Unknown country")
return countries[country_1][country_2]
def poke(self):
print("server: poke failure")
raise Thrift.TApplicationException(
type=Thrift.TApplicationException.INTERNAL_ERROR,
message="for poke",
)
def send(self, *args, **kwargs):
params = inspect.getcallargs(
getattr(service_module.Iface, method_name), self, *args, **kwargs
)
params.pop('self') # self is already known
# $methodName_args is the implicit struct containing the various
# method parameters.
call_args = args_type()
for name, value in params.items():
setattr(call_args, name, value)
tracer = tracing.ClientTracer(channel=self.tchannel)
span, headers = tracer.start_span(
service=service_name, endpoint=method_name, headers={}
)
body = serializer.serialize_body(call_args)
header = serializer.serialize_header(headers)
# Glue for old API.
if hasattr(self.tchannel, 'request'):
tracing.apply_trace_flag(span, self.trace, True)
with span:
response = yield self.tchannel.request(
hostport=self.hostport, service=self.service
).send(
arg1=endpoint,
arg2=header,
arg3=body, # body
headers=self.protocol_headers,
)
body = yield response.get_body()
else:
with span:
response = yield self.tchannel.call(
scheme=schemes.THRIFT,
service=self.service,
arg1=endpoint,
arg2=header,
arg3=body,
hostport=self.hostport,
trace=self.trace,
tracing_span=span
# TODO: Need to handle these!
# headers=self.protocol_headers,
)
body = response.body
call_result = serializer.deserialize_body(body)
if not result_spec:
# void return type and no exceptions allowed
raise gen.Return(None)
for exc_spec in result_spec[1:]:
# May have failed with an exception
exc = getattr(call_result, exc_spec[2])
if exc is not None:
raise exc
if result_spec[0]:
# Non-void return type. Return the result.
success = getattr(call_result, result_spec[0][2])
if success is not None:
raise gen.Return(success)
else:
# No return type specified and no exceptions raised.
raise gen.Return(None)
# Expected a result but nothing was present in the object. Something
# went wrong.
from thrift import Thrift
raise Thrift.TApplicationException(
Thrift.TApplicationException.MISSING_RESULT,
'%s failed: did not receive a result as expected' % method_name
)
