38 lines
801 B
Go
38 lines
801 B
Go
|
package etcd
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
"github.com/unistack-org/micro/v3/registry"
|
||
|
|
"go.uber.org/zap"
|
||
|
|
)
|
||
|
|
|
||
|
|
type authKey struct{}
|
||
|
|
|
||
|
|
type logConfigKey struct{}
|
||
|
|
|
||
|
|
type authCreds struct {
|
||
|
|
Username string
|
||
|
|
Password string
|
||
|
|
}
|
||
|
|
|
||
|
|
// Auth allows you to specify username/password
|
||
|
|
func Auth(username, password string) registry.Option {
|
||
|
|
return func(o *registry.Options) {
|
||
|
|
if o.Context == nil {
|
||
|
|
o.Context = context.Background()
|
||
|
|
}
|
||
|
|
o.Context = context.WithValue(o.Context, authKey{}, &authCreds{Username: username, Password: password})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// LogConfig allows you to set etcd log config
|
||
|
|
func LogConfig(config *zap.Config) registry.Option {
|
||
|
|
return func(o *registry.Options) {
|
||
|
|
if o.Context == nil {
|
||
|
|
o.Context = context.Background()
|
||
|
|
}
|
||
|
|
o.Context = context.WithValue(o.Context, logConfigKey{}, config)
|
||
|
|
}
|
||
|
|
}
|