fix presence bug
[svn42.git] / 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 }