mirror of
https://github.com/php-win-ext/grpc.git
synced 2026-03-24 17:12:19 +01:00
05f37c8143
This commit resolves #18331. This commit resolves #18256. This commit resolves... another TODO that apparently didn't have an associated github issue. We swap out pubref's implementation of py_proto_library with our own, which more closely mirrors the interface of the internal py_proto_library, taking the descriptor file output of a proto_library rule as input. One minor change in behavior was introduced for simplicity. When a py_proto_library depends on a proto_library with a source proto file in a subdirectory of the bazel package, the import module of the resultant python library will reflect the package, *not* the full directory of the proto file, including both the bazel package and the subdirectories, as pubref did previously. This behavior also more closely mirrors google internal behavior. This commit also introduces a slightly more stringent bazel format script. Buildifier on its own will not take care of long lines, but by running yapf first, we end up with a more legible file. At the moment, there is no sanity check associated with this formatter.
91 lines
2.6 KiB
Python
91 lines
2.6 KiB
Python
# Copyright 2019 The 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.
|
|
"""This example sends out rich error status from server-side."""
|
|
|
|
from concurrent import futures
|
|
import time
|
|
import logging
|
|
import threading
|
|
|
|
import grpc
|
|
from grpc_status import rpc_status
|
|
|
|
from google.protobuf import any_pb2
|
|
from google.rpc import code_pb2, status_pb2, error_details_pb2
|
|
|
|
from examples import helloworld_pb2
|
|
from examples import helloworld_pb2_grpc
|
|
|
|
_ONE_DAY_IN_SECONDS = 60 * 60 * 24
|
|
|
|
|
|
def create_greet_limit_exceed_error_status(name):
|
|
detail = any_pb2.Any()
|
|
detail.Pack(
|
|
error_details_pb2.QuotaFailure(
|
|
violations=[
|
|
error_details_pb2.QuotaFailure.Violation(
|
|
subject="name: %s" % name,
|
|
description="Limit one greeting per person",
|
|
)
|
|
],))
|
|
return status_pb2.Status(
|
|
code=code_pb2.RESOURCE_EXHAUSTED,
|
|
message='Request limit exceeded.',
|
|
details=[detail],
|
|
)
|
|
|
|
|
|
class LimitedGreeter(helloworld_pb2_grpc.GreeterServicer):
|
|
|
|
def __init__(self):
|
|
self._lock = threading.RLock()
|
|
self._greeted = set()
|
|
|
|
def SayHello(self, request, context):
|
|
with self._lock:
|
|
if request.name in self._greeted:
|
|
rich_status = create_greet_limit_exceed_error_status(
|
|
request.name)
|
|
context.abort_with_status(rpc_status.to_status(rich_status))
|
|
else:
|
|
self._greeted.add(request.name)
|
|
return helloworld_pb2.HelloReply(message='Hello, %s!' % request.name)
|
|
|
|
|
|
def create_server(server_address):
|
|
server = grpc.server(futures.ThreadPoolExecutor())
|
|
helloworld_pb2_grpc.add_GreeterServicer_to_server(LimitedGreeter(), server)
|
|
port = server.add_insecure_port(server_address)
|
|
return server, port
|
|
|
|
|
|
def serve(server):
|
|
server.start()
|
|
try:
|
|
while True:
|
|
time.sleep(_ONE_DAY_IN_SECONDS)
|
|
except KeyboardInterrupt:
|
|
server.stop(None)
|
|
|
|
|
|
def main():
|
|
server, unused_port = create_server('[::]:50051')
|
|
serve(server)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
logging.basicConfig()
|
|
main()
|