changeset 26681:35f4e0be6025

ZMQ example using python3 and asyncio
author Bob McElrath <bob_git@mcelrath.org>
date Fri, 06 Jan 2017 18:23:44 -0500
parents 346bd9f70860
children d850210d5738
files contrib/zmq/zmq_sub3.py
diffstat 1 files changed, 52 insertions(+), 0 deletions(-) [+]
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/contrib/zmq/zmq_sub3.py	Fri Jan 06 18:23:44 2017 -0500
@@ -0,0 +1,52 @@
+#!/usr/bin/env python3
+# Copyright (c) 2014-2016 The Bitcoin Core developers
+# Distributed under the MIT software license, see the accompanying
+# file COPYING or http://www.opensource.org/licenses/mit-license.php.
+
+import array
+import binascii
+import asyncio, zmq, zmq.asyncio
+import struct
+
+port = 28332
+
+zmqContext = zmq.asyncio.Context()
+
+async def recv_and_process():
+    zmqSubSocket = zmqContext.socket(zmq.SUB)
+    zmqSubSocket.setsockopt_string(zmq.SUBSCRIBE, "hashblock")
+    zmqSubSocket.setsockopt_string(zmq.SUBSCRIBE, "hashtx")
+    zmqSubSocket.setsockopt_string(zmq.SUBSCRIBE, "rawblock")
+    zmqSubSocket.setsockopt_string(zmq.SUBSCRIBE, "rawtx")
+    zmqSubSocket.connect("tcp://127.0.0.1:%i" % port)
+
+    poller = zmq.asyncio.Poller()
+    poller.register(zmqSubSocket, zmq.POLLIN)
+    while True:
+        s = await poller.poll()
+        msg = await s[0][0].recv_multipart()
+        topic = msg[0]
+        body = msg[1]
+        sequence = "Unknown";
+        if len(msg[-1]) == 4:
+          msgSequence = struct.unpack('<I', msg[-1])[-1]
+          sequence = str(msgSequence)
+        if topic == b"hashblock":
+            print('- HASH BLOCK ('+sequence+') -')
+            print(binascii.hexlify(body))
+        elif topic == b"hashtx":
+            print('- HASH TX  ('+sequence+') -')
+            print(binascii.hexlify(body))
+        elif topic == b"rawblock":
+            print('- RAW BLOCK HEADER ('+sequence+') -')
+            print(binascii.hexlify(body[:80]))
+        elif topic == b"rawtx":
+            print('- RAW TX ('+sequence+') -')
+            print(binascii.hexlify(body))
+
+try:
+    loop = zmq.asyncio.ZMQEventLoop()
+    asyncio.set_event_loop(loop)
+    loop.run_until_complete(setup())
+except KeyboardInterrupt:
+    zmqContext.destroy()