def write_results_after_future( result, event_handler, handler_ctx, seqid, oprot, fn_name, known_exceptions, future, ): """Result/exception handler for asyncio futures.""" try: try: result.success = future.result() reply_type = TMessageType.REPLY except TException as e: for exc_name, exc_type in known_exceptions.items(): setattr(result, exc_name, e) reply_type = TMessageType.REPLY event_handler.handlerException(handler_ctx, fn_name, e) break else: raise except Exception as e: result = TApplicationException(message=str(e)) reply_type = TMessageType.EXCEPTION event_handler.handlerError(handler_ctx, fn_name, e) event_handler.preWrite(handler_ctx, fn_name, result) oprot.writeMessageBegin(fn_name, reply_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush() event_handler.postWrite(handler_ctx, fn_name, result)
def process_sendSupervisorWorkerHeartbeat(self, seqid, iprot, oprot): args = sendSupervisorWorkerHeartbeat_args() args.read(iprot) iprot.readMessageEnd() result = sendSupervisorWorkerHeartbeat_result() try: self._handler.sendSupervisorWorkerHeartbeat(args.heartbeat) msg_type = TMessageType.REPLY except TTransport.TTransportException: raise except AuthorizationException as aze: msg_type = TMessageType.REPLY result.aze = aze except TApplicationException as ex: logging.exception('TApplication exception in handler') msg_type = TMessageType.EXCEPTION result = ex except Exception: logging.exception('Unexpected exception in handler') msg_type = TMessageType.EXCEPTION result = TApplicationException(TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("sendSupervisorWorkerHeartbeat", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_getBotUseInfo(self, seqid, iprot, oprot): args = getBotUseInfo_args() args.read(iprot) iprot.readMessageEnd() result = getBotUseInfo_result() try: result.success = self._handler.getBotUseInfo(args.botMid) msg_type = TMessageType.REPLY except TTransport.TTransportException: raise except TalkException as e: msg_type = TMessageType.REPLY result.e = e except TApplicationException as ex: logging.exception('TApplication exception in handler') msg_type = TMessageType.EXCEPTION result = ex except Exception: logging.exception('Unexpected exception in handler') msg_type = TMessageType.EXCEPTION result = TApplicationException(TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("getBotUseInfo", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_ping(self, seqid, iprot, oprot): args = ping_args() args.read(iprot) iprot.readMessageEnd() result = ping_result() try: result.success = self._handler.ping() msg_type = TMessageType.REPLY except TTransport.TTransportException: raise except AboutToShutDownException as shutdown_exception: msg_type = TMessageType.REPLY result.shutdown_exception = shutdown_exception except TApplicationException as ex: logging.exception('TApplication exception in handler') msg_type = TMessageType.EXCEPTION result = ex except Exception: logging.exception('Unexpected exception in handler') msg_type = TMessageType.EXCEPTION result = TApplicationException(TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("ping", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_notifyLeaveRoom(self, seqid, iprot, oprot): args = notifyLeaveRoom_args() args.read(iprot) iprot.readMessageEnd() result = notifyLeaveRoom_result() try: self._handler.notifyLeaveRoom(args.roomMid) msg_type = TMessageType.REPLY except TTransport.TTransportException: raise except TalkException as e: msg_type = TMessageType.REPLY result.e = e except TApplicationException as ex: logging.exception('TApplication exception in handler') msg_type = TMessageType.EXCEPTION result = ex except Exception: logging.exception('Unexpected exception in handler') msg_type = TMessageType.EXCEPTION result = TApplicationException(TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("notifyLeaveRoom", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_retrieveRequestToken(self, seqid, iprot, oprot): args = retrieveRequestToken_args() args.read(iprot) iprot.readMessageEnd() result = retrieveRequestToken_result() try: result.success = self._handler.retrieveRequestToken(args.carrier) msg_type = TMessageType.REPLY except TTransport.TTransportException: raise except TalkException as e: msg_type = TMessageType.REPLY result.e = e except TApplicationException as ex: logging.exception('TApplication exception in handler') msg_type = TMessageType.EXCEPTION result = ex except Exception: logging.exception('Unexpected exception in handler') msg_type = TMessageType.EXCEPTION result = TApplicationException(TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("retrieveRequestToken", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_failRequestV2(self, seqid, iprot, oprot): args = failRequestV2_args() args.read(iprot) iprot.readMessageEnd() result = failRequestV2_result() try: self._handler.failRequestV2(args.id, args.e) msg_type = TMessageType.REPLY except TTransport.TTransportException: raise except AuthorizationException as aze: msg_type = TMessageType.REPLY result.aze = aze except TApplicationException as ex: logging.exception('TApplication exception in handler') msg_type = TMessageType.EXCEPTION result = ex except Exception: logging.exception('Unexpected exception in handler') msg_type = TMessageType.EXCEPTION result = TApplicationException(TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("failRequestV2", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def _handle_request(self, listener_name, sock, addr): client = TFileObjectTransport(sock.makefile()) itrans = self.tfactory.getTransport(client) otrans = self.tfactory.getTransport(client) iprot = self.pfactory.getProtocol(itrans) oprot = self.pfactory.getProtocol(otrans) try: while True: (name, type, seqid) = iprot.readMessageBegin() request_start = time.time() try: timeout_con = Timeout(self.cfg.timeout, Timeout) timeout_con.start() if name not in self.wsgi._processMap: iprot.skip(TType.STRUCT) iprot.readMessageEnd() x = TApplicationException( TApplicationException.UNKNOWN_METHOD, "Unknown function %s" % (name)) oprot.writeMessageBegin( name, TMessageType.EXCEPTION, seqid) x.write(oprot) oprot.writeMessageEnd() oprot.trans.flush() raise ThriftFuncNotFound else: self.wsgi._processMap[name](self.wsgi, seqid, iprot, oprot) except ThriftFuncNotFound, ex: self.log.error("Unknown function %s" % (name)) self.log.access( addr, name, "FUNC_NOT_FOUND", time.time() - request_start) except Timeout, ex: self.log.error("A greenlet process timeout.") self.log.access( addr, name, "TIMEOUT", time.time() - request_start)
def process_fetchMessageOperations(self, seqid, iprot, oprot): args = fetchMessageOperations_args() args.read(iprot) iprot.readMessageEnd() result = fetchMessageOperations_result() try: result.success = self._handler.fetchMessageOperations(args.localRevision, args.lastOpTimestamp, args.count) msg_type = TMessageType.REPLY except TTransport.TTransportException: raise except TalkException as e: msg_type = TMessageType.REPLY result.e = e except TApplicationException as ex: logging.exception('TApplication exception in handler') msg_type = TMessageType.EXCEPTION result = ex except Exception: logging.exception('Unexpected exception in handler') msg_type = TMessageType.EXCEPTION result = TApplicationException(TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("fetchMessageOperations", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_getCapabilities(self, seqid, iprot, oprot): args = getCapabilities_args() args.read(iprot) iprot.readMessageEnd() result = getCapabilities_result() try: result.success = self._handler.getCapabilities() msg_type = TMessageType.REPLY except TTransport.TTransportException: raise except concrete.services.ttypes.ServicesException as ex: msg_type = TMessageType.REPLY result.ex = ex except TApplicationException as ex: logging.exception('TApplication exception in handler') msg_type = TMessageType.EXCEPTION result = ex except Exception: logging.exception('Unexpected exception in handler') msg_type = TMessageType.EXCEPTION result = TApplicationException(TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("getCapabilities", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_postSnsInvitationMessage(self, seqid, iprot, oprot): args = postSnsInvitationMessage_args() args.read(iprot) iprot.readMessageEnd() result = postSnsInvitationMessage_result() try: self._handler.postSnsInvitationMessage(args.snsIdType, args.snsAccessToken, args.toSnsUserId) msg_type = TMessageType.REPLY except TTransport.TTransportException: raise except TalkException as e: msg_type = TMessageType.REPLY result.e = e except TApplicationException as ex: logging.exception('TApplication exception in handler') msg_type = TMessageType.EXCEPTION result = ex except Exception: logging.exception('Unexpected exception in handler') msg_type = TMessageType.EXCEPTION result = TApplicationException(TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("postSnsInvitationMessage", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_execute(self, seqid, iprot, oprot): args = execute_args() args.read(iprot) iprot.readMessageEnd() result = execute_result() try: result.success = self._handler.execute(args.functionName, args.funcArgs) msg_type = TMessageType.REPLY except TTransport.TTransportException: raise except DRPCExecutionException as e: msg_type = TMessageType.REPLY result.e = e except AuthorizationException as aze: msg_type = TMessageType.REPLY result.aze = aze except TApplicationException as ex: logging.exception('TApplication exception in handler') msg_type = TMessageType.EXCEPTION result = ex except Exception: logging.exception('Unexpected exception in handler') msg_type = TMessageType.EXCEPTION result = TApplicationException(TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("execute", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_sendChatCheckedByWatermark(self, seqid, iprot, oprot): args = sendChatCheckedByWatermark_args() args.read(iprot) iprot.readMessageEnd() result = sendChatCheckedByWatermark_result() try: self._handler.sendChatCheckedByWatermark(args.seq, args.mid, args.watermark, args.sessionId) msg_type = TMessageType.REPLY except TTransport.TTransportException: raise except TalkException as e: msg_type = TMessageType.REPLY result.e = e except TApplicationException as ex: logging.exception('TApplication exception in handler') msg_type = TMessageType.EXCEPTION result = ex except Exception: logging.exception('Unexpected exception in handler') msg_type = TMessageType.EXCEPTION result = TApplicationException(TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("sendChatCheckedByWatermark", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_annotate(self, seqid, iprot, oprot): args = annotate_args() args.read(iprot) iprot.readMessageEnd() result = annotate_result() try: result.success = self._handler.annotate(args.original) msg_type = TMessageType.REPLY except TTransport.TTransportException: raise except concrete.exceptions.ttypes.ConcreteThriftException as ex: msg_type = TMessageType.REPLY result.ex = ex except TApplicationException as ex: logging.exception('TApplication exception in handler') msg_type = TMessageType.EXCEPTION result = ex except Exception: logging.exception('Unexpected exception in handler') msg_type = TMessageType.EXCEPTION result = TApplicationException(TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("annotate", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_create(self, seqid, iprot, oprot): args = create_args() args.read(iprot) iprot.readMessageEnd() result = create_result() try: result.success = self._handler.create(args.request) msg_type = TMessageType.REPLY except (TTransport.TTransportException, KeyboardInterrupt, SystemExit): raise except ServiceUnavailable as service_error: msg_type = TMessageType.REPLY result.service_error = service_error except ImageNotSupported as error: msg_type = TMessageType.REPLY result.error = error except UploadImageError as image_error: msg_type = TMessageType.REPLY result.image_error = image_error except Exception as ex: msg_type = TMessageType.EXCEPTION logging.exception(ex) result = TApplicationException(TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("create", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_checkUserAgeWithDocomo(self, seqid, iprot, oprot): args = checkUserAgeWithDocomo_args() args.read(iprot) iprot.readMessageEnd() result = checkUserAgeWithDocomo_result() try: result.success = self._handler.checkUserAgeWithDocomo(args.openIdRedirectUrl, args.standardAge, args.verifier) msg_type = TMessageType.REPLY except TTransport.TTransportException: raise except TalkException as e: msg_type = TMessageType.REPLY result.e = e except TApplicationException as ex: logging.exception('TApplication exception in handler') msg_type = TMessageType.EXCEPTION result = ex except Exception: logging.exception('Unexpected exception in handler') msg_type = TMessageType.EXCEPTION result = TApplicationException(TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("checkUserAgeWithDocomo", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_similar(self, seqid, iprot, oprot): args = similar_args() args.read(iprot) iprot.readMessageEnd() result = similar_result() try: result.success = self._handler.similar(args.query, args.n, args.sorting) msg_type = TMessageType.REPLY except TTransport.TTransportException: raise except TSearchException as error: msg_type = TMessageType.REPLY result.error = error except TApplicationException as ex: logging.exception('TApplication exception in handler') msg_type = TMessageType.EXCEPTION result = ex except Exception: logging.exception('Unexpected exception in handler') msg_type = TMessageType.EXCEPTION result = TApplicationException(TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("similar", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_getLocalAssignmentForStorm(self, seqid, iprot, oprot): args = getLocalAssignmentForStorm_args() args.read(iprot) iprot.readMessageEnd() result = getLocalAssignmentForStorm_result() try: result.success = self._handler.getLocalAssignmentForStorm(args.id) msg_type = TMessageType.REPLY except TTransport.TTransportException: raise except NotAliveException as e: msg_type = TMessageType.REPLY result.e = e except AuthorizationException as aze: msg_type = TMessageType.REPLY result.aze = aze except TApplicationException as ex: logging.exception('TApplication exception in handler') msg_type = TMessageType.EXCEPTION result = ex except Exception: logging.exception('Unexpected exception in handler') msg_type = TMessageType.EXCEPTION result = TApplicationException(TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("getLocalAssignmentForStorm", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_getAllServersWithPreferredServer(self, seqid, iprot, oprot): args = getAllServersWithPreferredServer_args() args.read(iprot) iprot.readMessageEnd() result = getAllServersWithPreferredServer_result() try: result.success = self._handler.getAllServersWithPreferredServer(args.serverTypes, args.serverGroups, args.failedServers) msg_type = TMessageType.REPLY except TTransport.TTransportException: raise except SnappyException as error: msg_type = TMessageType.REPLY result.error = error except TApplicationException as ex: logging.exception('TApplication exception in handler') msg_type = TMessageType.EXCEPTION result = ex except Exception: logging.exception('Unexpected exception in handler') msg_type = TMessageType.EXCEPTION result = TApplicationException(TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("getAllServersWithPreferredServer", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process(self, iprot, oprot): (name, type, seqid) = yield gen.Task(iprot.readMessageBegin) if name not in self._processMap: yield gen.Task(iprot.skip,TType.STRUCT) yield gen.Task(iprot.readMessageEnd) x = TApplicationException(TApplicationException.UNKNOWN_METHOD, 'Unknown function %s' % (name)) oprot.writeMessageBegin(name, TMessageType.EXCEPTION, seqid) x.write(oprot) oprot.writeMessageEnd() oprot.trans.flush() else: yield gen.Task(self._processMap[name], self, seqid, iprot, oprot)
def process(self, iprot, oprot): (name, type, seqid) = iprot.readMessageBegin() if name not in self._processMap: iprot.skip(TType.STRUCT) iprot.readMessageEnd() x = TApplicationException(TApplicationException.UNKNOWN_METHOD, 'Unknown function %s' % (name)) oprot.writeMessageBegin(name, TMessageType.EXCEPTION, seqid) x.write(oprot) oprot.writeMessageEnd() oprot.trans.flush() return else: return self._processMap[name](self, seqid, iprot, oprot)
def nested(self, iprot, oprot, server_ctx=None): name, _, seqid = iprot.readMessageBegin() if isinstance(name, bytes): name = name.decode("utf8") if name not in self._processMap: iprot.skip(TType.STRUCT) iprot.readMessageEnd() x = TApplicationException(TApplicationException.UNKNOWN_METHOD, "Unknown function {!r}".format(name)) oprot.writeMessageBegin(name, TMessageType.EXCEPTION, seqid) x.write(oprot) oprot.writeMessageEnd() oprot.trans.flush() else: yield from self._processMap[name](self, seqid, iprot, oprot, server_ctx)
def nested(self, iprot, oprot, server_ctx=None): (name, type, seqid) = iprot.readMessageBegin() name = name.decode() if name not in self._processMap: iprot.skip(TType.STRUCT) iprot.readMessageEnd() x = TApplicationException(TApplicationException.UNKNOWN_METHOD, 'Unknown function %s' % (name)) oprot.writeMessageBegin(name, TMessageType.EXCEPTION, seqid) x.write(oprot) oprot.writeMessageEnd() oprot.trans.flush() else: yield from self._processMap[name](self, seqid, iprot, oprot, server_ctx)
def handle_request(self, proto_input, queue, request_meta): """ 从 proto_input中读取数据,然后调用processor处理请求,结果暂时缓存在内存中, 最后一口气交给 queue, 由专门的 greenlet将数据写回到socket上 request_meta = (name, type, seqid, start_time) """ start_time0 = time.time() # 1. 获取一个可用的trans_output if len(self.out_protocols) > 0: trans_output, proto_output = self.out_protocols.popleft() trans_output.prepare_4_frame() # 预留4个字节的Frame Size else: trans_output = TCyMemoryBuffer() trans_output.prepare_4_frame() proto_output = TCyBinaryProtocol(trans_output, client=False) # 无状态的 try: # 2.1 处理正常的请求 self.processor.process(proto_input, proto_output) queue.put(trans_output) except Exception as e: # 2.2 处理异常(主要是结果序列化时参数类型不对的情况) trans_output.prepare_4_frame() name = request_meta[0] seqId = request_meta[2] msg = '%s, Exception: %s, Trace: %s' % (name, e, traceback.format_exc()) x = TApplicationException(TApplicationException.INVALID_PROTOCOL, msg) proto_output.writeMessageBegin(name, TMessageType.EXCEPTION, seqId) x.write(proto_output) proto_output.writeMessageEnd() proto_output.trans.flush() queue.put(trans_output) finally: start_time = request_meta[3] now = time.time() elapsed = now - start_time if elapsed > 2: # 做异常记录 exception_logger.info("Exception Request: %s %s seqId: %s_%s, Elaspsed: %.3f, Execute: %.3f", request_meta[0], request_meta[1], self.pid, request_meta[2], elapsed, now - start_time0) # 3. 回收 transport 和 protocol self.out_protocols.append((trans_output, proto_output))
def process_getDllCaptchaFromImageBinary(self, seqid, iprot, oprot): args = getDllCaptchaFromImageBinary_args() args.read(iprot) iprot.readMessageEnd() result = getDllCaptchaFromImageBinary_result() try: result.success = self._handler.getDllCaptchaFromImageBinary(args.path) msg_type = TMessageType.REPLY except (TTransport.TTransportException, KeyboardInterrupt, SystemExit): raise except Exception as ex: msg_type = TMessageType.EXCEPTION logging.exception(ex) result = TApplicationException(TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("getDllCaptchaFromImageBinary", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_matchGPS2Link(self, seqid, iprot, oprot): args = matchGPS2Link_args() args.read(iprot) iprot.readMessageEnd() result = matchGPS2Link_result() try: result.success = self._handler.matchGPS2Link(args.trace, args.link_list) msg_type = TMessageType.REPLY except (TTransport.TTransportException, KeyboardInterrupt, SystemExit): raise except Exception as ex: msg_type = TMessageType.EXCEPTION logging.exception(ex) result = TApplicationException(TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("matchGPS2Link", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_destroy(self, seqid, iprot, oprot): args = destroy_args() args.read(iprot) iprot.readMessageEnd() result = destroy_result() try: self._handler.destroy(args.contactId) msg_type = TMessageType.REPLY except (TTransport.TTransportException, KeyboardInterrupt, SystemExit): raise except Exception as ex: msg_type = TMessageType.EXCEPTION logging.exception(ex) result = TApplicationException(TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("destroy", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_myMethod(self, seqid, iprot, oprot): args = myMethod_args() args.read(iprot) iprot.readMessageEnd() result = myMethod_result() try: self._handler.myMethod(args.first, args.second, args.third, args.fourth) msg_type = TMessageType.REPLY except (TTransport.TTransportException, KeyboardInterrupt, SystemExit): raise except Exception as ex: msg_type = TMessageType.EXCEPTION logging.exception(ex) result = TApplicationException(TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("myMethod", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_stopLogging(self, seqid, iprot, oprot): args = stopLogging_args() args.read(iprot) iprot.readMessageEnd() result = stopLogging_result() try: result.success = self._handler.stopLogging() msg_type = TMessageType.REPLY except (TTransport.TTransportException, KeyboardInterrupt, SystemExit): raise except Exception as ex: msg_type = TMessageType.EXCEPTION logging.exception(ex) result = TApplicationException(TApplicationException.INTERNAL_ERROR, "Internal error") oprot.writeMessageBegin("stopLogging", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_submitSort(self, seqid, iprot, oprot): args = submitSort_args() args.read(iprot) iprot.readMessageEnd() result = submitSort_result() try: self._handler.submitSort(args.sessionId, args.unitIds) msg_type = TMessageType.REPLY except (TTransport.TTransportException, KeyboardInterrupt, SystemExit): raise except Exception as ex: msg_type = TMessageType.EXCEPTION logging.exception(ex) result = TApplicationException(TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("submitSort", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_updateChoice(self, seqid, iprot, oprot): args = updateChoice_args() args.read(iprot) iprot.readMessageEnd() result = updateChoice_result() try: result.success = self._handler.updateChoice( args.choiceUpdateRequest) msg_type = TMessageType.REPLY except (TTransport.TTransportException, KeyboardInterrupt, SystemExit): raise except P13nServiceException as p13nServiceException: msg_type = TMessageType.REPLY result.p13nServiceException = p13nServiceException except Exception as ex: msg_type = TMessageType.EXCEPTION logging.exception(ex) result = TApplicationException( TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("updateChoice", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_initQuery(self, seqid, iprot, oprot): args = initQuery_args() args.read(iprot) iprot.readMessageEnd() result = initQuery_result() try: result.success = self._handler.initQuery(args.selects, args.keyspace, args.table, args.area, args.precision, args.maxResults, args.blockIDs) msg_type = TMessageType.REPLY except (TTransport.TTransportException, KeyboardInterrupt, SystemExit): raise except Exception as ex: msg_type = TMessageType.EXCEPTION logging.exception(ex) result = TApplicationException( TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("initQuery", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_createUser(self, seqid, iprot, oprot): args = createUser_args() args.read(iprot) iprot.readMessageEnd() result = createUser_result() try: self._handler.createUser(args.username, args.password, args.profile) msg_type = TMessageType.REPLY except (TTransport.TTransportException, KeyboardInterrupt, SystemExit): raise except UserExists as err1: msg_type = TMessageType.REPLY result.err1 = err1 except Exception as ex: msg_type = TMessageType.EXCEPTION logging.exception(ex) result = TApplicationException( TApplicationException.INTERNAL_ERROR, "Internal error") oprot.writeMessageBegin("createUser", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_getMusicRecommendDetail(self, seqid, iprot, oprot): args = getMusicRecommendDetail_args() args.read(iprot) iprot.readMessageEnd() result = getMusicRecommendDetail_result() try: result.success = self._handler.getMusicRecommendDetail(args.mid) msg_type = TMessageType.REPLY except TTransport.TTransportException: raise except TApplicationException as ex: logging.exception('TApplication exception in handler') msg_type = TMessageType.EXCEPTION result = ex except Exception: logging.exception('Unexpected exception in handler') msg_type = TMessageType.EXCEPTION result = TApplicationException( TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("getMusicRecommendDetail", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_is_healthy(self, seqid, iprot, oprot): args = is_healthy_args() args.read(iprot) iprot.readMessageEnd() result = is_healthy_result() try: result.success = self._handler.is_healthy() msg_type = TMessageType.REPLY except TTransport.TTransportException: raise except TApplicationException as ex: logging.exception("TApplication exception in handler") msg_type = TMessageType.EXCEPTION result = ex except Exception: logging.exception("Unexpected exception in handler") msg_type = TMessageType.EXCEPTION result = TApplicationException( TApplicationException.INTERNAL_ERROR, "Internal error") oprot.writeMessageBegin("is_healthy", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def nested(self, iprot, oprot, server_ctx=None): (name, type, seqid) = iprot.readMessageBegin() if sys.version_info[0] >= 3: name = name.decode() if name not in self._processMap: iprot.skip(TType.STRUCT) iprot.readMessageEnd() x = TApplicationException(TApplicationException.UNKNOWN_METHOD, 'Unknown function %s' % (name)) oprot.writeMessageBegin(name, TMessageType.EXCEPTION, seqid) x.write(oprot) oprot.writeMessageEnd() oprot.trans.flush() if twisted is True: from twisted.internet import defer return defer.succeed(None) else: ret = self._processMap[name](self, seqid, iprot, oprot, server_ctx) if twisted is True: return ret else: return True
def process_balance_history(self, seqid, iprot, oprot): args = balance_history_args() args.read(iprot) iprot.readMessageEnd() result = balance_history_result() try: result.success = self._handler.balance_history(args.ownerId) msg_type = TMessageType.REPLY except TTransport.TTransportException: raise except TApplicationException as ex: logging.exception('TApplication exception in handler') msg_type = TMessageType.EXCEPTION result = ex except Exception: logging.exception('Unexpected exception in handler') msg_type = TMessageType.EXCEPTION result = TApplicationException( TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("balance_history", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def nested(self, iprot, oprot, server_ctx=None): name, _, seqid = iprot.readMessageBegin() if isinstance(name, bytes): name = name.decode('utf8') if name not in self._processMap: iprot.skip(TType.STRUCT) iprot.readMessageEnd() x = TApplicationException( TApplicationException.UNKNOWN_METHOD, 'Unknown function {!r}'.format(name), ) oprot.writeMessageBegin(name, TMessageType.EXCEPTION, seqid) x.write(oprot) oprot.writeMessageEnd() oprot.trans.flush() else: yield from self._processMap[name]( self, seqid, iprot, oprot, server_ctx, )
def process_ping(self, seqid, iprot, oprot): args = ping_args() args.read(iprot) iprot.readMessageEnd() result = ping_result() try: self._handler.ping() msg_type = TMessageType.REPLY except TTransport.TTransportException: raise except TApplicationException as ex: logging.exception('TApplication exception in handler') msg_type = TMessageType.EXCEPTION result = ex except Exception: logging.exception('Unexpected exception in handler') msg_type = TMessageType.EXCEPTION result = TApplicationException( TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("ping", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_fetchMessageOperations(self, seqid, iprot, oprot): args = fetchMessageOperations_args() args.read(iprot) iprot.readMessageEnd() result = fetchMessageOperations_result() try: result.success = self._handler.fetchMessageOperations( args.localRevision, args.lastOpTimestamp, args.count) msg_type = TMessageType.REPLY except (TTransport.TTransportException, KeyboardInterrupt, SystemExit): raise except TalkException as e: msg_type = TMessageType.REPLY result.e = e except Exception as ex: msg_type = TMessageType.EXCEPTION logging.exception(ex) result = TApplicationException( TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("fetchMessageOperations", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_checkUserAgeWithDocomo(self, seqid, iprot, oprot): args = checkUserAgeWithDocomo_args() args.read(iprot) iprot.readMessageEnd() result = checkUserAgeWithDocomo_result() try: result.success = self._handler.checkUserAgeWithDocomo( args.openIdRedirectUrl, args.standardAge, args.verifier) msg_type = TMessageType.REPLY except (TTransport.TTransportException, KeyboardInterrupt, SystemExit): raise except TalkException as e: msg_type = TMessageType.REPLY result.e = e except Exception as ex: msg_type = TMessageType.EXCEPTION logging.exception(ex) result = TApplicationException( TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("checkUserAgeWithDocomo", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_calculateStats(self, seqid, iprot, oprot): args = calculateStats_args() args.read(iprot) iprot.readMessageEnd() result = calculateStats_result() try: result.success = self._handler.calculateStats(args.numbers) msg_type = TMessageType.REPLY except TTransport.TTransportException: raise except TApplicationException as ex: logging.exception('TApplication exception in handler') msg_type = TMessageType.EXCEPTION result = ex except Exception: logging.exception('Unexpected exception in handler') msg_type = TMessageType.EXCEPTION result = TApplicationException( TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("calculateStats", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_getSnsMyProfile(self, seqid, iprot, oprot): args = getSnsMyProfile_args() args.read(iprot) iprot.readMessageEnd() result = getSnsMyProfile_result() try: result.success = self._handler.getSnsMyProfile( args.snsIdType, args.snsAccessToken) msg_type = TMessageType.REPLY except (TTransport.TTransportException, KeyboardInterrupt, SystemExit): raise except TalkException as e: msg_type = TMessageType.REPLY result.e = e except Exception as ex: msg_type = TMessageType.EXCEPTION logging.exception(ex) result = TApplicationException( TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("getSnsMyProfile", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_createSquareChat(self, seqid, iprot, oprot): args = createSquareChat_args() args.read(iprot) iprot.readMessageEnd() result = createSquareChat_result() try: result.success = self._handler.createSquareChat( args.reqSeq, args.squareChat, args.squareMemberMids) msg_type = TMessageType.REPLY except (TTransport.TTransportException, KeyboardInterrupt, SystemExit): raise except SquareException as e: msg_type = TMessageType.REPLY result.e = e except Exception as ex: msg_type = TMessageType.EXCEPTION logging.exception(ex) result = TApplicationException( TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("createSquareChat", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_retrieveOpenIdAuthUrlWithDocomo(self, seqid, iprot, oprot): args = retrieveOpenIdAuthUrlWithDocomo_args() args.read(iprot) iprot.readMessageEnd() result = retrieveOpenIdAuthUrlWithDocomo_result() try: result.success = self._handler.retrieveOpenIdAuthUrlWithDocomo() msg_type = TMessageType.REPLY except (TTransport.TTransportException, KeyboardInterrupt, SystemExit): raise except TalkException as e: msg_type = TMessageType.REPLY result.e = e except Exception as ex: msg_type = TMessageType.EXCEPTION logging.exception(ex) result = TApplicationException( TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("retrieveOpenIdAuthUrlWithDocomo", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_getJoinedSquares(self, seqid, iprot, oprot): args = getJoinedSquares_args() args.read(iprot) iprot.readMessageEnd() result = getJoinedSquares_result() try: result.success = self._handler.getJoinedSquares( args.continuationToken, args.limit) msg_type = TMessageType.REPLY except (TTransport.TTransportException, KeyboardInterrupt, SystemExit): raise except SquareException as e: msg_type = TMessageType.REPLY result.e = e except Exception as ex: msg_type = TMessageType.EXCEPTION logging.exception(ex) result = TApplicationException( TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("getJoinedSquares", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_saytext(self, seqid, iprot, oprot): args = saytext_args() args.read(iprot) iprot.readMessageEnd() result = saytext_result() try: result.success = self._handler.saytext(args.from_usename, args.msg) msg_type = TMessageType.REPLY except TTransport.TTransportException: raise except TApplicationException as ex: logging.exception('TApplication exception in handler') msg_type = TMessageType.EXCEPTION result = ex except Exception: logging.exception('Unexpected exception in handler') msg_type = TMessageType.EXCEPTION result = TApplicationException( TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("saytext", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_multiGetLastReadMessageIds(self, seqid, iprot, oprot): args = multiGetLastReadMessageIds_args() args.read(iprot) iprot.readMessageEnd() result = multiGetLastReadMessageIds_result() try: result.success = self._handler.multiGetLastReadMessageIds( args.chatIds) msg_type = TMessageType.REPLY except (TTransport.TTransportException, KeyboardInterrupt, SystemExit): raise except TalkException as e: msg_type = TMessageType.REPLY result.e = e except Exception as ex: msg_type = TMessageType.EXCEPTION logging.exception(ex) result = TApplicationException( TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("multiGetLastReadMessageIds", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_neuralNetArchitectureEval(self, seqid, iprot, oprot): args = neuralNetArchitectureEval_args() args.read(iprot) iprot.readMessageEnd() result = neuralNetArchitectureEval_result() try: result.success = self._handler.neuralNetArchitectureEval(args.arch) msg_type = TMessageType.REPLY except TTransport.TTransportException: raise except TApplicationException as ex: logging.exception('TApplication exception in handler') msg_type = TMessageType.EXCEPTION result = ex except Exception: logging.exception('Unexpected exception in handler') msg_type = TMessageType.EXCEPTION result = TApplicationException( TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("neuralNetArchitectureEval", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_getPostProcessingData(self, seqid, iprot, oprot): args = getPostProcessingData_args() args.read(iprot) iprot.readMessageEnd() result = getPostProcessingData_result() try: result.success = self._handler.getPostProcessingData( args.simulationDataSetRef) msg_type = TMessageType.REPLY except (TTransport.TTransportException, KeyboardInterrupt, SystemExit): raise except ThriftDataAccessException as dataAccessException: msg_type = TMessageType.REPLY result.dataAccessException = dataAccessException except Exception as ex: msg_type = TMessageType.EXCEPTION logging.exception(ex) result = TApplicationException( TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("getPostProcessingData", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_inferImage(self, seqid, iprot, oprot): args = inferImage_args() args.read(iprot) iprot.readMessageEnd() result = inferImage_result() try: result.success = self._handler.inferImage(args.image, args.width, args.height) msg_type = TMessageType.REPLY except TTransport.TTransportException: raise except TApplicationException as ex: logging.exception('TApplication exception in handler') msg_type = TMessageType.EXCEPTION result = ex except Exception: logging.exception('Unexpected exception in handler') msg_type = TMessageType.EXCEPTION result = TApplicationException( TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("inferImage", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_add(self, seqid, iprot, oprot): args = add_args() args.read(iprot) iprot.readMessageEnd() result = add_result() msg_type = TMessageType.REPLY try: result.success = yield gen.maybe_future( self._handler.add(args.num1, args.num2)) except TTransport.TTransportException: raise except TApplicationException as ex: logging.exception('TApplication exception in handler') msg_type = TMessageType.EXCEPTION result = ex except Exception: logging.exception('Unexpected exception in handler') msg_type = TMessageType.EXCEPTION result = TApplicationException( TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("add", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_endServiceDelivery(self, seqid, iprot, oprot): args = endServiceDelivery_args() args.read(iprot) iprot.readMessageEnd() result = endServiceDelivery_result() try: self._handler.endServiceDelivery(args.serviceID, args.serviceDeliveryToken, args.unitsReceived) msg_type = TMessageType.REPLY except (TTransport.TTransportException, KeyboardInterrupt, SystemExit): raise except wpthrift_types.ttypes.Error as err: msg_type = TMessageType.REPLY result.err = err except Exception as ex: msg_type = TMessageType.EXCEPTION logging.exception(ex) result = TApplicationException( TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("endServiceDelivery", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_requestCredit(self, seqid, iprot, oprot): args = requestCredit_args() args.read(iprot) iprot.readMessageEnd() result = requestCredit_result() try: result.success = self._handler.requestCredit( args.pesel, args.encryptedkey, args.amount, args.timeInMonths) msg_type = TMessageType.REPLY except TTransport.TTransportException: raise except NoSuchAccount as nsa: msg_type = TMessageType.REPLY result.nsa = nsa except WrongKey as wk: msg_type = TMessageType.REPLY result.wk = wk except UnsupportedCurrency as uc: msg_type = TMessageType.REPLY result.uc = uc except WrongMoney as wm: msg_type = TMessageType.REPLY result.wm = wm except TApplicationException as ex: logging.exception('TApplication exception in handler') msg_type = TMessageType.EXCEPTION result = ex except Exception: logging.exception('Unexpected exception in handler') msg_type = TMessageType.EXCEPTION result = TApplicationException( TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("requestCredit", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def _handle_request(self, listener_name, sock, addr): client = TFileObjectTransport(sock.makefile()) itrans = self.tfactory.getTransport(client) otrans = self.tfactory.getTransport(client) iprot = self.pfactory.getProtocol(itrans) oprot = self.pfactory.getProtocol(otrans) try: while True: (name, type, seqid) = iprot.readMessageBegin() request_start = time.time() try: timeout_con = Timeout(self.cfg.timeout, Timeout) timeout_con.start() if name not in self.wsgi._processMap: iprot.skip(TType.STRUCT) iprot.readMessageEnd() x = TApplicationException( TApplicationException.UNKNOWN_METHOD, "Unknown function %s" % (name)) oprot.writeMessageBegin( name, TMessageType.EXCEPTION, seqid) x.write(oprot) oprot.writeMessageEnd() oprot.trans.flush() raise ThriftFuncNotFound else: self.wsgi._processMap[name](self.wsgi, seqid, iprot, oprot) except ThriftFuncNotFound, ex: self.log.error("Unknown function %s" % (name)) self.log.access( addr, name, "FUNC_NOT_FOUND", time.time() - request_start) break except Timeout, ex: self.log.error("A greenlet process timeout.") self.log.access( addr, name, "TIMEOUT", time.time() - request_start) break
def process_myMethod(self, seqid, iprot, oprot): args = myMethod_args() args.read(iprot) iprot.readMessageEnd() result = myMethod_result() try: self._handler.myMethod(args.first, args.second, args.third, args.fourth) msg_type = TMessageType.REPLY except TTransport.TTransportException: raise except TApplicationException as ex: logging.exception('TApplication exception in handler') msg_type = TMessageType.EXCEPTION result = ex except Exception: logging.exception('Unexpected exception in handler') msg_type = TMessageType.EXCEPTION result = TApplicationException( TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("myMethod", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_cancelTicketExecution(self, seqid, iprot, oprot): args = cancelTicketExecution_args() args.read(iprot) iprot.readMessageEnd() result = cancelTicketExecution_result() try: result.success = self._handler.cancelTicketExecution( args.ticketId, args.mayInterruptIfRunning) msg_type = TMessageType.REPLY except TTransport.TTransportException: raise except TApplicationException as ex: logging.exception('TApplication exception in handler') msg_type = TMessageType.EXCEPTION result = ex except Exception: logging.exception('Unexpected exception in handler') msg_type = TMessageType.EXCEPTION result = TApplicationException( TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("cancelTicketExecution", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_test(self, seqid, iprot, oprot): args = test_args() args.read(iprot) iprot.readMessageEnd() result = test_result() try: result.success = self._handler.test(args.test) msg_type = TMessageType.REPLY except TTransport.TTransportException: raise except TApplicationException as ex: logging.exception('TApplication exception in handler') msg_type = TMessageType.EXCEPTION result = ex except Exception: logging.exception('Unexpected exception in handler: %s', traceback.format_exc()) msg_type = TMessageType.EXCEPTION result = TApplicationException( TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("test", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_route(self, seqid, iprot, oprot): args = route_args() args.read(iprot) iprot.readMessageEnd() result = route_result() try: result.success = self._handler.route(args.fromSolarSystemId, args.toSolarSystemId, args.connections, args.opts) msg_type = TMessageType.REPLY except (TTransport.TTransportException, KeyboardInterrupt, SystemExit): raise except NewEden.ttypes.LogicalError as le: msg_type = TMessageType.REPLY result.le = le except Exception as ex: msg_type = TMessageType.EXCEPTION logging.exception(ex) result = TApplicationException( TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("route", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()
def process_retrieveOpenIdAuthUrlWithDoCoMoRx(self, seqid, iprot, oprot): args = retrieveOpenIdAuthUrlWithDoCoMoRx_args() args.read(iprot) iprot.readMessageEnd() result = retrieveOpenIdAuthUrlWithDoCoMoRx_result() try: result.success = self._handler.retrieveOpenIdAuthUrlWithDoCoMoRx() msg_type = TMessageType.REPLY except TTransport.TTransportException: raise except TApplicationException as ex: logging.exception('TApplication exception in handler') msg_type = TMessageType.EXCEPTION result = ex except Exception: logging.exception('Unexpected exception in handler') msg_type = TMessageType.EXCEPTION result = TApplicationException( TApplicationException.INTERNAL_ERROR, 'Internal error') oprot.writeMessageBegin("retrieveOpenIdAuthUrlWithDoCoMoRx", msg_type, seqid) result.write(oprot) oprot.writeMessageEnd() oprot.trans.flush()