-
Notifications
You must be signed in to change notification settings - Fork 0
/
consul_catalog_nodes.go
60 lines (51 loc) · 1.27 KB
/
consul_catalog_nodes.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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
package discovery
import (
"encoding/json"
"fmt"
"io/ioutil"
"net/http"
)
const (
catalogNodesPath = "/v1/catalog/nodes" //Lists nodes in a given DC
)
// Node single consul node
type Node struct {
Node string `json:"Node"`
Address string `json:"Address"`
TaggedAddresses map[string]string `json:"TaggedAddresses,omitempty"`
}
// Nodes slice of nodes
type Nodes []Node
// Noder interface
type Noder interface {
GetNodes(addr, path string) (Nodes, error)
}
// DefaultNoder struct to append GetNodes
type DefaultNoder struct{}
// GetNodes make GET request on consul api
func (dn *DefaultNoder) GetNodes(addr, path string) (Nodes, error) {
qs := "?near=_agent"
url := fmt.Sprintf("http://%s%s%s", addr, path, qs)
resp, err := http.Get(url)
if err != nil {
return nil, err
}
defer resp.Body.Close()
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
return nil, err
}
nodes := Nodes{}
if err := json.Unmarshal(body, &nodes); err != nil {
return nil, err
}
return nodes, nil
}
// GetNodes function to get slice of nodes
func GetNodes(n Noder, addr string) (Nodes, error) {
nodes, err := n.GetNodes(addr, catalogNodesPath)
if err != nil {
return nil, fmt.Errorf("Error querying Consul API: %s", err)
}
return nodes, nil
}