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 }
  48
  49 func ZmqsRequestAnswer(sock *zmq.Socket, request [][]byte) (answer [][]byte) {
  50     if err := sock.Send(request); err != nil {
  51         panic(err)
  52     }
  53     parts, err := sock.Recv()
  54     if err != nil {
  55         panic(err)
  56     }
  57     return parts
  58 }
  59
  60 func ZmqsAskQuestionsAndClose(ctx *zmq.Context, addr string, questions [][][]byte) [][][]byte {
  61     if len(addr) == 0 || ctx == nil { return nil }
  62
  63     req_sock, err := ctx.Socket(zmq.Req)
  64     if err != nil {
  65         return nil
  66     }
  67     defer req_sock.Close()
  68
  69     if err = req_sock.Connect(addr); err != nil {
  70         return nil
  71     }
  72
  73     rv := make([][][]byte, len(questions))
  74     for index, q := range(questions) {
  75         rv[index] = ZmqsRequestAnswer(req_sock, q)
  76     }
  77     return rv
  78 }