def handle(self, message): print('group', message) self.conn.SendMessage(pb.Message( id=shortuuid.uuid(), type=message.type_name, data=json.dumps(vars(message)), ))
def handle(self, message): # extract user id set from rider_id_list target = set() target = target.union(_extract_user_id_list_from( message.rider_id_list)) self.conn.SendMessage( pb.Message(id=shortuuid.uuid(), target=list(target), type=message.type_name, data=json.dumps(vars(message))))
def handle(self, message): # extract user id set from rider_id_list # also add target with all driver's user id target = set() target = target.union(_extract_user_id_list_from( message.rider_id_list)) target = target.union(_get_all_driver_user_id_list()) self.conn.SendMessage( pb.Message( id=shortuuid.uuid(), target=list(target), type=message.type_name, data=json.dumps(vars(message)), ))
def StreamMessage(self, request_iterator, context): self._record_peer(context=context) latest_message_index = len(self.__message_list) while True: while len(self.__message_list) > latest_message_index: msg = self.__message_list[latest_message_index] latest_message_index += 1 yield pb.Message( id=msg.id, type=msg.type, data=msg.data, )
def StreamMessage(self, request, context): self.__record_peer(context=context, conn_id=request.id) latest_message_index = len(self.__message_list) while True: while len(self.__message_list) > latest_message_index: msg = self.__message_list[latest_message_index] latest_message_index += 1 # if the message is not for this connection, continue print('message.target', msg.target) if request.id not in msg.target: print('not my message {}'.format(request.id)) continue yield pb.Message( id=msg.id, type=msg.type, data=msg.data, )