fix bookmark

0xPolygonHermez · Apr 30, 2024 · 90f8258 · 90f8258
1 parent 575ba7a
commit 90f8258
Show file tree

Hide file tree

Showing 5 changed files with 12 additions and 658 deletions.
diff --git a/sequencesender/README.md → docs/README.md b/sequencesender/README.md → docs/README.md
diff --git a/sequencesender/SequenceSender.png → docs/SequenceSender.png b/sequencesender/SequenceSender.png → docs/SequenceSender.png
diff --git a/sequencesender/sequencesender.go b/sequencesender/sequencesender.go
@@ -2,7 +2,6 @@ package sequencesender
 
 import (
 	"context"
-	"encoding/binary"
 	"encoding/json"
 	"errors"
 	"fmt"
@@ -159,8 +158,17 @@ func (s *SequenceSender) Start(ctx context.Context) {
 
 	// Set starting point of the streaming
 	s.fromStreamBatch = s.latestVirtualBatch
-	bookmark := []byte{state.BookMarkTypeBatch}
-	bookmark = binary.BigEndian.AppendUint64(bookmark, s.fromStreamBatch)
+
+	bookmark := &datastream.BookMark{
+		Type:  datastream.BookmarkType_BOOKMARK_TYPE_BATCH,
+		Value: s.fromStreamBatch,
+	}
+
+	marshalledBookmark, err := proto.Marshal(bookmark)
+	if err != nil {
+		log.Fatalf("[SeqSender] failed to marshal bookmark, error: %v", err)
+	}
+
 	log.Infof("[SeqSender] stream client from bookmark %v", bookmark)
 
 	// Current batch to sequence
@@ -171,7 +179,7 @@ func (s *SequenceSender) Start(ctx context.Context) {
 	go s.sequenceSending(ctx)
 
 	// Start receiving the streaming
-	err = s.streamClient.ExecCommandStartBookmark(bookmark)
+	err = s.streamClient.ExecCommandStartBookmark(marshalledBookmark)
 	if err != nil {
 		log.Fatalf("[SeqSender] failed to connect to the streaming")
 	}