xref: /aosp_15_r20/external/grpc-grpc/examples/python/hellostreamingworld/async_greeter_client.py (revision cc02d7e222339f7a4f6ba5f422e6413f4bd931f2)
1# Copyright 2021 gRPC authors.
2#
3# Licensed under the Apache License, Version 2.0 (the "License");
4# you may not use this file except in compliance with the License.
5# You may obtain a copy of the License at
6#
7#     http://www.apache.org/licenses/LICENSE-2.0
8#
9# Unless required by applicable law or agreed to in writing, software
10# distributed under the License is distributed on an "AS IS" BASIS,
11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12# See the License for the specific language governing permissions and
13# limitations under the License.
14"""The Python AsyncIO implementation of the GRPC hellostreamingworld.MultiGreeter client."""
15
16import asyncio
17import logging
18
19import grpc
20import hellostreamingworld_pb2
21import hellostreamingworld_pb2_grpc
22
23
24async def run() -> None:
25    async with grpc.aio.insecure_channel("localhost:50051") as channel:
26        stub = hellostreamingworld_pb2_grpc.MultiGreeterStub(channel)
27
28        # Read from an async generator
29        async for response in stub.sayHello(
30            hellostreamingworld_pb2.HelloRequest(name="you")
31        ):
32            print(
33                "Greeter client received from async generator: "
34                + response.message
35            )
36
37        # Direct read from the stub
38        hello_stream = stub.sayHello(
39            hellostreamingworld_pb2.HelloRequest(name="you")
40        )
41        while True:
42            response = await hello_stream.read()
43            if response == grpc.aio.EOF:
44                break
45            print(
46                "Greeter client received from direct read: " + response.message
47            )
48
49
50if __name__ == "__main__":
51    logging.basicConfig()
52    asyncio.run(run())
53