Merge pull request #1 from NebulousLabs/holes

Patch holes
NebulousLabs · Oct 28, 2015 · 8233989 · 8233989
2 parents 8488501 + ddef166
commit 8233989
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/bucket.go b/bucket.go
@@ -325,9 +325,6 @@ func (b *Bucket) Delete(key []byte) error {
 	// Delete the node if we have a matching key.
 	c.node().del(key)
 
-	// Rebalance the bucket.
-	b.rebalance()
-
 	return nil
 }
 

diff --git a/cursor.go b/cursor.go
@@ -38,6 +38,9 @@ func (c *Cursor) First() (key []byte, value []byte) {
 	if (flags & uint32(bucketLeafFlag)) != 0 {
 		return k, nil
 	}
+	if k == nil && v == nil && flags == 0 {
+		k, v, flags = c.next()
+	}
 	return k, v
 
 }
@@ -230,6 +233,11 @@ func (c *Cursor) next() (key []byte, value []byte, flags uint32) {
 	// first element of the first leaf page.
 	c.stack = c.stack[:i+1]
 	c.first()
+
+	if c.stack[len(c.stack)-1].count() == 0 {
+		return c.next()
+	}
+
 	return c.keyValue()
 }