go/r3-eventbroker_zmq/zeromq.go

   1 // (c) Bernhard Tittelbach, 2013
   2
   3 package main
   4
   5 import (
   6     zmq "github.com/vaughan0/go-zmq"
   7     "bytes"
   8     "errors"
   9  )
  10
  11 // ---------- ZeroMQ Code -------------
  12
  13 func ZmqsInit(sub_connect_port, sub_listen_port, pub_port, keylookup_port string)  (ctx *zmq.Context, sub_chans *zmq.Channels, pub_sock *zmq.Socket, keylookup_sock *zmq.Socket) {
  14     var err error
  15     ctx, err = zmq.NewContext()
  16     if err != nil {
  17         panic(err)
  18     }
  19     //close only on later panic, otherwise leave open:
  20     defer func(){ if r:= recover(); r != nil { ctx.Close(); panic(r) } }()
  21
  22     if len(sub_connect_port) > 0 && len(sub_listen_port) > 0 {
  23         sub_sock, err := ctx.Socket(zmq.Sub)
  24         if err != nil {
  25             panic(err)
  26         }
  27         defer func() { if r:= recover(); r != nil { sub_sock.Close(); panic(r) } }()
  28
  29         sub_sock.Subscribe([]byte{}) //subscribe empty filter -> aka to all messages
  30
  31             if err = sub_sock.Bind(sub_listen_port); err != nil {
  32             panic(err)
  33         }
  34
  35             if err = sub_sock.Connect(sub_connect_port); err != nil {
  36             panic(err)
  37         }
  38
  39         sub_chans = sub_sock.ChannelsBuffer(10)
  40         go zmqsHandleError(sub_chans)
  41     } else {
  42         sub_chans = nil
  43     }
  44
  45     if len(pub_port) > 0 {
  46         pub_sock, err = ctx.Socket(zmq.Pub)
  47         if err != nil {
  48             panic(err)
  49         }
  50         defer func() { if r:= recover(); r != nil { pub_sock.Close(); panic(r) } }()
  51
  52         if err = pub_sock.Bind(pub_port); err != nil {
  53             panic(err)
  54         }
  55     } else {
  56         pub_sock = nil
  57     }
  58
  59     if len(keylookup_port) > 0 {
  60         keylookup_sock, err = ctx.Socket(zmq.Req)
  61         if err != nil {
  62             panic(err)
  63         }
  64         defer func() { if r:= recover(); r != nil { keylookup_sock.Close(); panic(r) } }()
  65
  66         if err = keylookup_sock.Connect(keylookup_port); err != nil {
  67             panic(err)
  68         }
  69     } else {
  70         keylookup_sock = nil
  71     }
  72
  73     return
  74 }
  75
  76 func zmqsHandleError(chans *zmq.Channels) {
  77     for error := range(chans.Errors()) {
  78         chans.Close()
  79         panic(error)
  80     }
  81 }
  82
  83 func ZmqsBindNewReplySocket(ctx *zmq.Context, addr string) (chans *zmq.Channels, err error) {
  84     if len(addr) == 0 {
  85         return nil, errors.New("No listen address given")
  86     }
  87     sock, err := ctx.Socket(zmq.Rep)
  88     if err != nil { return nil, err}
  89
  90     if err = sock.Bind(addr); err != nil {
  91         sock.Close()
  92         return nil, err
  93     }
  94
  95     chans = sock.ChannelsBuffer(10)
  96     go zmqsHandleError(chans)
  97
  98     return chans, nil
  99 }
 100
 101 func ZmqsRequestAnswer(sock *zmq.Socket, request [][]byte) (answer [][]byte) {
 102     if err := sock.Send(request); err != nil {
 103         panic(err)
 104     }
 105     parts, err := sock.Recv()
 106     if err != nil {
 107         panic(err)
 108     }
 109     return parts
 110 }
 111
 112 func ZmqsAskQuestionsAndClose(ctx *zmq.Context, addr string, questions [][][]byte) [][][]byte {
 113     if len(addr) == 0 || ctx == nil { return nil }
 114
 115     req_sock, err := ctx.Socket(zmq.Req)
 116     if err != nil {
 117         return nil
 118     }
 119     defer req_sock.Close()
 120
 121     if err = req_sock.Connect(addr); err != nil {
 122         return nil
 123     }
 124
 125     rv := make([][][]byte, len(questions))
 126     for index, q := range(questions) {
 127         rv[index] = ZmqsRequestAnswer(req_sock, q)
 128     }
 129     return rv
 130 }
 131
 132 func LookupCardIdNick(s *zmq.Socket, hexbytes []byte) (string, error) {
 133     answ := ZmqsRequestAnswer(s, [][]byte{hexbytes})
 134     if len(answ) == 0 {
 135         return "", errors.New("Empty reply received")
 136     }
 137     if bytes.Compare(answ[0], []byte("ERROR")) == 0 {
 138         return "", errors.New(string(bytes.Join(answ[1:],[]byte(" "))))
 139     }
 140     if bytes.Compare(answ[0], []byte("RESULT")) != 0 || len(answ) != 3{
 141         return "", errors.New("Unknown reply received")
 142     }
 143     if bytes.Compare(answ[1], hexbytes) != 0 {
 144         return "", errors.New("Wrong reply received")
 145     }
 146     return string(answ[2]), nil
 147 }