mirror of
https://github.com/unclechu/gRPC-haskell.git
synced 2024-11-30 06:59:43 +01:00
42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
|
from simple_pb2 import *
|
||
|
|
from uuid import uuid4
|
||
|
|
import random
|
||
|
|
import Queue
|
||
|
|
import grpc
|
||
|
|
|
||
|
|
print "Starting python server"
|
||
|
|
|
||
|
|
done_queue = Queue.Queue()
|
||
|
|
|
||
|
|
class SimpleServiceServer(BetaSimpleServiceServicer):
|
||
|
|
def done(self, request, context):
|
||
|
|
global server
|
||
|
|
done_queue.put_nowait(())
|
||
|
|
|
||
|
|
return SimpleServiceDone()
|
||
|
|
|
||
|
|
def normalCall(self, request, context):
|
||
|
|
return SimpleServiceResponse(response = "NormalRequest", num = sum(request.num))
|
||
|
|
|
||
|
|
def clientStreamingCall(self, requests, context):
|
||
|
|
cur_name = ""
|
||
|
|
cur_sum = 0
|
||
|
|
for request in requests:
|
||
|
|
cur_name += request.request
|
||
|
|
cur_sum += sum(request.num)
|
||
|
|
return SimpleServiceResponse(response = cur_name, num = cur_sum)
|
||
|
|
|
||
|
|
def serverStreamingCall(self, request, context):
|
||
|
|
for num in request.num:
|
||
|
|
yield SimpleServiceResponse(response = request.request, num = num)
|
||
|
|
|
||
|
|
def biDiStreamingCall(self, requests, context):
|
||
|
|
for request in requests:
|
||
|
|
yield SimpleServiceResponse(response = request.request, num = sum(request.num))
|
||
|
|
|
||
|
|
server = beta_create_SimpleService_server(SimpleServiceServer())
|
||
|
|
server.add_insecure_port('[::]:50051')
|
||
|
|
server.start()
|
||
|
|
|
||
|
|
done_queue.get()
|