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             if err = sub_sock.Bind(sub_listen_port); err != nil {
  30             panic(err)
  31         }
  32
  33             if err = sub_sock.Connect(sub_connect_port); err != nil {
  34             panic(err)
  35         }
  36
  37         sub_chans = sub_sock.ChannelsBuffer(10)
  38         go zmqsHandleError(sub_chans)
  39     } else {
  40         sub_chans = nil
  41     }
  42
  43     if len(pub_port) > 0 {
  44         pub_sock, err := ctx.Socket(zmq.Pub)
  45         if err != nil {
  46             panic(err)
  47         }
  48         defer func() { if r:= recover(); r != nil { pub_sock.Close(); panic(r) } }()
  49
  50         if err = pub_sock.Bind(pub_port); err != nil {
  51             panic(err)
  52         }
  53     } else {
  54         pub_sock = nil
  55     }
  56
  57     if len(keylookup_port) > 0 {
  58         keylookup_sock, err := ctx.Socket(zmq.Req)
  59         if err != nil {
  60             panic(err)
  61         }
  62         defer func() { if r:= recover(); r != nil { keylookup_sock.Close(); panic(r) } }()
  63
  64         if err = keylookup_sock.Connect(keylookup_port); err != nil {
  65             panic(err)
  66         }
  67     } else {
  68         keylookup_sock = nil
  69     }
  70
  71     return
  72 }
  73
  74 func zmqsHandleError(chans *zmq.Channels) {
  75     for error := range(chans.Errors()) {
  76         chans.Close()
  77         panic(error)
  78     }
  79 }
  80
  81 func (sock *zmq.Socket) ZmqsRequestAnswer(request [][]byte) (answer [][]byte) {
  82     if err := sock.Send(request); err != nil {
  83         panic(err)
  84     }
  85     parts, err := sock.Recv()
  86     if err != nil {
  87         panic(err)
  88     }
  89     return parts
  90 }
  91
  92 func (s *zmq.Socket) LookupCardIdNick(hexbytes []byte) (nick string, error) {
  93     answ := s.ZmqsRequestAnswer([][]byte{hexbytes})
  94     if len(answ) == 0 {
  95         return "", errors.New("Empty reply received")
  96     }
  97     if answ[0] == []byte("ERROR") {
  98         return "", errors.New(string(bytes.Join(answ[1:],[]byte(" "))))
  99     }
 100     if answ[0] !=  []byte("RESULT") || len(answ) != 3{
 101         return "", errors.New("Unknown reply received")
 102     }
 103     if answ[1] !=  hexbytes {
 104         return "", errors.New("Wrong reply received")
 105     }
 106     return string(answ[2]), nil
 107 }