feat(store): add readTail method

celestiaorg · Aug 6, 2024 · d2c6088 · d2c6088
1 parent 7d9aad6
commit d2c6088
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 1 deletion.
diff --git a/store/keys.go b/store/keys.go
@@ -8,7 +8,10 @@ import (
 	"github.com/celestiaorg/go-header"
 )
 
-var headKey = datastore.NewKey("head")
+var (
+	headKey = datastore.NewKey("head")
+	tailKey = datastore.NewKey("tail")
+)
 
 func heightKey(h uint64) datastore.Key {
 	return datastore.NewKey(strconv.Itoa(int(h)))

diff --git a/store/store.go b/store/store.go
@@ -482,6 +482,22 @@ func (s *Store[H]) readHead(ctx context.Context) (H, error) {
 	return s.Get(ctx, head)
 }
 
+// readTail loads the tail from the datastore.
+func (s *Store[H]) readTail(ctx context.Context) (H, error) {
+	var zero H
+	b, err := s.ds.Get(ctx, tailKey)
+	if err != nil {
+		return zero, err
+	}
+
+	var tail header.Hash
+	if err := tail.UnmarshalJSON(b); err != nil {
+		return zero, err
+	}
+
+	return s.Get(ctx, tail)
+}
+
 func (s *Store[H]) get(ctx context.Context, hash header.Hash) ([]byte, error) {
 	startTime := time.Now()
 	data, err := s.ds.Get(ctx, datastore.NewKey(hash.String()))