solana: allow to request all balances of transaction (#313)

subsquid · Jul 21, 2024 · 9a75781 · 9a75781
1 parent 36ebb4c
commit 9a75781
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 0 deletions.
diff --git a/common/changes/@subsquid/solana-stream/transaction-balances_2024-07-17-10-01.json b/common/changes/@subsquid/solana-stream/transaction-balances_2024-07-17-10-01.json
@@ -0,0 +1,10 @@
+{
+  "changes": [
+    {
+      "packageName": "@subsquid/solana-stream",
+      "comment": "allow to request all balances of transaction",
+      "type": "minor"
+    }
+  ],
+  "packageName": "@subsquid/solana-stream"
+}
diff --git a/solana/solana-stream/src/data/request.ts b/solana/solana-stream/src/data/request.ts
@@ -66,6 +66,7 @@ export interface InstructionRequestWhere {
 
 export interface InstructionRequestRelations {
     transaction?: boolean
+    transactionBalances?: boolean
     transactionTokenBalances?: boolean
     transactionInstructions?: boolean
     logs?: boolean

diff --git a/solana/solana-stream/src/rpc/filter.ts b/solana/solana-stream/src/rpc/filter.ts
@@ -173,6 +173,11 @@ class BlockFilter {
         return groupBy(this.block.instructions, i => i.transactionIndex)
     }
 
+    @def
+    private balancesByTx(): Map<number, Balance[]> {
+        return groupBy(this.block.balances, b => b.transactionIndex)
+    }
+
     @def
     private tokenBalancesByTx(): Map<number, TokenBalance[]> {
         return groupBy(this.block.tokenBalances, b => b.transactionIndex)
@@ -220,6 +225,9 @@ class BlockFilter {
             if (rel.transaction) {
                 this.include.transactions.add(this.getTransaction(ins.transactionIndex))
             }
+            if (rel.transactionBalances) {
+                include(this.include.balances, this.balancesByTx().get(ins.transactionIndex))
+            }
             if (rel.transactionTokenBalances) {
                 include(this.include.tokenBalances, this.tokenBalancesByTx().get(ins.transactionIndex))
             }