Add working pyzmq socket support

.gitignore

@@ -0,0 +1,2 @@
+venv
+__pycache__

examples/closing.py

@@ -0,0 +1,58 @@
+import trio
+import trzmq
+import zmq
+
+async def req(s, lock, n):
+    i = 1
+    while True:
+        topic = b"ZMQ-Test"
+        message = f"Hello {i}".encode()
+        async with lock:
+            await s.send(b"%b %b" % (topic, message))
+            print(f"{n} Sent: {topic} {message}")
+            reply = await s.recv()
+            print(f"{n} Got response: {reply}")
+        i += 1
+        await trio.sleep(1)
+        if n == 1 and i > 30:
+            s.close()
+            break
+
+async def rep():
+    await trio.sleep(10)
+
+    context = zmq.Context()
+    socket = context.socket(zmq.REP)
+    socket.setsockopt(zmq.SNDHWM, 2)
+    socket.bind("tcp://0.0.0.0:5556")
+
+    s = trzmq.Socket(socket)
+    i = 1
+    while True:
+        msg = await s.recv()
+        await s.send(b'Reply to: ' + msg)
+        if i > 30:
+            s.close()
+            break
+
+async def printer():
+    while True:
+        print("Still alive")
+        await trio.sleep(1)
+
+async def run():
+    context = zmq.Context()
+    socket = context.socket(zmq.REQ)
+    socket.setsockopt(zmq.SNDHWM, 5)
+    socket.connect("tcp://0.0.0.0:5556")
+
+    s = trzmq.Socket(socket)
+    lock = trio.Lock()
+    async with trio.open_nursery() as nursery:
+        nursery.start_soon(req, s, lock, 1)
+        nursery.start_soon(req, s, lock, 2)
+        nursery.start_soon(rep)
+        nursery.start_soon(printer)
+
+
+trio.run(run)

examples/pub.py

@@ -0,0 +1,22 @@
+# publisher
+
+import trzmq
+import trio
+import zmq
+
+async def run():
+    context = zmq.Context()
+    socket = context.socket(zmq.PUB)
+    socket.connect("tcp://0.0.0.0:5556")
+
+    s = trzmq.Socket(socket)
+    i = 1
+    while True:
+        topic = b"ZMQ-Test"
+        message = "Hello, NORM " + str(i) + "..."
+        await s.send(b"%b %b" % (topic, message.encode()))
+        print("%s %s" % (topic, message))
+        i += 1
+        await trio.sleep(1)
+
+trio.run(run)

examples/pubsub.py

@@ -0,0 +1,41 @@
+
+import trzmq
+import trio
+import zmq
+
+
+async def pub(n):
+    context = zmq.Context()
+    socket = context.socket(zmq.PUB)
+    socket.connect("tcp://0.0.0.0:5556")
+
+    s = trzmq.Socket(socket)
+    i = 1
+    while True:
+        topic = b"ZMQ-Test"
+        message = f"Hello, from {n}: {i}..."
+        await s.send(b"%b %b" % (topic, message.encode()))
+        print("%s %s" % (topic, message))
+        i += 1
+        await trio.sleep(1)
+
+
+async def sub():
+    context = zmq.Context()
+    socket = context.socket(zmq.SUB)
+    socket.bind("tcp://0.0.0.0:5556")
+    socket.setsockopt_string(zmq.SUBSCRIBE, '')
+    s = trzmq.Socket(socket)
+    while True:
+        string = await s.recv()
+        print(string)
+
+
+async def run():
+    async with trio.open_nursery() as nursery:
+        nursery.start_soon(sub)
+        nursery.start_soon(pub, 1)
+        nursery.start_soon(pub, 2)
+
+
+trio.run(run)

examples/reqrep.py

@@ -0,0 +1,49 @@
+import trio
+import trzmq
+import zmq
+
+async def req():
+    context = zmq.Context()
+    socket = context.socket(zmq.REQ)
+    socket.setsockopt(zmq.SNDHWM, 2)
+    socket.connect("tcp://0.0.0.0:5556")
+
+    s = trzmq.Socket(socket)
+    i = 1
+    while True:
+        topic = b"ZMQ-Test"
+        message = f"Hello {i}".encode()
+        await s.send(b"%b %b" % (topic, message))
+        print("Sent: %s %s" % (topic, message))
+        reply = await s.recv()
+        print("Got response: %s" % reply)
+        i += 1
+        await trio.sleep(1)
+
+async def rep():
+    await trio.sleep(10)
+
+    context = zmq.Context()
+    socket = context.socket(zmq.REP)
+    socket.setsockopt(zmq.SNDHWM, 2)
+    socket.bind("tcp://0.0.0.0:5556")
+
+    s = trzmq.Socket(socket)
+    i = 1
+    while True:
+        msg = await s.recv()
+        await s.send(b'Reply to: ' + msg)
+
+async def printer():
+    while True:
+        print("Still alive")
+        await trio.sleep(1)
+
+async def run():
+    async with trio.open_nursery() as nursery:
+        nursery.start_soon(req)
+        nursery.start_soon(rep)
+        nursery.start_soon(printer)
+
+
+trio.run(run)

examples/sub.py

@@ -0,0 +1,17 @@
+# subscriber
+
+import trio
+import trzmq
+import zmq
+
+async def run():
+    context = zmq.Context()
+    socket = context.socket(zmq.SUB)
+    socket.bind("tcp://0.0.0.0:5556")
+    socket.setsockopt_string(zmq.SUBSCRIBE, '')
+    s = trzmq.Socket(socket)
+    while True:
+        string = await s.recv()
+        print(string)
+
+trio.run(run)

requirements.txt

@@ -0,0 +1,2 @@
+pyzmq
+trio

trzmq/__init__.py

@@ -5,3 +5,6 @@
 
 from ._proxy import *
 __all__ += _proxy.__all__
+
+from ._socket import *
+__all__ += _socket.__all__