diff options
author | Bob McElrath <bob_git@mcelrath.org> | 2017-01-06 18:23:44 -0500 |
---|---|---|
committer | Bob McElrath <bob_git@mcelrath.org> | 2017-01-19 17:17:37 -0500 |
commit | 5ea5368b3a71ab91032ed08b919a4bf740cd676b (patch) | |
tree | d1d44bb33259f3cf8fc2fa4584eda7fe6906a881 /contrib/zmq/zmq_sub3.py | |
parent | 82274c02ed2d82537dc55f008a29edb1bc09bbc4 (diff) | |
download | bitcoin-5ea5368b3a71ab91032ed08b919a4bf740cd676b.tar.xz |
ZMQ example using python3 and asyncio
Diffstat (limited to 'contrib/zmq/zmq_sub3.py')
-rwxr-xr-x | contrib/zmq/zmq_sub3.py | 52 |
1 files changed, 52 insertions, 0 deletions
diff --git a/contrib/zmq/zmq_sub3.py b/contrib/zmq/zmq_sub3.py new file mode 100755 index 0000000000..79cb5b484c --- /dev/null +++ b/contrib/zmq/zmq_sub3.py @@ -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() |