go/door_daemon_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, pub_port string)  (ctx *zmq.Context, cmd_chans, pub_chans *zmq.Channels) {
  13
  14     var err error
  15     ctx, err = zmq.NewContext()
  16     if err != nil {
  17         panic(err)
  18     }
  19     //close only on panic, otherwise leave open:
  20     defer func(){ if r:= recover(); r != nil { ctx.Close(); panic(r) } }()
  21
  22     cmd_sock, err := ctx.Socket(zmq.Rep)
  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     pub_sock, err := ctx.Socket(zmq.Pub)
  32     if err != nil {
  33         panic(err)
  34     }
  35     defer func() { if r:= recover(); r != nil { pub_sock.Close(); panic(r) } }()
  36
  37     if err = cmd_sock.Bind(cmd_port); err != nil { // "tcp://*:5555"
  38         panic(err)
  39     }
  40
  41     if err = pub_sock.Bind(pub_port); err != nil { // "tcp://*:5556"
  42         panic(err)
  43     }
  44
  45     cmd_chans = cmd_sock.ChannelsBuffer(10)
  46     pub_chans = cmd_sock.ChannelsBuffer(10)
  47     go zmqsHandleError(cmd_chans, pub_chans)
  48     return
  49 }
  50
  51 func zmqsHandleError(cmd_chans, pub_chans *zmq.Channels) {
  52     for {
  53         select {
  54             case cmd_error := <- cmd_chans.Errors():
  55                 cmd_chans.Close()
  56                 pub_chans.Close()
  57                 panic(cmd_error)
  58             case pub_error := <- pub_chans.Errors():
  59                 cmd_chans.Close()
  60                 pub_chans.Close()
  61                 panic(pub_error)
  62         }
  63     }
  64 }