add chain id to nft balance lookups

taikoxyz · Aug 8, 2023 · 33bdb41 · 33bdb41
1 parent cfdc095
commit 33bdb41
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/...711_alter_nft_balances_table_add_address_contract_address_token_id_and_chain_id_index.sql b/...711_alter_nft_balances_table_add_address_contract_address_token_id_and_chain_id_index.sql
@@ -0,0 +1,9 @@
+-- +goose Up
+-- +goose StatementBegin
+ALTER TABLE `nft_balances` ADD INDEX `address_contract_address_token_id_and_chain_id_index` (`address`, `contract_address`, `token_id`, `chain_id`);
+
+-- +goose StatementEnd
+-- +goose Down
+-- +goose StatementBegin
+DROP INDEX address_contract_address_token_id_and_chain_id_index on nft_balances;
+-- +goose StatementEnd
diff --git a/packages/eventindexer/repo/nft_balance.go b/packages/eventindexer/repo/nft_balance.go
@@ -42,6 +42,7 @@ func (r *NFTBalanceRepository) IncreaseBalance(
 		Where("contract_address = ?", opts.ContractAddress).
 		Where("token_id = ?", opts.TokenID).
 		Where("address = ?", opts.Address).
+		Where("chain_id = ?", opts.ChainID).
 		First(b).
 		Error
 	if err != nil {
@@ -80,6 +81,7 @@ func (r *NFTBalanceRepository) SubtractBalance(
 		Where("contract_address = ?", opts.ContractAddress).
 		Where("token_id = ?", opts.TokenID).
 		Where("address = ?", opts.Address).
+		Where("chain_id = ?", opts.ChainID).
 		First(b).
 		Error
 	if err != nil {