22 lines
567 B
Python
22 lines
567 B
Python
import queue
|
|
from collections import deque
|
|
|
|
from federatedscope.core.proto import gRPC_comm_manager_pb2, \
|
|
gRPC_comm_manager_pb2_grpc
|
|
|
|
|
|
class gRPCComServeFunc(gRPC_comm_manager_pb2_grpc.gRPCComServeFuncServicer):
|
|
def __init__(self):
|
|
self.msg_queue = deque()
|
|
|
|
def sendMessage(self, request, context):
|
|
self.msg_queue.append(request)
|
|
|
|
return gRPC_comm_manager_pb2.MessageResponse(msg='ACK')
|
|
|
|
def receive(self):
|
|
while len(self.msg_queue) == 0:
|
|
continue
|
|
msg = self.msg_queue.popleft()
|
|
return msg
|