Exemplo n.º 1
0
 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)),
         ))
Exemplo n.º 4
0
    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,
                )
Exemplo n.º 5
0
    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,
                )