golang 操作etcd租约以及监听kv变化

定义错误常量:

const (
    NewLeaseErr  = 101
    LeasTtlErr   = 102
    KeepAliveErr = 103
    PutErr       = 104
    GetErr       = 105
    RevokeErr    = 106
)

创建client:

var conf = clientv3.Config{
        Endpoints:   []string{"172.16.196.129:2380", "192.168.50.250:2380"},
        DialTimeout: 5 * time.Second,
}
client, err := clientv3.New(conf)
defer client.Close()
if err != nil {
    fmt.Printf("创建client失败:\n", err.Error())
    os.Exit(NewLeaseErr)
}

创建租约:

//创建租约
lease := clientv3.NewLease(client)

//设置租约时间
leaseResp, err := lease.Grant(context.TODO(), 10)
if err != nil {
    fmt.Printf("设置租约时间失败:%s\n", err.Error())
    os.Exit(LeasTtlErr)
}

设置续租:

//设置续租
leaseID := leaseResp.ID
ctx, cancelFunc := context.WithCancel(context.TODO())
leaseRespChan, err := lease.KeepAlive(ctx, leaseID)
if err != nil {
    fmt.Printf("续租失败:%s\n", err.Error())
    os.Exit(KeepAliveErr)
}

监听租约:

go func() {
        for  {
            select {
            case leaseKeepResp := <-leaseRespChan:
                if leaseKeepResp == nil {
                    fmt.Printf("已经关闭续租功能\n")
                    return
                } else {
                    fmt.Printf("续租成功\n")
                    goto END
                }
            }
            END:
                time.Sleep(500*time.Millisecond)
        }
    }()

监听某个key的变化

//ctx1, _ := context.WithTimeout(context.TODO(),20)  //设置超时ctx传入Watch里会使watch监听失败,可能是watch是个永久监听,不支持设置timeCtx(我也不太清楚)
go func() {
    wc := client.Watch(context.TODO(), "/job/v3/1", clientv3.WithPrevKV())
    for v := range wc {
        for _, e := range v.Events {
        fmt.Printf("type:%v kv:%v  prevKey:%v \n ", e.Type, string(e.Kv.Key), e.PrevKv)
    }
    }
}()

put操作:

kv := clientv3.NewKV(client)
time.Sleep(3*time.Second)
//通过租约put
putResp, err := kv.Put(context.TODO(), "/job/v3/1", "koock",clientv3.WithLease(leaseID))
if err != nil {
    fmt.Printf("put 失败:%s", err.Error())
    os.Exit(PutErr)
}
fmt.Printf("%v\n",putResp.Header)

取消续租以及撤销租约:

撤销租约会使当前租约的所关联的key-value失效

 //关闭续租
cancelFunc() 
//撤销租约
_, err = lease.Revoke(context.TODO(), leaseID)
    if err != nil {
        fmt.Printf("撤销租约失败:%s\n",err.Error())
        os.Exit(RevokeErr)
    }
fmt.Printf("撤销租约成功")

完整代码段:

package main

import (
    "context"
    "fmt"
    "go.etcd.io/etcd/clientv3"
    "os"
    "time"
)

const (
    NewLeaseErr  = 101
    LeasTtlErr   = 102
    KeepAliveErr = 103
    PutErr       = 104
    GetErr       = 105
    RevokeErr    = 106
)

func main() {

    var conf = clientv3.Config{
        Endpoints:   []string{"172.16.196.129:2380", "192.168.50.250:2380"},
        DialTimeout: 5 * time.Second,
    }

    client, err := clientv3.New(conf)
    defer client.Close()
    if err != nil {
        fmt.Printf("创建client失败:\n", err.Error())
        os.Exit(NewLeaseErr)
    }

    //创建租约
    lease := clientv3.NewLease(client)

    //设置租约时间
    leaseResp, err := lease.Grant(context.TODO(), 10)
    if err != nil {
        fmt.Printf("设置租约时间失败:%s\n", err.Error())
        os.Exit(LeasTtlErr)
    }

    //设置续租
    leaseID := leaseResp.ID
    ctx, cancelFunc := context.WithCancel(context.TODO())
    leaseRespChan, err := lease.KeepAlive(ctx, leaseID)
    if err != nil {
        fmt.Printf("续租失败:%s\n", err.Error())
        os.Exit(KeepAliveErr)
    }

    //监听租约
    go func() {
        for  {
            select {
            case leaseKeepResp := <-leaseRespChan:
                if leaseKeepResp == nil {
                    fmt.Printf("已经关闭续租功能\n")
                    return
                } else {
                    fmt.Printf("续租成功\n")
                    goto END
                }
            }
            END:
                time.Sleep(500*time.Millisecond)
        }

    }()

    //监听某个key的变化
    //ctx1, _ := context.WithTimeout(context.TODO(),20)
    go func() {
        wc := client.Watch(context.TODO(), "/job/v3/1", clientv3.WithPrevKV())
            for v := range wc {
                for _, e := range v.Events {
                    fmt.Printf("type:%v kv:%v  prevKey:%v \n ", e.Type, string(e.Kv.Key), e.PrevKv)
                }
            }
    }()

    kv := clientv3.NewKV(client)
    //通过租约put
    putResp, err := kv.Put(context.TODO(), "/job/v3/1", "koock",clientv3.WithLease(leaseID))
    if err != nil {
        fmt.Printf("put 失败:%s", err.Error())
        os.Exit(PutErr)
    }
    fmt.Printf("%v\n",putResp.Header)

    cancelFunc()

    time.Sleep(2*time.Second)
    _, err = lease.Revoke(context.TODO(), leaseID)
    if err != nil {
        fmt.Printf("撤销租约失败:%s\n",err.Error())
        os.Exit(RevokeErr)
    }
    fmt.Printf("撤销租约成功")
    getResp,err := kv.Get(context.TODO(),"/job/v3/1")
    if err != nil{
        fmt.Printf("get 失败:%s",err.Error())
        os.Exit(GetErr)
    }
    fmt.Printf("%v",getResp.Kvs)
    time.Sleep(20 * time.Second)


}

在put操作时,watch会监听到操作,打印出:type:PUT kv:/job/v3/1 prevKey:<nil>
撤销租约后,watch会监听到delete操作,打印出:type:DELETE kv:/job/v3/1 prevKey:key:"/job/v3/1"

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 220,367评论 6 512
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 93,959评论 3 396
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 166,750评论 0 357
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 59,226评论 1 295
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 68,252评论 6 397
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,975评论 1 308
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,592评论 3 420
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 39,497评论 0 276
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 46,027评论 1 319
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 38,147评论 3 340
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 40,274评论 1 352
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,953评论 5 347
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,623评论 3 331
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 32,143评论 0 23
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 33,260评论 1 272
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 48,607评论 3 375
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 45,271评论 2 358

推荐阅读更多精彩内容