go/r3-netstatus/zeromq.go

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