/
server.py
68 lines (56 loc) · 2.13 KB
/
server.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
import asyncio
from grpclib.utils import graceful_exit
from grpclib.server import Server, Stream
from .helloworld_pb2 import HelloRequest, HelloReply
from .helloworld_grpc import GreeterBase
class Greeter(GreeterBase):
# UNARY_UNARY - simple RPC
async def UnaryUnaryGreeting(
self,
stream: Stream[HelloRequest, HelloReply],
) -> None:
request = await stream.recv_message()
assert request is not None
message = f'Hello, {request.name}!'
await stream.send_message(HelloReply(message=message))
# UNARY_STREAM - response streaming RPC
async def UnaryStreamGreeting(
self,
stream: Stream[HelloRequest, HelloReply],
) -> None:
request = await stream.recv_message()
assert request is not None
await stream.send_message(
HelloReply(message=f'Hello, {request.name}!'))
await stream.send_message(
HelloReply(message=f'Goodbye, {request.name}!'))
# STREAM_UNARY - request streaming RPC
async def StreamUnaryGreeting(
self,
stream: Stream[HelloRequest, HelloReply],
) -> None:
names = []
async for request in stream:
names.append(request.name)
message = 'Hello, {}!'.format(' and '.join(names))
await stream.send_message(HelloReply(message=message))
# STREAM_STREAM - bidirectional streaming RPC
async def StreamStreamGreeting(
self,
stream: Stream[HelloRequest, HelloReply],
) -> None:
async for request in stream:
message = f'Hello, {request.name}!'
await stream.send_message(HelloReply(message=message))
# Send another message to demonstrate responses are not
# coupled to requests.
message = 'Goodbye, all!'
await stream.send_message(HelloReply(message=message))
async def main(*, host: str = '127.0.0.1', port: int = 50051) -> None:
server = Server([Greeter()])
with graceful_exit([server]):
await server.start(host, port)
print(f'Serving on {host}:{port}')
await server.wait_closed()
if __name__ == '__main__':
asyncio.run(main())