door_daemon_zmq

[svn42.git] / go / door_daemon_zmq / zeromq.go

diff --git a/go/door_daemon_zmq/zeromq.go b/go/door_daemon_zmq/zeromq.go
new file mode 100644 (file)
index 0000000..70fe795
--- /dev/null
+++ b/go/door_daemon_zmq/zeromq.go
@@ -0,0 +1,65 @@
+// (c) Bernhard Tittelbach, 2013
+
+package main
+
+import (
+    zmq "github.com/vaughan0/go-zmq"
+ )
+
+// ---------- ZeroMQ Code -------------
+
+func ZmqsInit(cmd_port, pub_port string)  (cmd_chans, pub_chans *zmq.Channels) {
+
+    cmd_ctx, err := zmq.NewContext()
+    if err != nil {
+        panic(err)
+    }
+    //close only on panic, otherwise leave open:
+    defer func(){ if r:= recover(); r != nil { cmd_ctx.Close(); panic(r) } }()
+    
+    pub_ctx, err := zmq.NewContext()
+    if err != nil {
+        panic(err)
+    }
+    defer func() { if r:= recover(); r != nil { pub_ctx.Close(); panic(r) } }()
+    
+    cmd_sock, err := cmd_ctx.Socket(zmq.Rep)
+    if err != nil {
+        panic(err)
+    }
+    defer func() { if r:= recover(); r != nil { cmd_sock.Close(); panic(r) } }()
+
+    pub_sock, err := pub_ctx.Socket(zmq.Pub)
+    if err != nil {
+        panic(err)
+    }
+    defer func() { if r:= recover(); r != nil { pub_sock.Close(); panic(r) } }()
+
+    if err = cmd_sock.Bind(cmd_port); err != nil { // "tcp://*:5555"
+        panic(err)
+    }
+
+    if err = pub_sock.Bind(pub_port); err != nil { // "tcp://*:5556"
+        panic(err)
+    }
+    
+    cmd_chans = cmd_sock.Channels()
+    pub_chans = cmd_sock.Channels()
+    go zmqsHandleError(cmd_chans, pub_chans)
+    return
+}
+
+func zmqsHandleError(cmd_chans, pub_chans *zmq.Channels) {
+    for {
+        select {
+            case cmd_error := <- cmd_chans.Errors():
+                cmd_chans.Close()
+                pub_chans.Close()
+                panic(cmd_error)
+            case pub_error := <- pub_chans.Errors():
+                cmd_chans.Close()
+                pub_chans.Close()
+                panic(pub_error)
+        }
+    }
+}
\ No newline at end of file