Issue #182: Initialize Vault client better

Use the DefaultConfig and the environment variable overrides of the Vault API when creating the client.
fabiolb · Nov 4, 2016 · 1b5df2c · 1b5df2c
1 parent 0a88383
commit 1b5df2c
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/cert/vault_source.go b/cert/vault_source.go
@@ -34,7 +34,14 @@ type VaultSource struct {
 }
 
 func (s *VaultSource) client() (*api.Client, error) {
-	c, err := api.NewClient(&api.Config{Address: s.Addr})
+	conf := api.DefaultConfig()
+	if err := conf.ReadEnvironment(); err != nil {
+		return nil, err
+	}
+	if s.Addr != "" {
+		conf.Address = s.Addr
+	}
+	c, err := api.NewClient(conf)
 	if err != nil {
 		return nil, err
 	}