def unary_stream(self, request, context): metadata = extract_metadata(context) maybe_echo_metadata(context) message = request.value * (request.multiplier or 1) for i in range(request.response_count): maybe_sleep(request) yield ExampleReply(message=message, seqno=i + 1, metadata=metadata)
def stream_stream(self, request, context): metadata = extract_metadata(context) maybe_echo_metadata(context) for index, req in enumerate(request): maybe_sleep(req) message = req.value * (req.multiplier or 1) yield ExampleReply(message=message, seqno=index + 1, metadata=metadata)
def stream_error(self, request, context): metadata = extract_metadata(context) maybe_echo_metadata(context) message = request.value * (request.multiplier or 1) for i in range(request.response_count): maybe_sleep(request) # raise on the last message if i == request.response_count - 1: raise Error("boom") yield ExampleReply(message=message, seqno=i + 1, metadata=metadata)
def stream_unary(self, request, context): metadata = extract_metadata(context) maybe_echo_metadata(context) messages = [] for index, req in enumerate(request): maybe_sleep(req) message = req.value * (req.multiplier or 1) messages.append(message) return ExampleReply(message=",".join(messages), metadata=metadata)
def stream_grpc_error(self, request, context): metadata = extract_metadata(context) maybe_echo_metadata(context) message = request.value * (request.multiplier or 1) for i in range(request.response_count): maybe_sleep(request) # raise on the last message if i == request.response_count - 1: code = StatusCode.RESOURCE_EXHAUSTED message = "Out of tokens!" raise GrpcError( code=code, message=message, status=make_status(code, message), ) yield ExampleReply(message=message, seqno=i + 1, metadata=metadata)
def unary_unary(self, request, context): metadata = extract_metadata(context) maybe_echo_metadata(context) maybe_sleep(request) message = request.value * (request.multiplier or 1) return ExampleReply(message=message, metadata=metadata)