fix: missing first elem in graphql tx iterator (#33)

Thanks!
gnolang · Apr 15, 2024 · f026544 · f026544
1 parent d48e8b1
commit f026544
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 8 deletions.
diff --git a/storage/pebble.go b/storage/pebble.go
@@ -261,17 +261,12 @@ type PebbleTxIter struct {
 func (pi *PebbleTxIter) Next() bool {
 	for {
 		if !pi.init {
-			pi.init = true
 			if !pi.i.First() {
 				return false
 			}
-		}
-
-		if !pi.i.Valid() {
-			return false
-		}
 
-		if !pi.i.Next() {
+			pi.init = true
+		} else if !pi.i.Next() {
 			return false
 		}
 

diff --git a/storage/pebble_test.go b/storage/pebble_test.go
@@ -145,7 +145,7 @@ func TestStorageIters(t *testing.T) {
 		txCount++
 	}
 
-	require.Equal(t, 2, txCount)
+	require.Equal(t, 3, txCount)
 
 	defer require.NoError(t, it.Close())