修改集群模式使用封装的方法
This commit is contained in:
+20
-1
@@ -3,6 +3,7 @@ package priority
|
||||
import (
|
||||
"time"
|
||||
|
||||
"github.com/google/uuid"
|
||||
"github.com/yuninks/timerx/logger"
|
||||
)
|
||||
|
||||
@@ -10,15 +11,21 @@ type Options struct {
|
||||
getInterval time.Duration // 查询周期
|
||||
updateInterval time.Duration // 更新间隔
|
||||
expireTime time.Duration // 有效时间
|
||||
logger logger.Logger
|
||||
logger logger.Logger // 日志
|
||||
source string // 来源服务
|
||||
instanceId string // 实例ID
|
||||
}
|
||||
|
||||
func defaultOptions() Options {
|
||||
|
||||
u, _ := uuid.NewV7()
|
||||
|
||||
return Options{
|
||||
getInterval: time.Second * 2,
|
||||
updateInterval: time.Second * 4,
|
||||
expireTime: time.Second * 8,
|
||||
logger: logger.NewLogger(),
|
||||
instanceId: u.String(),
|
||||
}
|
||||
}
|
||||
|
||||
@@ -49,3 +56,15 @@ func WithUpdateInterval(d time.Duration) Option {
|
||||
o.getInterval = d / 3
|
||||
}
|
||||
}
|
||||
|
||||
func WithInstanceId(instanceId string) Option {
|
||||
return func(o *Options) {
|
||||
o.instanceId = instanceId
|
||||
}
|
||||
}
|
||||
|
||||
func WithSource(s string) Option {
|
||||
return func(o *Options) {
|
||||
o.source = s
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user