Repository navigation
Expand file tree
/
Copy pathclient.go
More file actions
63 lines (58 loc) · 1.61 KB
/
Copy pathclient.go
File metadata and controls
63 lines (58 loc) · 1.61 KB
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
61
62
63
package backends
import (
"errors"
"net/url"
"strings"
"github.com/kelseyhightower/confd/backends/consul"
"github.com/kelseyhightower/confd/backends/env"
"github.com/kelseyhightower/confd/backends/etcd"
"github.com/kelseyhightower/confd/log"
)
// The StoreClient interface is implemented by objects that can retrieve
// key/value pairs from a backend store.
type StoreClient interface {
GetValues(keys []string) (map[string]string, error)
}
// New is used to create a storage client based on our configuration.
func New(config Config) (StoreClient, error) {
if config.Backend == "" {
config.Backend = "etcd"
}
var err error
backendNodes := config.BackendNodes
if config.Backend == "etcd" {
backendNodes, err = addScheme(config.Scheme, config.BackendNodes)
if err != nil {
return nil, err
}
}
log.Notice("Backend nodes set to " + strings.Join(backendNodes, ", "))
switch config.Backend {
case "consul":
return consul.NewConsulClient(backendNodes)
case "etcd":
// Create the etcd client upfront and use it for the life of the process.
// The etcdClient is an http.Client and designed to be reused.
return etcd.NewEtcdClient(backendNodes, config.ClientCert, config.ClientKey, config.ClientCaKeys)
case "env":
return env.NewEnvClient()
}
return nil, errors.New("Invalid backend")
}
func addScheme(scheme string, nodes []string) ([]string, error) {
ns := make([]string, 0)
if scheme == "" {
scheme = "http"
}
for _, node := range nodes {
u, err := url.Parse(node)
if err != nil {
return nil, err
}
if u.Scheme == "" {
u.Scheme = scheme
}
ns = append(ns, u.String())
}
return ns, nil
}