Exemple #1
0
 def service(self) -> purerpc.Service:
     service_obj = purerpc.Service("Greeter")
     service_obj.add_method(
         "SayHello", self.SayHello,
         purerpc.RPCSignature(
             purerpc.Cardinality.UNARY_UNARY,
             generated.greeter_pb2.HelloRequest,
             generated.greeter_pb2.HelloReply,
         ))
     service_obj.add_method(
         "SayHelloGoodbye", self.SayHelloGoodbye,
         purerpc.RPCSignature(
             purerpc.Cardinality.UNARY_STREAM,
             generated.greeter_pb2.HelloRequest,
             generated.greeter_pb2.HelloReply,
         ))
     service_obj.add_method(
         "SayHelloToMany", self.SayHelloToMany,
         purerpc.RPCSignature(
             purerpc.Cardinality.STREAM_STREAM,
             generated.greeter_pb2.HelloRequest,
             generated.greeter_pb2.HelloReply,
         ))
     service_obj.add_method(
         "SayHelloToManyAtOnce", self.SayHelloToManyAtOnce,
         purerpc.RPCSignature(
             purerpc.Cardinality.STREAM_UNARY,
             generated.greeter_pb2.HelloRequest,
             generated.greeter_pb2.HelloReply,
         ))
     return service_obj
Exemple #2
0
 def service(self) -> purerpc.Service:
     service_obj = purerpc.Service("challenges.Challenge2")
     service_obj.add_method(
         "create", self.create,
         purerpc.RPCSignature(
             purerpc.Cardinality.UNARY_UNARY,
             api_pb2.ChallengeRequest2,
             api_pb2.ChallengeResponse,
         ))
     return service_obj
Exemple #3
0
def purerpc_server_wrong_method_name_port(greeter_pb2):
    service = purerpc.Service("Greeter")

    @service.rpc("SomeOtherMethod")
    async def say_hello(
            message: greeter_pb2.HelloRequest) -> greeter_pb2.HelloReply:
        return greeter_pb2.HelloReply(message="Hello, " + message.name)

    with run_purerpc_service_in_process(service) as port:
        yield port
Exemple #4
0
def purerpc_server_wrong_service_name_port(greeter_pb2):
    service = purerpc.Service("some_package.SomeWrongServiceName")

    @service.rpc("SayHello")
    async def say_hello(
            message: greeter_pb2.HelloRequest) -> greeter_pb2.HelloReply:
        return greeter_pb2.HelloReply(message="Hello, " + message.name)

    with run_purerpc_service_in_process(service) as port:
        yield port
Exemple #5
0
def purerpc_port(greeter_pb2):
    service = purerpc.Service("Greeter")

    @service.rpc("SayHello")
    async def say_hello(
            message: greeter_pb2.HelloRequest) -> greeter_pb2.HelloReply:
        status_code_tuple = pickle.loads(base64.b64decode(message.name))
        raise status_code_tuple[0](status_code_tuple[2])

    with run_purerpc_service_in_process(service) as port:
        yield port
Exemple #6
0
def purerpc_simple_greeter_port(greeter_pb2):
    service = purerpc.Service("Greeter")

    @service.rpc("SayHello")
    async def say_hello(
            message: greeter_pb2.HelloRequest) -> greeter_pb2.HelloReply:
        return greeter_pb2.HelloReply(message="Hello, " + message.name)

    @service.rpc("SayHelloGoodbye")
    @async_generator
    async def say_hello_goodbye(
        message: greeter_pb2.HelloRequest
    ) -> purerpc.Stream[greeter_pb2.HelloReply]:
        await yield_(greeter_pb2.HelloReply(message="Hello, " + message.name))
        await yield_(greeter_pb2.HelloReply(message="Goodbye, " + message.name)
                     )

    @service.rpc("SayHelloToManyAtOnce")
    async def say_hello_to_many_at_once(
        messages: purerpc.Stream[greeter_pb2.HelloRequest]
    ) -> greeter_pb2.HelloReply:
        names = []
        async for message in messages:
            names.append(message.name)
        return greeter_pb2.HelloReply(message="Hello, " + ', '.join(names))

    @service.rpc("SayHelloToMany")
    @async_generator
    async def say_hello_to_many(
        messages: purerpc.Stream[greeter_pb2.HelloRequest]
    ) -> purerpc.Stream[greeter_pb2.HelloReply]:
        async for message in messages:
            await yield_(
                greeter_pb2.HelloReply(message="Hello, " + message.name))

    with run_purerpc_service_in_process(service) as port:
        yield port
Exemple #7
0
 def service(self) -> purerpc.Service:
     service_obj = purerpc.Service("challenges.Challenge")
     service_obj.add_method(
         "create", self.create,
         purerpc.RPCSignature(
             purerpc.Cardinality.UNARY_UNARY,
             api_pb2.ChallengeRequest,
             api_pb2.ChallengeResponse,
         ))
     service_obj.add_method(
         "bulk_create", self.bulk_create,
         purerpc.RPCSignature(
             purerpc.Cardinality.STREAM_STREAM,
             api_pb2.ChallengeRequest,
             api_pb2.ChallengeResponse,
         ))
     service_obj.add_method(
         "list", self.list,
         purerpc.RPCSignature(
             purerpc.Cardinality.UNARY_STREAM,
             google.protobuf.empty_pb2.Empty,
             api_pb2.ChallengeResponse,
         ))
     return service_obj
Exemple #8
0
def dummy_server_port():
    with run_purerpc_service_in_process(purerpc.Service("Greeter")) as port:
        yield port