mirror of https://github.com/grpc/grpc.git
The C based gRPC (C++, Python, Ruby, Objective-C, PHP, C#)
https://grpc.io/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
123 lines
4.7 KiB
123 lines
4.7 KiB
# Copyright 2019 gRPC authors. |
|
# |
|
# Licensed under the Apache License, Version 2.0 (the "License"); |
|
# you may not use this file except in compliance with the License. |
|
# You may obtain a copy of the License at |
|
# |
|
# http://www.apache.org/licenses/LICENSE-2.0 |
|
# |
|
# Unless required by applicable law or agreed to in writing, software |
|
# distributed under the License is distributed on an "AS IS" BASIS, |
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|
# See the License for the specific language governing permissions and |
|
# limitations under the License. |
|
"""The example of four ways of data transmission using gRPC in Python.""" |
|
|
|
import time |
|
import grpc |
|
|
|
import demo_pb2_grpc |
|
import demo_pb2 |
|
|
|
__all__ = [ |
|
'simple_method', 'client_streaming_method', 'server_streaming_method', |
|
'bidirectional_streaming_method' |
|
] |
|
|
|
SERVER_ADDRESS = "localhost:23333" |
|
CLIENT_ID = 1 |
|
|
|
# 中文注释和英文翻译 |
|
# Note that this example was contributed by an external user using Chinese comments. |
|
# In all cases, the Chinese comment text is translated to English just below it. |
|
|
|
|
|
# 一元模式(在一次调用中, 客户端只能向服务器传输一次请求数据, 服务器也只能返回一次响应) |
|
# unary-unary(In a single call, the client can only send request once, and the server can |
|
# only respond once.) |
|
def simple_method(stub): |
|
print("--------------Call SimpleMethod Begin--------------") |
|
request = demo_pb2.Request(client_id=CLIENT_ID, |
|
request_data="called by Python client") |
|
response = stub.SimpleMethod(request) |
|
print("resp from server(%d), the message=%s" % |
|
(response.server_id, response.response_data)) |
|
print("--------------Call SimpleMethod Over---------------") |
|
|
|
|
|
# 客户端流模式(在一次调用中, 客户端可以多次向服务器传输数据, 但是服务器只能返回一次响应) |
|
# stream-unary (In a single call, the client can transfer data to the server several times, |
|
# but the server can only return a response once.) |
|
def client_streaming_method(stub): |
|
print("--------------Call ClientStreamingMethod Begin--------------") |
|
|
|
# 创建一个生成器 |
|
# create a generator |
|
def request_messages(): |
|
for i in range(5): |
|
request = demo_pb2.Request( |
|
client_id=CLIENT_ID, |
|
request_data=("called by Python client, message:%d" % i)) |
|
yield request |
|
|
|
response = stub.ClientStreamingMethod(request_messages()) |
|
print("resp from server(%d), the message=%s" % |
|
(response.server_id, response.response_data)) |
|
print("--------------Call ClientStreamingMethod Over---------------") |
|
|
|
|
|
# 服务端流模式(在一次调用中, 客户端只能一次向服务器传输数据, 但是服务器可以多次返回响应) |
|
# unary-stream (In a single call, the client can only transmit data to the server at one time, |
|
# but the server can return the response many times.) |
|
def server_streaming_method(stub): |
|
print("--------------Call ServerStreamingMethod Begin--------------") |
|
request = demo_pb2.Request(client_id=CLIENT_ID, |
|
request_data="called by Python client") |
|
response_iterator = stub.ServerStreamingMethod(request) |
|
for response in response_iterator: |
|
print("recv from server(%d), message=%s" % |
|
(response.server_id, response.response_data)) |
|
|
|
print("--------------Call ServerStreamingMethod Over---------------") |
|
|
|
|
|
# 双向流模式 (在一次调用中, 客户端和服务器都可以向对方多次收发数据) |
|
# stream-stream (In a single call, both client and server can send and receive data |
|
# to each other multiple times.) |
|
def bidirectional_streaming_method(stub): |
|
print( |
|
"--------------Call BidirectionalStreamingMethod Begin---------------") |
|
|
|
# 创建一个生成器 |
|
# create a generator |
|
def request_messages(): |
|
for i in range(5): |
|
request = demo_pb2.Request( |
|
client_id=CLIENT_ID, |
|
request_data=("called by Python client, message: %d" % i)) |
|
yield request |
|
time.sleep(1) |
|
|
|
response_iterator = stub.BidirectionalStreamingMethod(request_messages()) |
|
for response in response_iterator: |
|
print("recv from server(%d), message=%s" % |
|
(response.server_id, response.response_data)) |
|
|
|
print("--------------Call BidirectionalStreamingMethod Over---------------") |
|
|
|
|
|
def main(): |
|
with grpc.insecure_channel(SERVER_ADDRESS) as channel: |
|
stub = demo_pb2_grpc.GRPCDemoStub(channel) |
|
|
|
simple_method(stub) |
|
|
|
client_streaming_method(stub) |
|
|
|
server_streaming_method(stub) |
|
|
|
bidirectional_streaming_method(stub) |
|
|
|
|
|
if __name__ == '__main__': |
|
main()
|
|
|