src/dedo.go: Add Bucket.NextID() that returns []bytes instead of uint64

1 files changed, 25 insertions, 0 deletions

diff --git a/src/dedo.go b/src/dedo.go
index e69dbd4..92317da 100644
--- a/src/dedo.go
+++ b/src/dedo.go
@@ -40,6 +40,7 @@ package dedo
 
 import (
 	"bytes"
+	"encoding/binary"
 	"errors"
 	"flag"
 	"fmt"
@@ -828,6 +829,30 @@ func (b *Bucket) Delete(key []byte) error {
 	return nil
 }
 
+func bytesLE(num uint64) []byte {
+	arr := make([]byte, 8)
+	binary.LittleEndian.PutUint64(arr, uint64(num))
+	return arr
+}
+
+func (b *Bucket) NextID() []byte {
+	if b.tx.db == nil {
+		panic(ErrTxClosed)
+	} else if !b.tx.writable {
+		panic(ErrTxNotWritable)
+	}
+
+	// Materialize the root node if it hasn't been already so that the
+	// bucket will be saved during commit.
+	if b.rootNode == nil {
+		_ = b.node(b.ref.root, nil)
+	}
+
+	// Increment and return the sequence.
+	b.ref.sequence++
+	return bytesLE(b.ref.sequence)
+}
+
 /// Bucket.NextSequence() returns an autoincrementing integer for the bucket.
 func (b *Bucket) NextSequence() (uint64, error) {
 	if b.tx.db == nil {