tracker: record burn script along with loss details

RaghavSood · Jun 7, 2024 · 6cdc82d · 6cdc82d
1 parent 6823002
commit 6cdc82d
Show file tree

Hide file tree

Showing 5 changed files with 17 additions and 3 deletions.
diff --git a/storage/sqlite/loss.go b/storage/sqlite/loss.go
@@ -52,7 +52,7 @@ func scanLosses(rows *sql.Rows) ([]types.Loss, error) {
 	var losses []types.Loss
 	for rows.Next() {
 		var loss types.Loss
-		err := rows.Scan(&loss.ID, &loss.TxID, &loss.BlockHash, &loss.BlockHeight, &loss.Vout, &loss.Amount, &loss.CreatedAt)
+		err := rows.Scan(&loss.ID, &loss.TxID, &loss.BlockHash, &loss.BlockHeight, &loss.Vout, &loss.Amount, &loss.CreatedAt, &loss.BurnScript)
 		if err != nil {
 			return nil, err
 		}

diff --git a/storage/sqlite/migrations/20240607043408_add_burn_script_to_losses.sql b/storage/sqlite/migrations/20240607043408_add_burn_script_to_losses.sql
@@ -0,0 +1,11 @@
+-- +goose Up
+-- +goose StatementBegin
+ALTER TABLE losses ADD COLUMN burn_script TEXT DEFAULT '';
+CREATE INDEX losses_burn_script_idx ON losses (burn_script);
+-- +goose StatementEnd
+
+-- +goose Down
+-- +goose StatementBegin
+DROP INDEX losses_burn_script_idx;
+ALTER TABLE losses DROP COLUMN burn_script;
+-- +goose StatementEnd
diff --git a/storage/sqlite/tracker.go b/storage/sqlite/tracker.go
@@ -36,7 +36,7 @@ func (d *SqliteBackend) RecordBlockIndexResults(block types.Block, txoutset type
 
 	// Insert the losses records into the losses table
 	for _, loss := range losses {
-		_, err = tx.Exec("INSERT INTO losses (tx_id, block_hash, block_height, vout, amount) VALUES (?, ?, ?, ?, ?) ON CONFLICT DO NOTHING", loss.TxID, loss.BlockHash, loss.BlockHeight, loss.Vout, loss.Amount)
+		_, err = tx.Exec("INSERT INTO losses (tx_id, block_hash, block_height, vout, amount, burn_script) VALUES (?, ?, ?, ?, ?, ?) ON CONFLICT (tx_id, vout) DO UPDATE SET block_hash=excluded.block_hash, block_height=excluded.block_height, vout=excluded.vout, amount=excluded.amount, burn_script=excluded.burn_script", loss.TxID, loss.BlockHash, loss.BlockHeight, loss.Vout, loss.Amount, loss.BurnScript)
 		if err != nil {
 			tx.Rollback()
 			return fmt.Errorf("failed to insert loss record: %v", err)
@@ -74,7 +74,7 @@ func (d *SqliteBackend) RecordTransactionIndexResults(losses []types.Loss, trans
 
 	// Insert the losses records into the losses table
 	for _, loss := range losses {
-		_, err = tx.Exec("INSERT INTO losses (tx_id, block_hash, block_height, vout, amount) VALUES (?, ?, ?, ?, ?) ON CONFLICT (tx_id, vout) DO NOTHING", loss.TxID, loss.BlockHash, loss.BlockHeight, loss.Vout, loss.Amount)
+		_, err = tx.Exec("INSERT INTO losses (tx_id, block_hash, block_height, vout, amount, burn_script) VALUES (?, ?, ?, ?, ?, ?) ON CONFLICT (tx_id, vout) DO UPDATE SET block_hash=excluded.block_hash, block_height=excluded.block_height, vout=excluded.vout, amount=excluded.amount, burn_script=excluded.burn_script", loss.TxID, loss.BlockHash, loss.BlockHeight, loss.Vout, loss.Amount, loss.BurnScript)
 		if err != nil {
 			tx.Rollback()
 			return fmt.Errorf("failed to insert loss record: %v", err)

diff --git a/tracker/tracker.go b/tracker/tracker.go
@@ -340,6 +340,7 @@ func (t *Tracker) scanTransactions(blockhash string, blockHeight int64, transact
 					BlockHeight: blockHeight,
 					Vout:        vout.N,
 					Amount:      types.FromBTCFloat64(vout.Value),
+					BurnScript:  vout.ScriptPubKey.Hex,
 				})
 
 				atLeastOneBurn = true
@@ -359,6 +360,7 @@ func (t *Tracker) scanTransactions(blockhash string, blockHeight int64, transact
 						BlockHeight: blockHeight,
 						Vout:        vout.N,
 						Amount:      types.FromBTCFloat64(vout.Value),
+						BurnScript:  vout.ScriptPubKey.Hex,
 					})
 
 					atLeastOneBurn = true

diff --git a/types/loss.go b/types/loss.go
@@ -10,4 +10,5 @@ type Loss struct {
 	Vout        int
 	Amount      *BigInt
 	CreatedAt   time.Time
+	BurnScript  string
 }