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 22:52:41 +00:00
|
|
|
import sys
|
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__":
|
2018-06-14 22:52:41 +00:00
|
|
|
# get port
|
|
|
|
if len(sys.argv) > 1:
|
|
|
|
port = sys.argv[1]
|
|
|
|
else:
|
|
|
|
port = "8080"
|
|
|
|
|
2018-06-14 19:45:44 +00:00
|
|
|
# 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:52:41 +00:00
|
|
|
print("Listening on port " + port)
|
|
|
|
server.add_insecure_port('[::]:'+port)
|
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)
|