etcd3 watcher: use prevKV for prevObj

This commit is contained in:
Hongchao Deng
2016-10-05 15:42:32 -07:00
parent c060417ecb
commit 2516ab058d
2 changed files with 20 additions and 10 deletions

View File

@@ -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
} }

View File

@@ -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
} }