Repository navigation
Expand file tree
/
Copy pathconfig.go
More file actions
210 lines (199 loc) · 6.25 KB
/
Copy pathconfig.go
File metadata and controls
210 lines (199 loc) · 6.25 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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
// Copyright (c) 2013 Kelsey Hightower. All rights reserved.
// Use of this source code is governed by the Apache License, Version 2.0
// that can be found in the LICENSE file.
package main
import (
"errors"
"flag"
"io/ioutil"
"net"
"os"
"path/filepath"
"strconv"
"strings"
"github.com/BurntSushi/toml"
"github.com/kelseyhightower/confd/backends"
"github.com/kelseyhightower/confd/log"
"github.com/kelseyhightower/confd/resource/template"
)
var (
configFile = ""
defaultConfigFile = "/etc/confd/confd.toml"
backend string
clientCaKeys string
clientCert string
clientKey string
confdir string
config Config // holds the global confd config.
debug bool
interval int
keepStageFile bool
nodes Nodes
noop bool
onetime bool
prefix string
printVersion bool
quiet bool
scheme string
srvDomain string
templateConfig template.Config
backendsConfig backends.Config
verbose bool
)
// A Config structure is used to configure confd.
type Config struct {
Backend string `toml:"backend"`
BackendNodes []string `toml:"nodes"`
ClientCaKeys string `toml:"client_cakeys"`
ClientCert string `toml:"client_cert"`
ClientKey string `toml:"client_key"`
ConfDir string `toml:"confdir"`
Debug bool `toml:"debug"`
Interval int `toml:"interval"`
Noop bool `toml:"noop"`
Prefix string `toml:"prefix"`
Quiet bool `toml:"quiet"`
SRVDomain string `toml:"srv_domain"`
Scheme string `toml:"scheme"`
Verbose bool `toml:"verbose"`
}
func init() {
flag.StringVar(&backend, "backend", "etcd", "backend to use")
flag.StringVar(&clientCaKeys, "client-ca-keys", "", "client ca keys")
flag.StringVar(&clientCert, "client-cert", "", "the client cert")
flag.StringVar(&clientKey, "client-key", "", "the client key")
flag.StringVar(&confdir, "confdir", "/etc/confd", "confd conf directory")
flag.StringVar(&configFile, "config-file", "", "the confd config file")
flag.BoolVar(&debug, "debug", false, "enable debug logging")
flag.IntVar(&interval, "interval", 600, "backend polling interval")
flag.BoolVar(&keepStageFile, "keep-stage-file", false, "keep staged files")
flag.Var(&nodes, "node", "list of backend nodes")
flag.BoolVar(&noop, "noop", false, "only show pending changes")
flag.BoolVar(&onetime, "onetime", false, "run once and exit")
flag.StringVar(&prefix, "prefix", "/", "key path prefix")
flag.BoolVar(&printVersion, "version", false, "print version and exit")
flag.BoolVar(&quiet, "quiet", false, "enable quiet logging")
flag.StringVar(&scheme, "scheme", "http", "the backend URI scheme (http or https)")
flag.StringVar(&srvDomain, "srv-domain", "", "the name of the resource record")
flag.BoolVar(&verbose, "verbose", false, "enable verbose logging")
}
// initConfig initializes the confd configuration by first setting defaults,
// then overriding setting from the confd config file, and finally overriding
// settings from flags set on the command line.
// It returns an error if any.
func initConfig() error {
if configFile == "" {
if _, err := os.Stat(defaultConfigFile); !os.IsNotExist(err) {
configFile = defaultConfigFile
}
}
// Set defaults.
config = Config{
Backend: "etcd",
BackendNodes: []string{"127.0.0.1:4001"},
ConfDir: "/etc/confd",
Interval: 600,
Prefix: "/",
Scheme: "http",
}
// Update config from the TOML configuration file.
if configFile == "" {
log.Warning("Skipping confd config file.")
} else {
log.Debug("Loading " + configFile)
configBytes, err := ioutil.ReadFile(configFile)
if err != nil {
return err
}
_, err = toml.Decode(string(configBytes), &config)
if err != nil {
return err
}
}
// Update config from commandline flags.
processFlags()
// Configure logging.
log.SetQuiet(config.Quiet)
log.SetVerbose(config.Verbose)
log.SetDebug(config.Debug)
// Update BackendNodes from SRV records.
if config.Backend != "env" && config.SRVDomain != "" {
log.Info("SRV domain set to " + config.SRVDomain)
srvNodes, err := getBackendNodesFromSRV(config.Backend, config.SRVDomain)
if err != nil {
return errors.New("Cannot get nodes from SRV records " + err.Error())
}
config.BackendNodes = srvNodes
}
// Initialize the storage client
log.Notice("Backend set to " + config.Backend)
backendsConfig = backends.Config{
Backend: config.Backend,
ClientCaKeys: config.ClientCaKeys,
ClientCert: config.ClientCert,
ClientKey: config.ClientKey,
BackendNodes: config.BackendNodes,
Scheme: config.Scheme,
}
// Template configuration.
templateConfig = template.Config{
ConfDir: config.ConfDir,
ConfigDir: filepath.Join(config.ConfDir, "conf.d"),
KeepStageFile: keepStageFile,
Noop: config.Noop,
Prefix: config.Prefix,
TemplateDir: filepath.Join(config.ConfDir, "templates"),
}
return nil
}
func getBackendNodesFromSRV(backend, domain string) ([]string, error) {
nodes := make([]string, 0)
// Ignore the CNAME as we don't need it.
_, addrs, err := net.LookupSRV(backend, "tcp", domain)
if err != nil {
return nodes, err
}
for _, srv := range addrs {
host := strings.TrimRight(srv.Target, ".")
port := strconv.FormatUint(uint64(srv.Port), 10)
nodes = append(nodes, net.JoinHostPort(host, port))
}
return nodes, nil
}
// processFlags iterates through each flag set on the command line and
// overrides corresponding configuration settings.
func processFlags() {
flag.Visit(setConfigFromFlag)
}
func setConfigFromFlag(f *flag.Flag) {
switch f.Name {
case "backend":
config.Backend = backend
case "debug":
config.Debug = debug
case "client-cert":
config.ClientCert = clientCert
case "client-key":
config.ClientKey = clientKey
case "client-cakeys":
config.ClientCaKeys = clientCaKeys
case "confdir":
config.ConfDir = confdir
case "node":
config.BackendNodes = nodes
case "interval":
config.Interval = interval
case "noop":
config.Noop = noop
case "prefix":
config.Prefix = prefix
case "quiet":
config.Quiet = quiet
case "scheme":
config.Scheme = scheme
case "srv-domain":
config.SRVDomain = srvDomain
case "verbose":
config.Verbose = verbose
}
}