From ca5d112ea5fb446d1dd6aa6b665fe306c18a4c28 Mon Sep 17 00:00:00 2001
From: Andrew Svetlov <andrew.svetlov@gmail.com>
Date: Sun, 23 Nov 2014 13:50:35 +0200
Subject: [PATCH] Add example for core level

---
 docs/examples.rst              |  7 ++++
 examples/core_dealer_router.py | 74 ++++++++++++++++++++++++++++++++++
 2 files changed, 81 insertions(+)
 create mode 100644 examples/core_dealer_router.py

diff --git a/docs/examples.rst b/docs/examples.rst
index d95dab2..3145e5c 100644
--- a/docs/examples.rst
+++ b/docs/examples.rst
@@ -5,6 +5,13 @@ There is a list of examples from `aiozmq/examples <https://github.com/aio-libs/a
 
 Every example is a correct tiny python program.
 
+.. _aiozmq-examples-core-dealer-router:
+
+Simple DEALER-ROUTER pair implemented on Core level
+---------------------------------------------------
+
+.. literalinclude:: ../examples/core_dealer_router.py
+
 .. _aiozmq-examples-rpc-rpc:
 
 Remote Procedure Call
diff --git a/examples/core_dealer_router.py b/examples/core_dealer_router.py
new file mode 100644
index 0000000..2ec0fee
--- /dev/null
+++ b/examples/core_dealer_router.py
@@ -0,0 +1,74 @@
+import asyncio
+import aiozmq
+import zmq
+
+
+class ZmqDealerProtocol(aiozmq.ZmqProtocol):
+
+    transport = None
+
+    def __init__(self, queue, on_close):
+        self.queue = queue
+        self.on_close = on_close
+
+    def connection_made(self, transport):
+        self.transport = transport
+
+    def msg_received(self, msg):
+        self.queue.put_nowait(msg)
+
+    def connection_lost(self, exc):
+        self.on_close.set_result(exc)
+
+
+class ZmqRouterProtocol(aiozmq.ZmqProtocol):
+
+    transport = None
+
+    def __init__(self, on_close):
+        self.on_close = on_close
+
+    def connection_made(self, transport):
+        self.transport = transport
+
+    def msg_received(self, msg):
+        self.transport.write(msg)
+
+    def connection_lost(self, exc):
+        self.on_close.set_result(exc)
+
+
+@asyncio.coroutine
+def go():
+    router_closed = asyncio.Future()
+    dealer_closed = asyncio.Future()
+    router, _ = yield from aiozmq.create_zmq_connection(
+        lambda: ZmqRouterProtocol(router_closed),
+        zmq.ROUTER,
+        bind='tcp://127.0.0.1:*')
+
+    addr = next(iter(router.bindings()))
+    queue = asyncio.Queue()
+    dealer, _ = yield from aiozmq.create_zmq_connection(
+        lambda: ZmqDealerProtocol(queue, dealer_closed),
+        zmq.DEALER,
+        connect=addr)
+
+    for i in range(10):
+        msg = (b'data', b'ask', str(i).encode('utf-8'))
+        dealer.write(msg)
+        answer = yield from queue.get()
+        print(answer)
+    dealer.close()
+    yield from dealer_closed
+    router.close()
+    yield from router_closed
+
+
+def main():
+    asyncio.get_event_loop().run_until_complete(go())
+    print("DONE")
+
+
+if __name__ == '__main__':
+    main()