greeter_server_with_reflection.py 1.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647
  1. # Copyright 2018 The 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 reflection-enabled version of gRPC helloworld.Greeter server."""
  15. from concurrent import futures
  16. import logging
  17. import grpc
  18. from grpc_reflection.v1alpha import reflection
  19. import helloworld_pb2
  20. import helloworld_pb2_grpc
  21. class Greeter(helloworld_pb2_grpc.GreeterServicer):
  22. def SayHello(self, request, context):
  23. return helloworld_pb2.HelloReply(message='Hello, %s!' % request.name)
  24. def serve():
  25. server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
  26. helloworld_pb2_grpc.add_GreeterServicer_to_server(Greeter(), server)
  27. SERVICE_NAMES = (
  28. helloworld_pb2.DESCRIPTOR.services_by_name['Greeter'].full_name,
  29. reflection.SERVICE_NAME,
  30. )
  31. reflection.enable_server_reflection(SERVICE_NAMES, server)
  32. server.add_insecure_port('[::]:50051')
  33. server.start()
  34. server.wait_for_termination()
  35. if __name__ == '__main__':
  36. logging.basicConfig()
  37. serve()