stomp协议

最新推荐文章于 2024-05-14 21:48:10 发布

nimade511

最新推荐文章于 2024-05-14 21:48:10 发布

阅读量487

点赞数

CC 4.0 BY-SA版权

分类专栏： python 文章标签： python stomp

本文链接：https://blog.youkuaiyun.com/nimade511/article/details/53945286

python 专栏收录该内容

12 篇文章

订阅专栏

本文介绍了STOMP协议，一种在消息框架中用于通信的文本协议，其结构类似HTTP，支持跨平台互操作。详细讲解了STOMP的命令，包括SEND、SUBSCRIBE等，并提及其实现。

stomp协议简介

stomp协议是消息框架中的通信协议
通信格式是文本结构类似于HTTP
可以互操作

stomp文本结构

COMMAND
header1:value1
header2:value2

Body^@

命令

SEND
SUBSCRIBE
UNSUBSCRIBE
BEGIN
COMMIT
ABORT
ACK
NACK
DISCONNECT

实现

# -*- coding:utf-8 -*-

import traceback
import gevent
from gevent import socket
from gevent.pool import Pool


class Stomp(object):
    def __init__(self, host, port, username, password):
        self.username = username
        self.password = password
        self.host = host
        self.port = port
        self.pool = Pool(3)
        self.connecting = False
        self._connect_auth()
        hb = self.pool.spawn(self._heart_beating)
        self.pool.start(hb)
    #连接
    def _connect_auth(self):
        print "re auth "
        self.sock = socket.create_connection((self.host, self.port))
        self._command(command="CONNECT", headers={"login": self.username, "passcode": self.password}, body="")
        self.connecting = True
    #心跳，断线重连
    def _heart_beating(self):
        while True:
            print "_heart_beating"
            try:
                if not self.connecting:
                    self._connect_auth()
                self._command(command="CONNECT", headers={"heart-beat":"0,60"}, body="")
                self.connecting = True
                gevent.sleep(30)
            except:
                print traceback.format_exc()
                self.connecting = False
    #发送请求
    def _command(self, command, headers, body):
        new_headers = {}
        new_headers.update(headers)
        frame = self.to_string(command, headers, body)
        self.sock.send(frame)

    #格式化请求，返回Frame
    def to_string(self, command, headers, body):
        header_list = [k + ":" + v + "\n" for k, v in headers.items()]
        headers_line = "".join(header_list)
        frame_string = command + "\n" + headers_line + "\n" + body + "\x00"
        return frame_string

    def send(self, headers, body):
        gl = self.pool.spawn(self._command,"SEND",headers, body)
        self.pool.start(gl)


    def dis_connect(self):
        gevent.sleep(5)
        self.sock.close()
        self.pool.kill()

这是协议文档