mirror of
https://github.com/k3s-io/kubernetes.git
synced 2025-09-07 04:03:20 +00:00
etcd3 watcher: use prevKV for prevObj
This commit is contained in:
@@ -24,15 +24,17 @@ import (
|
|||||||
type event struct {
|
type event struct {
|
||||||
key string
|
key string
|
||||||
value []byte
|
value []byte
|
||||||
|
prevValue []byte
|
||||||
rev int64
|
rev int64
|
||||||
isDeleted bool
|
isDeleted bool
|
||||||
isCreated bool
|
isCreated bool
|
||||||
}
|
}
|
||||||
|
|
||||||
func parseKV(kv *mvccpb.KeyValue) *event {
|
func parseKV(kv *mvccpb.KeyValue, prevVal []byte) *event {
|
||||||
return &event{
|
return &event{
|
||||||
key: string(kv.Key),
|
key: string(kv.Key),
|
||||||
value: kv.Value,
|
value: kv.Value,
|
||||||
|
prevValue: prevVal,
|
||||||
rev: kv.ModRevision,
|
rev: kv.ModRevision,
|
||||||
isDeleted: false,
|
isDeleted: false,
|
||||||
isCreated: kv.ModRevision == kv.CreateRevision,
|
isCreated: kv.ModRevision == kv.CreateRevision,
|
||||||
@@ -40,11 +42,15 @@ func parseKV(kv *mvccpb.KeyValue) *event {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func parseEvent(e *clientv3.Event) *event {
|
func parseEvent(e *clientv3.Event) *event {
|
||||||
return &event{
|
ret := &event{
|
||||||
key: string(e.Kv.Key),
|
key: string(e.Kv.Key),
|
||||||
value: e.Kv.Value,
|
value: e.Kv.Value,
|
||||||
rev: e.Kv.ModRevision,
|
rev: e.Kv.ModRevision,
|
||||||
isDeleted: e.Type == clientv3.EventTypeDelete,
|
isDeleted: e.Type == clientv3.EventTypeDelete,
|
||||||
isCreated: e.IsCreate(),
|
isCreated: e.IsCreate(),
|
||||||
}
|
}
|
||||||
|
if e.PrevKv != nil {
|
||||||
|
ret.prevValue = e.PrevKv.Value
|
||||||
|
}
|
||||||
|
return ret
|
||||||
}
|
}
|
||||||
|
@@ -154,7 +154,15 @@ func (wc *watchChan) sync() error {
|
|||||||
wc.initialRev = getResp.Header.Revision
|
wc.initialRev = getResp.Header.Revision
|
||||||
|
|
||||||
for _, kv := range getResp.Kvs {
|
for _, kv := range getResp.Kvs {
|
||||||
wc.sendEvent(parseKV(kv))
|
prevResp, err := wc.watcher.client.Get(wc.ctx, string(kv.Key), clientv3.WithRev(kv.ModRevision-1))
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
var prevVal []byte
|
||||||
|
if len(prevResp.Kvs) > 0 {
|
||||||
|
prevVal = prevResp.Kvs[0].Value
|
||||||
|
}
|
||||||
|
wc.sendEvent(parseKV(kv, prevVal))
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
@@ -170,7 +178,7 @@ func (wc *watchChan) startWatching(watchClosedCh chan struct{}) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
opts := []clientv3.OpOption{clientv3.WithRev(wc.initialRev + 1)}
|
opts := []clientv3.OpOption{clientv3.WithRev(wc.initialRev + 1), clientv3.WithPrevKV()}
|
||||||
if wc.recursive {
|
if wc.recursive {
|
||||||
opts = append(opts, clientv3.WithPrefix())
|
opts = append(opts, clientv3.WithPrefix())
|
||||||
}
|
}
|
||||||
@@ -331,16 +339,12 @@ func prepareObjs(ctx context.Context, e *event, client *clientv3.Client, codec r
|
|||||||
return nil, nil, err
|
return nil, nil, err
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if e.isDeleted || !e.isCreated {
|
if len(e.prevValue) > 0 {
|
||||||
getResp, err := client.Get(ctx, e.key, clientv3.WithRev(e.rev-1), clientv3.WithSerializable())
|
|
||||||
if err != nil {
|
|
||||||
return nil, nil, err
|
|
||||||
}
|
|
||||||
// Note that this sends the *old* object with the etcd revision for the time at
|
// Note that this sends the *old* object with the etcd revision for the time at
|
||||||
// which it gets deleted.
|
// which it gets deleted.
|
||||||
// We assume old object is returned only in Deleted event. Users (e.g. cacher) need
|
// We assume old object is returned only in Deleted event. Users (e.g. cacher) need
|
||||||
// to have larger than previous rev to tell the ordering.
|
// to have larger than previous rev to tell the ordering.
|
||||||
oldObj, err = decodeObj(codec, versioner, getResp.Kvs[0].Value, e.rev)
|
oldObj, err = decodeObj(codec, versioner, e.prevValue, e.rev)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, nil, err
|
return nil, nil, err
|
||||||
}
|
}
|
||||||
|
Reference in New Issue
Block a user