fix: skip list put operation as set when key exists

zongxingye · Jan 15, 2022 · 5f5cb00 · 5f5cb00
1 parent 9e0b271
commit 5f5cb00
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/arenaskl/iterator.go b/arenaskl/iterator.go
@@ -113,6 +113,9 @@ func (it *Iterator) SeekForPrev(key []byte) (found bool) {
 // on the most current value and returns ErrRecordExists. If there isn't enough
 // room in the arena, then Add returns ErrArenaFull.
 func (it *Iterator) Put(key []byte, val []byte) error {
+	if found := it.Seek(key); found {
+		return it.Set(val)
+	}
 	var spl [maxHeight]splice
 	if it.seekForSplice(key, &spl) {
 		// Found a matching node, but handle case where it's been deleted.