go/door_client_zmq/zeromq.go

   1 // (c) Bernhard Tittelbach, 2013
   2
   3 package main
   4
   5 import (
   6     zmq "github.com/vaughan0/go-zmq"
   7     "time"
   8  )
   9
  10 // ---------- ZeroMQ Code -------------
  11
  12 func ZmqsInit(cmd_port, sub_port string)  (ctx *zmq.Context, cmd_chans, sub_chans *zmq.Channels) {
  13     var err error
  14     ctx, err = zmq.NewContext()
  15     if err != nil {
  16         panic(err)
  17     }
  18     //close only on later panic, otherwise leave open:
  19     defer func(){ if r:= recover(); r != nil { ctx.Close(); panic(r) } }()
  20
  21     if len(cmd_port) > 0 {
  22         cmd_sock, err := ctx.Socket(zmq.Req)
  23         if err != nil {
  24             panic(err)
  25         }
  26         defer func() { if r:= recover(); r != nil { cmd_sock.Close(); panic(r) } }()
  27
  28         cmd_sock.SetRecvTimeout(2 * time.Second)
  29         cmd_sock.SetSendTimeout(2 * time.Second)
  30
  31         if err = cmd_sock.Connect(cmd_port); err != nil {
  32             panic(err)
  33         }
  34
  35         cmd_chans = cmd_sock.ChannelsBuffer(10)
  36         go zmqsHandleError(cmd_chans)
  37     } else {
  38         cmd_chans = nil
  39     }
  40
  41     if len(sub_port) > 0 {
  42         sub_sock, err := ctx.Socket(zmq.Sub)
  43         if err != nil {
  44             panic(err)
  45         }
  46         defer func() { if r:= recover(); r != nil { sub_sock.Close(); panic(r) } }()
  47
  48         sub_sock.Subscribe([]byte{}) //subscribe empty filter -> aka to all messages
  49
  50         if err = sub_sock.Connect(sub_port); err != nil {
  51             panic(err)
  52         }
  53
  54         sub_chans = sub_sock.ChannelsBuffer(10)
  55         go zmqsHandleError(sub_chans)
  56     } else {
  57         sub_chans = nil
  58     }
  59
  60     return
  61 }
  62
  63 func zmqsHandleError(chans *zmq.Channels) {
  64     for error := range(chans.Errors()) {
  65         chans.Close()
  66         panic(error)
  67     }
  68 }