microservices-demo/src/recommendationservice/recommendation_server.py

42 lines
1.4 KiB
Python
Raw Normal View History

2018-06-14 19:30:50 +00:00
import grpc
import demo_pb2
import demo_pb2_grpc
2018-06-14 19:45:44 +00:00
from concurrent import futures
import time
2018-06-14 19:30:50 +00:00
2018-06-14 19:45:44 +00:00
class RecommendationService(demo_pb2_grpc.RecommendationServiceServicer):
def ListRecommendations(self, request, context):
2018-06-14 22:31:06 +00:00
print("handling request")
2018-06-14 19:45:44 +00:00
response = demo_pb2.ListRecommendationsResponse()
2018-06-14 22:31:06 +00:00
prod_list = []
for i in range(3):
this_price = demo_pb2.MoneyAmount(decimal=i,
fractional=i)
this_prod = demo_pb2.Product(id=i,
name="test-"+str(i),
description="test product",
picture="test image",
price_usd=this_price)
prod_list.append(this_prod)
response.products.extend(prod_list)
2018-06-14 19:45:44 +00:00
return response
if __name__ == "__main__":
# create gRPC server
server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
# add class to gRPC server
demo_pb2_grpc.add_RecommendationServiceServicer_to_server(RecommendationService(), server)
# start server
2018-06-14 22:31:06 +00:00
print("Listening on port 8081")
server.add_insecure_port('[::]:8081')
2018-06-14 19:45:44 +00:00
server.start()
# keep alive
try:
while True:
2018-06-14 22:31:06 +00:00
time.sleep(10000)
2018-06-14 19:45:44 +00:00
except KeyboardInterrupt:
server.stop(0)