/
etcd.go
57 lines (46 loc) · 1.22 KB
/
etcd.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
package sources
import (
"github.com/coreos/etcd/Godeps/_workspace/src/golang.org/x/net/context"
etcd "github.com/coreos/etcd/client"
"strings"
)
type EtcdSource struct {
Endpoints []string `json:"endpoints"`
Prefix string `json:"prefix"`
KeepPrefix bool `json:"keepPrefix"`
}
func (etcdSource *EtcdSource) Get() (map[string]interface{}, error) {
cfg := etcd.Config{
Endpoints: etcdSource.Endpoints,
Transport: etcd.DefaultTransport,
}
client, err := etcd.New(cfg)
if err != nil {
return nil, err
}
keysAPI := etcd.NewKeysAPI(client)
response, err := keysAPI.Get(context.Background(), etcdSource.Prefix, &etcd.GetOptions{
Recursive: true,
})
if err != nil {
return nil, err
}
result := make(map[string]interface{})
etcdSource.nodeToMap(response.Node, result)
return result, nil
}
func (etcdSource *EtcdSource) nodeToMap(node *etcd.Node, result map[string]interface{}) {
key := node.Key
if !node.Dir {
key = strings.TrimPrefix(key, "/")
if !etcdSource.KeepPrefix {
key = strings.TrimPrefix(key, etcdSource.Prefix)
}
key = strings.Replace(key, "/", "_", -1)
result[key] = node.Value
} else {
for _, subNode := range node.Nodes {
etcdSource.nodeToMap(subNode, result)
}
}
}