feat: add AsyncPruner

Author: Eoous

block/async_store.go

@@ -0,0 +1,45 @@
+package block
+
+import (
+	"context"
+	"time"
+
+	"github.com/rollkit/rollkit/pkg/store"
+)
+
+const DefaultFlushInterval = 1 * time.Second
+
+// AsyncPruner is a service that periodically prunes block data in the background.
+type AsyncPruner struct {
+	ps store.PruningStore
+
+	flushInterval time.Duration
+}
+
+func NewAsyncPruner(pruningStore store.PruningStore, flushInterval time.Duration) *AsyncPruner {
+	if flushInterval <= 0 {
+		flushInterval = DefaultFlushInterval
+	}
+
+	return &AsyncPruner{
+		ps: pruningStore,
+
+		flushInterval: flushInterval,
+	}
+}
+
+// Start starts the async pruner that periodically prunes block data.
+func (s *AsyncPruner) Start(ctx context.Context) {
+	ticker := time.NewTicker(s.flushInterval)
+	defer ticker.Stop()
+
+	for {
+		select {
+		case <-ctx.Done():
+			return
+		case <-ticker.C:
+			// Currently PruneBlockData only returns nil.
+			_ = s.ps.PruneBlockData(ctx)
+		}
+	}
+}

node/full.go

@@ -61,6 +61,7 @@ type FullNode struct {
 	Store        store.Store
 	blockManager *block.Manager
 	reaper       *block.Reaper
+	asyncPruner  *block.AsyncPruner
 
 	prometheusSrv *http.Server
 	pprofSrv      *http.Server
@@ -129,12 +130,17 @@ func newFullNode(
 	// Connect the reaper to the manager for transaction notifications
 	reaper.SetManager(blockManager)
 
+	asyncPruner := block.NewAsyncPruner(
+		rktStore, block.DefaultFlushInterval,
+	)
+
 	node := &FullNode{
 		genesis:      genesis,
 		nodeConfig:   nodeConfig,
 		p2pClient:    p2pClient,
 		blockManager: blockManager,
 		reaper:       reaper,
+		asyncPruner:  asyncPruner,
 		da:           da,
 		Store:        rktStore,
 		hSyncService: headerSyncService,
@@ -390,6 +396,7 @@ func (n *FullNode) Run(parentCtx context.Context) error {
 		spawnWorker(func() { n.blockManager.SyncLoop(ctx, errCh) })
 		spawnWorker(func() { n.blockManager.DAIncluderLoop(ctx, errCh) })
 	}
+	spawnWorker(func() { n.asyncPruner.Start(ctx) })
 
 	select {
 	case err := <-errCh: