コード例 #1
0
ファイル: consul.go プロジェクト: joyent/containerpilot
// NewConsulConfig creates a new service discovery backend for Consul
func NewConsulConfig(config interface{}) (*Consul, error) {
	var consulConfig *consul.Config
	var err error
	switch t := config.(type) {
	case string:
		consulConfig, err = configFromURI(t)
	case map[string]interface{}:
		consulConfig, err = configFromMap(t)
	default:
		return nil, fmt.Errorf("Unexpected Consul config structure. Expected a string or map")
	}
	if err != nil {
		return nil, err
	}

	if token := os.Getenv("CONSUL_HTTP_TOKEN"); token != "" {
		consulConfig.Token = token
	}
	client, err := consul.NewClient(consulConfig)
	if err != nil {
		return nil, err
	}
	return &Consul{*client}, nil
}
コード例 #2
0
ファイル: consul.go プロジェクト: myENA/consul-backinator
// New returns an initialized consul client
func (cc *Config) New() (*Client, error) {
	var c *api.Config  // upstream client configuration
	var client *Client // client wrapper
	var err error      // general error holder

	// init upstream config
	c = api.DefaultConfig()

	// overwrite address if needed
	if cc.Address != "" {
		c.Address = cc.Address
	}

	// overwrite scheme if needed
	if cc.Scheme != "" {
		c.Scheme = cc.Scheme
	}

	// overwrite dc if needed
	if cc.Datacenter != "" {
		c.Datacenter = cc.Datacenter
	}

	// overwrite token if needed
	if cc.Token != "" {
		c.Token = cc.Token
	}

	// configure if any TLS specific options were passed
	if cc.TLS.CAFile != "" || cc.TLS.CertFile != "" || cc.TLS.KeyFile != "" || cc.TLS.InsecureSkipVerify {
		var tlsConfig *tls.Config // client TLS config
		// attempt to build tls config from passed options
		if tlsConfig, err = api.SetupTLSConfig(cc.TLS); err != nil {
			return nil, err
		}
		// build a new http client and transport
		httpClient := cleanhttp.DefaultClient()
		httpTransport := cleanhttp.DefaultTransport()
		httpTransport.TLSClientConfig = tlsConfig
		httpClient.Transport = httpTransport

		// set client
		c.HttpClient = httpClient
	}

	// init client wrapper
	client = new(Client)
	client.Client, err = api.NewClient(c)

	// return client and error
	return client, err
}