Merge pull request #170 from pescadores/msgpack

JSON->Msgpack

.github/environment-ci.yml

@@ -7,6 +7,7 @@ dependencies:
   - pip
   - pyzmq>=18.0
   - decorator>=4.0
+  - msgpack-python>=1.0
 
   # optional, but required for testing
   - pytest

.github/environment-minimal.yml

@@ -7,6 +7,7 @@ dependencies:
   - pip
   - pyzmq~=18.0
   - decorator~=4.0
+  - msgpack-python~=1.0
 
   # optional, but required for testing
   - pytest

pescador/zmq_stream.py

@@ -16,11 +16,7 @@
 import multiprocessing as mp
 import zmq
 import numpy as np
-
-try:
-    import ujson as json
-except ImportError:
-    import json
+import msgpack
 
 from .core import Streamer
 from .exceptions import DataError
@@ -51,7 +47,7 @@ def zmq_send_data(socket, data, flags=0, copy=True, track=False):
         payload.append(arr)
 
     # Send the header
-    msg = [json.dumps(header).encode("ascii")]
+    msg = [msgpack.packb(header)]
     msg.extend(payload)
 
     return socket.send_multipart(msg, flags, copy=copy, track=track)
@@ -63,7 +59,7 @@ def zmq_recv_data(socket, flags=0, copy=True, track=False):
 
     msg = socket.recv_multipart(flags=flags, copy=copy, track=track)
 
-    headers = json.loads(msg[0].decode("ascii"))
+    headers = msgpack.unpackb(msg[0], raw=False)
 
     if len(headers) == 0:
         raise StopIteration

setup.cfg

@@ -52,6 +52,7 @@ install_requires =
     pyzmq >= 18.0
     numpy >= 1.9
     decorator >= 4.0
+    msgpack >= 1.0
 
 [options.extras_require]
 docs =