39 lines
606 B
Go
39 lines
606 B
Go
|
package persistent
|
||
|
|
|
||
|
|
import (
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"go.uber.org/zap"
|
||
|
|
)
|
||
|
|
|
||
|
|
type cfg struct {
|
||
|
|
l *zap.Logger
|
||
|
|
timeout time.Duration
|
||
|
|
}
|
||
|
|
|
||
|
|
// Option allows setting optional parameters of the TokenStore.
|
||
|
|
type Option func(*cfg)
|
||
|
|
|
||
|
|
func defaultCfg() *cfg {
|
||
|
|
return &cfg{
|
||
|
|
l: zap.L(),
|
||
|
|
timeout: 100 * time.Millisecond,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// WithLogger returns an option to specify
|
||
|
|
// logger.
|
||
|
|
func WithLogger(v *zap.Logger) Option {
|
||
|
|
return func(c *cfg) {
|
||
|
|
c.l = v
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// WithTimeout returns option to specify
|
||
|
|
// database connection timeout.
|
||
|
|
func WithTimeout(v time.Duration) Option {
|
||
|
|
return func(c *cfg) {
|
||
|
|
c.timeout = v
|
||
|
|
}
|
||
|
|
}
|