增加watch方法

This commit is contained in:
白茶清欢 2021-11-23 15:13:50 +08:00
parent 0898107e92
commit c3f8e40133
2 changed files with 35 additions and 0 deletions

View File

@ -10,6 +10,8 @@ package etcd
import (
"math"
"time"
"go.etcd.io/etcd/clientv3"
)
// 初始化client各种默认配置
@ -33,3 +35,6 @@ const (
// DefaultGetTimeout get 默认超时时间
DefaultGetTimeout = time.Second
)
// WatcherHandler 监听key变化的处理函数
type WatcherHandler func(event *clientv3.Event)

30
middleware/etcd/watch.go Normal file
View File

@ -0,0 +1,30 @@
// Package etcd...
//
// Description : 监听key的变化
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 2021-11-23 2:58 下午
package etcd
import (
"context"
)
// WatchKey 监听key的变化
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 2:58 下午 2021/11/23
func WatchKey(ctx context.Context, watchKey string, callbackFunc WatcherHandler) {
if nil == callbackFunc {
// 变化之后,没有任何逻辑处理,视为不需要监听变化
return
}
rch := Client.Watch(context.Background(), watchKey) // <-chan WatchResponse
for watchResp := range rch {
for _, ev := range watchResp.Events {
callbackFunc(ev)
}
}
}