fix: add compute units to jsonrpc parser

solana-labs · Sep 1, 2022 · cb57f8f · cb57f8f
1 parent 31087b8
commit cb57f8f
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/web3.js/src/connection.ts b/web3.js/src/connection.ts
@@ -888,6 +888,8 @@ export type ParsedTransactionMeta = {
   err: TransactionError | null;
   /** The collection of addresses loaded using address lookup tables */
   loadedAddresses?: LoadedAddresses;
+  /** The compute units consumed after processing the transaction */
+  computeUnitsConsumed?: number;
 };
 
 export type CompiledInnerInstruction = {
@@ -917,6 +919,8 @@ export type ConfirmedTransactionMeta = {
   err: TransactionError | null;
   /** The collection of addresses loaded using address lookup tables */
   loadedAddresses?: LoadedAddresses;
+  /** The compute units consumed after processing the transaction */
+  computeUnitsConsumed?: number;
 };
 
 /**
@@ -1993,6 +1997,7 @@ const ConfirmedTransactionMetaResult = pick({
   preTokenBalances: optional(nullable(array(TokenBalanceResult))),
   postTokenBalances: optional(nullable(array(TokenBalanceResult))),
   loadedAddresses: optional(LoadedAddressesResult),
+  computeUnitsConsumed: optional(number()),
 });
 
 /**
@@ -2017,6 +2022,7 @@ const ParsedConfirmedTransactionMetaResult = pick({
   preTokenBalances: optional(nullable(array(TokenBalanceResult))),
   postTokenBalances: optional(nullable(array(TokenBalanceResult))),
   loadedAddresses: optional(LoadedAddressesResult),
+  computeUnitsConsumed: optional(number()),
 });
 
 const TransactionVersionStruct = union([literal(0), literal('legacy')]);

diff --git a/web3.js/test/connection.test.ts b/web3.js/test/connection.test.ts
@@ -4460,6 +4460,8 @@ describe('Connection', function () {
           readonly: [],
           writable: [lookupTableAddresses[0]],
         });
+        expect(fetchedTransaction.meta?.computeUnitsConsumed).to.not.be
+          .undefined;
         expect(
           fetchedTransaction.transaction.message.addressTableLookups,
         ).to.eql(addressTableLookups);
@@ -4489,6 +4491,8 @@ describe('Connection', function () {
           readonly: [],
           writable: [lookupTableAddresses[0]],
         });
+        expect(parsedTransaction?.meta?.computeUnitsConsumed).to.not.be
+          .undefined;
         expect(
           parsedTransaction?.transaction.message.addressTableLookups,
         ).to.eql(addressTableLookups);