scriptoriumadmin/featurebase
1
0
Fork 0
mirror of https://github.com/featurebasedb/featurebase.git synced 2026-09-05 08:10:50 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

force transactional etcd reads

Browse source
This commit is contained in:
Nia Weiss 2021-02-26 18:29:01 -05:00
parent 590dc3b226
commit 4be4097edd
No known key found for this signature in database
GPG key ID: 895E83409BFDA1BB
1 changed files with 9 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
etcd/embed.go
View file

@ -648,16 +648,20 @@ func (e *Etcd) putKey(ctx context.Context, key, val string, opts ...clientv3.OpO
func (e *Etcd) getKeyBytes(ctx context.Context, key string) ([]byte, error) {
// Get the current value for the key.
kv := e.e.Server.KV()
resp, err := kv.Range([]byte(key), nil, mvcc.RangeOptions{})
op := clientv3.OpGet(key)
resp, err := e.cli.Txn(ctx).Then(op).Commit()
if err != nil {
return nil, err
}
kvs := resp.KVs
// TODO: consider returning a "key does not exist" error instead of (nil, nil)
kvs := resp.Responses[0].GetResponseRange().Kvs
if !resp.Succeeded {
return nil, errors.New("tx failed")
}
if len(kvs) == 0 {
return nil, nil
return nil, errors.New("key does not exist")
}
return kvs[0].Value, nil