provide timeout option and lower default
Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
This commit is contained in:
parent
68696aebd5
commit
16cde1bfae
@ -1,6 +1,8 @@
|
|||||||
package vault
|
package vault
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"time"
|
||||||
|
|
||||||
"github.com/hashicorp/vault/api"
|
"github.com/hashicorp/vault/api"
|
||||||
"github.com/unistack-org/micro/v3/config"
|
"github.com/unistack-org/micro/v3/config"
|
||||||
)
|
)
|
||||||
@ -40,3 +42,9 @@ type secretIDKey struct{}
|
|||||||
func SecretID(secret string) config.Option {
|
func SecretID(secret string) config.Option {
|
||||||
return config.SetOption(secretIDKey{}, secret)
|
return config.SetOption(secretIDKey{}, secret)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type timeoutKey struct{}
|
||||||
|
|
||||||
|
func Timeout(td time.Duration) config.Option {
|
||||||
|
return config.SetOption(timeoutKey{}, td)
|
||||||
|
}
|
||||||
|
6
vault.go
6
vault.go
@ -4,6 +4,7 @@ import (
|
|||||||
"context"
|
"context"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"errors"
|
"errors"
|
||||||
|
"time"
|
||||||
|
|
||||||
"github.com/hashicorp/vault/api"
|
"github.com/hashicorp/vault/api"
|
||||||
"github.com/imdario/mergo"
|
"github.com/imdario/mergo"
|
||||||
@ -36,6 +37,7 @@ func (c *vaultConfig) Init(opts ...config.Option) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
cfg := api.DefaultConfig()
|
cfg := api.DefaultConfig()
|
||||||
|
cfg.Timeout = 3 * time.Second
|
||||||
path := ""
|
path := ""
|
||||||
token := ""
|
token := ""
|
||||||
roleID := ""
|
roleID := ""
|
||||||
@ -46,6 +48,10 @@ func (c *vaultConfig) Init(opts ...config.Option) error {
|
|||||||
cfg = v
|
cfg = v
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if v, ok := c.opts.Context.Value(timeoutKey{}).(time.Duration); ok {
|
||||||
|
cfg.Timeout = v
|
||||||
|
}
|
||||||
|
|
||||||
if v, ok := c.opts.Context.Value(addrKey{}).(string); ok {
|
if v, ok := c.opts.Context.Value(addrKey{}).(string); ok {
|
||||||
cfg.Address = v
|
cfg.Address = v
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user