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.
81 lines
2.5 KiB
81 lines
2.5 KiB
5 months ago
|
# Copyright 2024 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.
|
||
|
|
"""Example gRPC server that applies back-pressure on client"""
|
||
|
|
|
||
|
|
from concurrent import futures
|
||
|
|
import logging
|
||
|
|
import time
|
||
|
|
|
||
|
|
import grpc
|
||
|
|
import helloworld_pb2
|
||
|
|
import helloworld_pb2_grpc
|
||
|
|
import helpers
|
||
|
|
|
||
|
|
_PORT = "50051"
|
||
|
|
|
||
|
|
|
||
|
|
class Greeter(helloworld_pb2_grpc.GreeterServicer):
|
||
|
|
def SayHelloBidiStream(self, request_iterator, unused_context):
|
||
|
|
# Delay read on server side to apply back-pressure on client
|
||
|
|
time.sleep(5)
|
||
|
|
|
||
|
|
bytes_received = 0
|
||
|
|
|
||
|
|
# Read request from client on every iteration
|
||
|
|
for i, request in enumerate(request_iterator, start=1):
|
||
|
|
bytes_received += len(request.name)
|
||
|
|
if (i % 10) == 0:
|
||
|
|
print(
|
||
|
|
f"{helpers.get_current_time()} "
|
||
|
|
f"Request {i}: Received {bytes_received} bytes in total"
|
||
|
|
)
|
||
|
|
|
||
|
|
# Simulate server "work"
|
||
|
|
time.sleep(1)
|
||
|
|
|
||
|
|
# Send a response
|
||
|
|
msg = "Hello!"
|
||
|
|
yield helloworld_pb2.HelloReply(message=msg)
|
||
|
|
|
||
|
|
if (i % 10) == 0:
|
||
|
|
print(
|
||
|
|
f"{helpers.get_current_time()} "
|
||
|
|
f"Request {i}: Sent {bytes_received} bytes in total\n"
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def serve():
|
||
|
|
server = grpc.server(
|
||
|
|
futures.ThreadPoolExecutor(max_workers=10),
|
||
|
|
# Setting server options to minimal, to allow low number of maximum
|
||
|
|
# bytes in the window.
|
||
|
|
# `bdp_probe` is set to 0(false) to disable resizing of window
|
||
|
|
options=[
|
||
|
|
("grpc.http2.max_frame_size", 16384),
|
||
|
|
("grpc.http2.bdp_probe", 0),
|
||
|
|
("grpc.max_concurrent_streams", 1),
|
||
|
|
],
|
||
|
|
)
|
||
|
|
helloworld_pb2_grpc.add_GreeterServicer_to_server(Greeter(), server)
|
||
|
|
|
||
|
|
server.add_insecure_port("[::]:" + _PORT)
|
||
|
|
server.start()
|
||
|
|
print("Server started, listening on " + _PORT)
|
||
|
|
server.wait_for_termination()
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
logging.basicConfig()
|
||
|
|
serve()
|