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)  (cmd_chans, sub_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.Req)
  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     sub_sock, err := ctx.Socket(zmq.Sub)
  31     if err != nil {
  32         panic(err)
  33     }
  34     defer func() { if r:= recover(); r != nil { sub_sock.Close(); panic(r) } }()
  35
  36     if err = cmd_sock.Connect(cmd_port); err != nil {
  37         panic(err)
  38     }
  39
  40     if err = sub_sock.Connect(sub_port); err != nil {
  41         panic(err)
  42     }
  43
  44     cmd_chans = cmd_sock.ChannelsBuffer(10)
  45     sub_chans = cmd_sock.ChannelsBuffer(10)
  46
  47     go zmqsHandleError(cmd_chans, sub_chans)
  48     return
  49 }
  50
  51 func zmqsHandleError(cmd_chans, sub_chans *zmq.Channels) {
  52     for {
  53         select {
  54             case cmd_error := <- cmd_chans.Errors():
  55                 cmd_chans.Close()
  56                 sub_chans.Close()
  57                 panic(cmd_error)
  58             case sub_error := <- sub_chans.Errors():
  59                 cmd_chans.Close()
  60                 sub_chans.Close()
  61                 panic(sub_error)
  62         }
  63     }
  64 }