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