From 0260687fd4d507dff62cc272a24be76c001649c7 Mon Sep 17 00:00:00 2001
From: Neill Miller <neillm@thecodefactory.org>
Date: Wed, 14 Mar 2018 13:05:23 -0400
Subject: [PATCH] Add python based subscription clients for our block,
 heartbeat, and transaction services

---
 data/subscribe-block.py            | 24 ++++++++++++++++++++++++
 data/subscribe-heartbeat.py        | 22 ++++++++++++++++++++++
 data/subscribe-tx.py               | 22 ++++++++++++++++++++++
 src/services/heartbeat_service.cpp |  2 +-
 4 files changed, 69 insertions(+), 1 deletion(-)
 create mode 100644 data/subscribe-block.py
 create mode 100644 data/subscribe-heartbeat.py
 create mode 100644 data/subscribe-tx.py

diff --git a/data/subscribe-block.py b/data/subscribe-block.py
new file mode 100644
index 00000000..0b827085
--- /dev/null
+++ b/data/subscribe-block.py
@@ -0,0 +1,24 @@
+#!python3
+import sys
+import zmq
+import struct
+from binascii import hexlify, unhexlify
+
+# Connect to the public block service.
+url = 'tcp://mainnet.libbitcoin.net:9093'
+if len(sys.argv) > 1:
+    url = sys.argv[1]
+
+context = zmq.Context()
+socket = context.socket(zmq.SUB)
+socket.connect(url)
+socket.setsockopt(zmq.SUBSCRIBE, '')
+    
+while True:
+    # Collect the response in parts.
+    sequence = struct.unpack('<H', socket.recv())[0]
+    height = struct.unpack('<I', socket.recv())[0]
+    block_data = hexlify(socket.recv()).decode('ascii')
+
+    print '[Seq:{}] Block {} received {}'.format(
+        sequence, height, block_data)
diff --git a/data/subscribe-heartbeat.py b/data/subscribe-heartbeat.py
new file mode 100644
index 00000000..1e035078
--- /dev/null
+++ b/data/subscribe-heartbeat.py
@@ -0,0 +1,22 @@
+#!python3
+import sys
+import zmq
+import struct
+
+# Connect to the public heartbeat service.
+url = 'tcp://mainnet.libbitcoin.net:9092'
+if len(sys.argv) > 1:
+    url = sys.argv[1]
+
+context = zmq.Context()
+socket = context.socket(zmq.SUB)
+socket.connect(url)
+socket.setsockopt(zmq.SUBSCRIBE, '')
+    
+while True:
+    # Collect the response in parts.
+    sequence = struct.unpack('<H', socket.recv())[0]
+    height = struct.unpack('<Q', socket.recv())[0]
+
+    print '[Seq:{}] Heartbeat received with height {}'.format(
+        sequence, height)
diff --git a/data/subscribe-tx.py b/data/subscribe-tx.py
new file mode 100644
index 00000000..3d92c018
--- /dev/null
+++ b/data/subscribe-tx.py
@@ -0,0 +1,22 @@
+#!python3
+import sys
+import zmq
+import struct
+from binascii import hexlify, unhexlify
+
+# Connect to the public transaction service.
+url = 'tcp://mainnet.libbitcoin.net:9094'
+if len(sys.argv) > 1:
+    url = sys.argv[1]
+
+context = zmq.Context()
+socket = context.socket(zmq.SUB)
+socket.connect(url)
+socket.setsockopt(zmq.SUBSCRIBE, '')
+    
+while True:
+    # Collect the response in parts.
+    sequence = struct.unpack('<H', socket.recv())[0]
+    transaction = hexlify(socket.recv()).decode('ascii')
+
+    print '[Seq:{}] Transaction received: {}'.format(sequence, transaction)
diff --git a/src/services/heartbeat_service.cpp b/src/services/heartbeat_service.cpp
index 83f20e7c..cb62f79c 100644
--- a/src/services/heartbeat_service.cpp
+++ b/src/services/heartbeat_service.cpp
@@ -128,7 +128,7 @@ void heartbeat_service::publish(zmq::socket& publisher)
         return;
 
     // [ sequence:2 ]
-    // [ height:4 ]
+    // [ height:8 ]
     zmq::message message;
     message.enqueue_little_endian(++sequence_);
     message.enqueue_little_endian(node_.top_block().height());