forked from nsqio/nsq
-
Notifications
You must be signed in to change notification settings - Fork 0
/
lookupd_requests.go
40 lines (38 loc) · 900 Bytes
/
lookupd_requests.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
package util
import (
"../nsq"
"errors"
"fmt"
"log"
"net/url"
"sync"
)
func GetChannelsForTopic(topic string, lookupdHTTPAddrs []string) ([]string, error) {
channels := make([]string, 0)
var lock sync.Mutex
var wg sync.WaitGroup
success := false
for _, addr := range lookupdHTTPAddrs {
wg.Add(1)
endpoint := fmt.Sprintf("http://%s/lookup?topic=%s", addr, url.QueryEscape(topic))
log.Printf("LOOKUPD: querying %s", endpoint)
go func(endpiont string) {
data, err := nsq.ApiRequest(endpoint)
lock.Lock()
defer lock.Unlock()
defer wg.Done()
if err != nil {
log.Printf("ERROR: lookupd %s - %s", endpoint, err.Error())
return
}
success = true
c, _ := data.Get("channels").Array()
channels = StringUnion(channels, c)
}(endpoint)
}
wg.Wait()
if success == false {
return nil, errors.New("unable to query any lookupd")
}
return channels, nil
}