X-Git-Url: https://git.realraum.at/?a=blobdiff_plain;f=go%2Fdoor_client_zmq%2Fzeromq.go;h=8536c66e30d031839e0c7dc576811349359a4c6d;hb=83655ca2cf31b0e0af2196f87efbd0b25aa72f92;hp=4b3b87d25eecb430989492beb5f2ec81dc634d11;hpb=05f5975f0266e628e89cf0f60f84f47d43136747;p=svn42.git

diff --git a/go/door_client_zmq/zeromq.go b/go/door_client_zmq/zeromq.go
index 4b3b87d..8536c66 100644
--- a/go/door_client_zmq/zeromq.go
+++ b/go/door_client_zmq/zeromq.go
@@ -4,62 +4,66 @@ package main
 
 import (
     zmq "github.com/vaughan0/go-zmq"
+    "time"
  )
 
 // ---------- ZeroMQ Code -------------
 
-func ZmqsInit(cmd_port, sub_port string)  (cmd_chans, pub_chans *zmq.Channels) {
-
-    cmd_ctx, err := zmq.NewContext()
+func ZmqsInit(cmd_port, sub_port string)  (ctx *zmq.Context, cmd_chans, sub_chans *zmq.Channels) {
+    var subfilter []byte
+    var err error
+    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) } }()
+    //close only on later panic, otherwise leave open:
+    defer func(){ if r:= recover(); r != nil { 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) } }()
+    if len(cmd_port) > 0 {
+        cmd_sock, err := ctx.Socket(zmq.Req)
+        if err != nil {
+            panic(err)
+        }
+        defer func() { if r:= recover(); r != nil { cmd_sock.Close(); panic(r) } }()
 
-    cmd_sock, err := cmd_ctx.Socket(zmq.Req)
-    if err != nil {
-        panic(err)
-    }
-    defer func() { if r:= recover(); r != nil { cmd_sock.Close(); panic(r) } }()
+        cmd_sock.SetRecvTimeout(2 * time.Second)
+        cmd_sock.SetSendTimeout(2 * time.Second)
 
-    pub_sock, err := pub_ctx.Socket(zmq.Sub)
-    if err != nil {
-        panic(err)
+        if err = cmd_sock.Connect(cmd_port); err != nil {
+            panic(err)
+        }
+    
+        cmd_chans = cmd_sock.ChannelsBuffer(10)
+        go zmqsHandleError(cmd_chans)
+    } else {
+        cmd_chans = nil
     }
-    defer func() { if r:= recover(); r != nil { pub_sock.Close(); panic(r) } }()
 
-    if err = cmd_sock.Bind(cmd_port); err != nil {
-        panic(err)
-    }
+    if len(sub_port) > 0 {
+        sub_sock, err := ctx.Socket(zmq.Sub)
+        if err != nil {
+            panic(err)
+        }
+        defer func() { if r:= recover(); r != nil { sub_sock.Close(); panic(r) } }()
 
-    if err = pub_sock.Bind(sub_port); err != nil {
-        panic(err)
+        sub_sock.Subscribe(subfilter)
+
+        if err = sub_sock.Connect(sub_port); err != nil {
+            panic(err)
+        }
+
+        sub_chans = sub_sock.ChannelsBuffer(10)
+        go zmqsHandleError(sub_chans)
+    } else {
+        sub_chans = nil
     }
 
-    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)
-        }
+func zmqsHandleError(chans *zmq.Channels) {
+    for error := range(chans.Errors()) {
+        chans.Close()
+        panic(error)    
     }
-}
\ No newline at end of file
+}