Esempio n. 1
0
 async def stomp_subscribe(self, topic):
     self.log.debug('STOMP SUBSCRIBE topic=' + topic)
     frame = StompFrame()
     frame.set_command("SUBSCRIBE")
     frame.set_header('destination', topic)
     frame.set_header('id', 'my-id')
     out = StringIO()
     frame.write(out)
     await self.ws.send(out.getvalue().encode('utf-8'))
Esempio n. 2
0
 async def stomp_connect(self, hostname):
     self.log.debug('STOMP CONNECT host=' + hostname)
     frame = StompFrame()
     frame.set_command("CONNECT")
     frame.set_header('accept-version', '1.2')
     frame.set_header('host', hostname)
     out = StringIO()
     frame.write(out)
     await self.ws.send(out.getvalue().encode('utf-8'))
Esempio n. 3
0
 async def stomp_send(self, topic, message):
     print('STOMP SEND topic=' + topic)
     frame = StompFrame()
     frame.set_command("SEND")
     frame.set_header('destination', topic)
     frame.set_content(message)
     out = StringIO()
     frame.write(out)
     await self.ws.send(out.getvalue().encode('utf-8'))
Esempio n. 4
0
 async def stomp_disconnect(self, receipt=None):
     self.log.info('STOMP DISCONNECT receipt=' + receipt)
     frame = StompFrame()
     frame.set_command("DISCONNECT")
     if receipt is not None:
         frame.set_header('receipt', receipt)
     out = StringIO()
     frame.write(out)
     await self.ws.send(out.getvalue().encode('utf-8'))
Esempio n. 5
0
    def on_pika_message(self, channel, method, header, body):
        pika.log.info('PikaCient: Got message from brocker, delivery tag #%i' % method.delivery_tag)

        response = StompFrame.message(
            message=body,
            headers = {'subscription': self.subscription_id}
        )

        self.write_message(response.as_string())
Esempio n. 6
0
 async def stomp_read_message(self):
     while True:
         message = await self.ws.recv()
         s_in = StringIO(message.decode('utf-8'))
         stomp = StompFrame.parse(s_in)
         if stomp.get_command() == 'MESSAGE':
             return stomp.get_content()
         elif stomp.get_command() == 'CONNECTED':
             version = stomp.get_header('version')
             self.log.info('STOMP CONNECTED version=' + version)
         elif stomp.get_command() == 'RECEIPT':
             receipt = stomp.get_header('receipt-id')
             self.log.debug('STOMP RECEIPT id=' + receipt)
         elif stomp.get_command() == 'ERROR':
             self.log.error('STOMP ERROR content=' + stomp.get_content())
Esempio n. 7
0
 async def stomp_send(self, topic, message):
     logger.debug('STOMP SEND topic=' + topic)
     frame = StompFrame()
     frame.set_command("SEND")
     frame.set_header('destination', topic)
     frame.set_header('content-length', str(len(message)))
     frame.set_content(message)
     out = StringIO()
     frame.write(out)
     await self.ws.send(out.getvalue().encode('utf-8'))
     logger.debug('stomp_send completed')
Esempio n. 8
0
    def on_message(self, message):
        pika.log.info('Websocket: Got message from browser')

        request = StompFrame()
        request.from_string(message)

        pika.log.info('Websocket: Stomp ' + request.command)

        if request.is_connect():
            response = StompFrame.connected()

        elif request.is_subscribe():

            self.agent_id = '/'.join(request.get_header('destination').split('/')[1:])
            self.subscription_id = request.get_header('id')

            pika.log.debug(self.agent_id)
            self.application.pika.channel.queue_declare(exclusive=True, queue=self.queue_name, callback=self.on_queue_declared)

            response = StompFrame.ok()

        elif request.is_unsubscribe():
            #TODO same as on_close() but not delete queueu????
            response = StompFrame.ok()

        elif request.is_send():
            pika.log.info('Websocket: Not implemented')
            pika.log.debug(request.body)

            response = StompFrame.ok()

        else:
            pika.log.info('Websocket: Unknown command')
            response = StompFrame.error()

        self.write_message(response.as_string())