[Python学习笔记-006] 使用stomp.py校验JMS selector的正确性

了解Jenkins的人都知道，JMS selector是基于SQL92语法实现的，本文将介绍使用stomp.py和ActiveMQ来校验JMS selector的正确性。

Q: 什么是stomp.py?

A: stomp.py是一个基于STOMP（Simple (or Streaming) Text Orientated Messaging Protocol）协议实现的Python的客户端函数库，用来访问诸如ActiveMQ，RabbitMQ之类的消息服务器。

那么，为了使用stomp.py函数库，我们先安装一个ActiveMQ服务。

1.  下载ActiveMQ

$ export TGZBALL=5.15./apache-activemq-5.15.-bin.tar.gz
$ wget https://archive.apache.org/dist/activemq/$TGZBALL

2. 安装ActiveMQ

$ tar zxvf apache-activemq-5.15.-bin.tar.gz
$ sudo mv apache-activemq-5.15. /opt
$ cd /opt
$ sudo ln -s apache-activemq-5.15. activemq

3. 启动ActiveMQ

$ cd /opt/activemq/bin/linux-x86-
$ sudo ./activemq start
Starting ActiveMQ Broker...

注意ActiveMQ默认的监听端口是8161,

$ netstat -an | egrep
tcp6              :::                 :::*                    LISTEN

如果要关闭ActiveMQ,

$ cd /opt/activemq/bin/linux-x86-
$ sudo ./activemq stop
Stopping ActiveMQ Broker...
Stopped ActiveMQ Broker.

一旦ActiveMQ, 我们可以访问其web界面，默认的登录用户名/密码是admin/admin, 例如：

4. 克隆stomp.py的源代码

$ mkdir /var/tmp/sp
$ cd /var/tmp/sp
$ git clone https://github.com/jasonrbriggs/stomp.py.git

5. 准备一个Python脚本foo.py

 #!/usr/bin/python3
 import sys
 import json
 import collections
 import time
 import stomp

 g_rcvmsg_cnt = 0  # counter of received messages

 class MyListener(stomp.ConnectionListener):
     def on_error(self, headers, message):
         global g_rcvmsg_cnt
         g_rcvmsg_cnt = 0

         print('Oops!! received an error "%s"' % message)

     def on_message(self, headers, message):
         global g_rcvmsg_cnt
         g_rcvmsg_cnt += 1

         print('Bingo! received a message, rcv_cnt = %d' % g_rcvmsg_cnt)
         print('o message headers :\n%s' % json.dumps(headers, indent=4))
         print('o message body    : %s\n' % message)

 def validate():
     if g_rcvmsg_cnt > 0:
         print(">>> PASS")
         return 0
     else:
         print(">>> FAIL")
         return 1

 def usage(argv0):
     sys.stderr.write("Usage: %s <JMS selector> <message headers>\n" % argv0)

 def main(argc, argv):
     if argc != 3:
         usage(argv[0])
         return 1

     jms_selector = argv[1]
     msg_headers = argv[2]

     d_sub_headers = dict()
     d_sub_headers['selector'] = jms_selector
     print('>>> SUB HEADERS', d_sub_headers)

     d_snd_headers = json.loads(msg_headers,
                                object_pairs_hook=collections.OrderedDict)
     print('>>> SND HEADERS', d_snd_headers)
     print()

     conn = stomp.Connection()
     conn.set_listener('', MyListener())
     conn.start()

     # Note default user/password of ActiveMQ is admin/admin
     user = 'admin'
     password = 'admin'
     conn.connect(user, password, wait=True)

     conn.subscribe(destination='/queue/test', id=1, ack='auto',
                    headers=d_sub_headers)

     s_body = ' '.join(argv[1:])
     conn.send(destination='/queue/test', body=s_body, headers=d_snd_headers)

     time.sleep(2)
     conn.disconnect()

     return validate()

 if __name__ == '__main__':
     sys.exit(main(len(sys.argv), sys.argv))

在上面的代码中，

    67        conn.subscribe(destination='/queue/test', id=1, ack='auto',
    68                       headers=d_sub_headers)

L67-68: 根据用户输入的JMS selector进行订阅，

    71        conn.send(destination='/queue/test', body=s_body, headers=d_snd_headers)

L71: 将用户输入的JSON文本作为消息头发送。

6. 使用foo.py测试JMS selector

hdan$ export PYTHONPATH=/var/tmp/sp/stomp.py:$PYTHONPATH
hdan$
hdan$ ./foo.py "Type LIKE 'Tag%'" '{"Type": "tag", "foo": 123}'
>>> SUB HEADERS {'selector': "Type LIKE 'Tag%'"}
>>> SND HEADERS OrderedDict([('Type', 'tag'), ('foo', )])

>>> FAIL
hdan$
hdan$ ./foo.py "Type LIKE 'Tag%'" '{"Type": "Tag", "foo": 123}'
>>> SUB HEADERS {'selector': "Type LIKE 'Tag%'"}
>>> SND HEADERS OrderedDict([('Type', 'Tag'), ('foo', )])

Bingo! received a message, rcv_cnt =
o message headers :
{
    "subscription": "",
    "content-length": "",
    "timestamp": "",
    "destination": "/queue/test",
    "message-id": "ID:huangdan-40431-1550639739706-3:131:-1:1:1",
    "foo": "",
    "Type": "Tag",
    "expires": "",
    "priority": ""
}
o message body    : Type LIKE 'Tag%' {"Type": "Tag", "foo": }

>>> PASS
hdan$
hdan$ ./foo.py "Type LIKE 'Tag%'" '{"Type": "Tag123", "foo": 123}'
>>> SUB HEADERS {'selector': "Type LIKE 'Tag%'"}
>>> SND HEADERS OrderedDict([('Type', 'Tag123'), ('foo', )])

Bingo! received a message, rcv_cnt =
o message headers :
{
    "message-id": "ID:huangdan-40431-1550639739706-3:132:-1:1:1",
    "destination": "/queue/test",
    "timestamp": "",
    "expires": "",
    "priority": "",
    "subscription": "",
    "content-length": "",
    "foo": "",
    "Type": "Tag123"
}
o message body    : Type LIKE 'Tag%' {"Type": "Tag123", "foo": }

>>> PASS
hdan$ 

附录： SQL92的LIKE操作符 (来源戳这里）

A like string consists of a set of characters where the percent (%),
underscore (_) and escape character, such as backslash (\) are
treated differently.

* The percent    (%) matches any number of characters including
                             zero characters
* The underscore (_) matches only one character

e.g.
+-------------+---------------------------------------+
| Like string | Meaning                               |
+-------------+---------------------------------------+
| Topic%      | All strings starting with Topic       |
| %/abc/%     | Any string containing /abc/           |
| Name_       | Any string starting with Name and     |
|             | having exactly one more character     |
| Name\_2     | Only the string Name_2                |
+-------------+---------------------------------------+

参考资料：

• STOMP
• A simple example of using stomp.py