tapdb: add SupplySyncerStore

Add a store used by the syncer to persist synced supply commit leaves,
as well as syncer state (e.g., the latest synced block height for a
given asset group).

Author: ffranr

tapdb/supply_syncer.go

@@ -0,0 +1,84 @@
+package tapdb
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/lightninglabs/taproot-assets/asset"
+	"github.com/lightninglabs/taproot-assets/tapdb/sqlc"
+	"github.com/lightninglabs/taproot-assets/universe/supplycommit"
+)
+
+// SupplySyncerStore implements the persistent storage for supply syncing
+// operations. It provides methods to store supply updates without requiring
+// a supply commitment transition.
+type SupplySyncerStore struct {
+	db BatchedUniverseTree
+}
+
+// NewSupplySyncerStore creates a new supply syncer DB store handle.
+func NewSupplySyncerStore(db BatchedUniverseTree) *SupplySyncerStore {
+	return &SupplySyncerStore{
+		db: db,
+	}
+}
+
+// UpsertSupplyLeaves stores a batch of supply update events to the database
+// without requiring a supply commitment transition. It reuses the internal
+// supply update logic from SupplyTreeStore.
+func (s *SupplySyncerStore) UpsertSupplyLeaves(ctx context.Context,
+	spec asset.Specifier,
+	updates []supplycommit.SupplyUpdateEvent) error {
+
+	// If no updates were provided, return early without error.
+	if len(updates) == 0 {
+		return nil
+	}
+
+	// Find the highest block height from all the supply update events.
+	var maxBlockHeight uint32
+	for _, update := range updates {
+		if height := update.BlockHeight(); height > maxBlockHeight {
+			maxBlockHeight = height
+		}
+	}
+
+	// All updates must have a valid block height.
+	if maxBlockHeight == 0 {
+		return fmt.Errorf("all supply updates must have a valid " +
+			"block height greater than 0")
+	}
+
+	// Extract the group key for logging.
+	groupKey, err := spec.UnwrapGroupKeyOrErr()
+	if err != nil {
+		return fmt.Errorf("group key must be specified for supply "+
+			"syncer: %w", err)
+	}
+
+	var writeTx BaseUniverseStoreOptions
+	return s.db.ExecTx(ctx, &writeTx, func(dbTx BaseUniverseStore) error {
+		// Reuse the internal supply update logic which handles all
+		// the complex sub-tree and root tree updates within the
+		// transaction.
+		_, err := applySupplyUpdatesInternal(ctx, dbTx, spec, updates)
+		if err != nil {
+			return err
+		}
+
+		// Log the latest synced block height for this asset group.
+		groupKeyBytes := groupKey.SerializeCompressed()
+		err = dbTx.UpsertSupplySyncerLog(
+			ctx, sqlc.UpsertSupplySyncerLogParams{
+				GroupKey:              groupKeyBytes,
+				LatestSyncBlockHeight: int32(maxBlockHeight),
+			},
+		)
+		if err != nil {
+			return fmt.Errorf("failed to log synced block "+
+				"height: %w", err)
+		}
+
+		return nil
+	})
+}