feat: erc721 total actions

migrations/evm/20240925063851_erc721-contract-total-actions.ts

@@ -0,0 +1,76 @@
+import { Knex } from 'knex';
+import { Erc721Activity } from '../../src/models';
+import _ from 'lodash';
+import { ERC721_ACTION } from '../../src/services/evm/erc721_handler';
+
+export async function up(knex: Knex): Promise<void> {
+  await knex.schema.alterTable('erc721_contract', (table) => {
+    table.jsonb('total_actions').defaultTo('{}');
+  });
+  await knex.raw(`set statement_timeout to 0`);
+  const [totalTransfer, totalApproval, totalApprovalForAll] = await Promise.all(
+    [
+      _.keyBy(
+        (
+          await Erc721Activity.query(knex)
+            .where('action', ERC721_ACTION.TRANSFER)
+            .groupBy('erc721_activity.erc721_contract_address')
+            .select(
+              'erc721_activity.erc721_contract_address',
+              knex.raw('count(*)::integer as transfer')
+            )
+        ).map((e) => e.toJSON()),
+        'erc721_contract_address'
+      ),
+      _.keyBy(
+        (
+          await Erc721Activity.query(knex)
+            .where('action', ERC721_ACTION.APPROVAL)
+            .groupBy('erc721_activity.erc721_contract_address')
+            .select(
+              'erc721_activity.erc721_contract_address',
+              knex.raw('count(*)::integer as approval')
+            )
+        ).map((e) => e.toJSON()),
+        'erc721_contract_address'
+      ),
+      _.keyBy(
+        (
+          await Erc721Activity.query(knex)
+            .where('action', ERC721_ACTION.APPROVAL_FOR_ALL)
+            .groupBy('erc721_activity.erc721_contract_address')
+            .select(
+              'erc721_activity.erc721_contract_address',
+              knex.raw('count(*)::integer as approval_for_all')
+            )
+        ).map((e) => e.toJSON()),
+        'erc721_contract_address'
+      ),
+    ]
+  );
+  const totalAction = _.merge(
+    totalTransfer,
+    totalApproval,
+    totalApprovalForAll
+  );
+  const erc721ContractsAddr = Object.keys(totalAction);
+  if (erc721ContractsAddr.length > 0) {
+    const stringListUpdates = erc721ContractsAddr
+      .map(
+        (addr) =>
+          `('${addr}', '${JSON.stringify(
+            _.omit(totalAction[addr], 'erc721_contract_address')
+          )}'::jsonb)`
+      )
+      .join(',');
+    await knex.raw(
+      `UPDATE erc721_contract SET total_actions = temp.total_actions from (VALUES ${stringListUpdates}) as temp(address, total_actions) where temp.address = erc721_contract.address`
+    );
+  }
+}
+
+export async function down(knex: Knex): Promise<void> {
+  await knex.schema.alterTable('erc721_contract', (table) => {
+    table.dropColumn('total_actions');
+  });
+}

src/models/erc721_contract.ts

@@ -27,6 +27,8 @@ export class Erc721Contract extends BaseModel {
 
   total_supply!: string;
 
+  total_actions!: any;
+
   static get tableName() {
     return 'erc721_contract';
   }

src/services/evm/erc721_handler.ts

@@ -92,6 +92,11 @@ export class Erc721Handler {
       if (erc721Activity.action === ERC721_ACTION.TRANSFER) {
         this.handlerErc721Transfer(erc721Activity);
       }
+      this.erc721Contracts[
+        erc721Activity.erc721_contract_address
+      ].total_actions[erc721Activity.action] =
+        (this.erc721Contracts[erc721Activity.erc721_contract_address]
+          .total_actions[erc721Activity.action] || 0) + 1;
     });
   }
 
@@ -321,12 +326,14 @@ export class Erc721Handler {
       const stringListUpdates = erc721Contracts
         .map(
           (erc721Contract) =>
-            `(${erc721Contract.id}, ${erc721Contract.total_supply})`
+            `(${erc721Contract.id}, ${
+              erc721Contract.total_supply
+            }, '${JSON.stringify(erc721Contract.total_actions)}'::jsonb)`
         )
         .join(',');
       await knex
         .raw(
-          `UPDATE erc721_contract SET total_supply = temp.total_supply from (VALUES ${stringListUpdates}) as temp(id, total_supply) where temp.id = erc721_contract.id`
+          `UPDATE erc721_contract SET total_supply = temp.total_supply, total_actions = temp.total_actions from (VALUES ${stringListUpdates}) as temp(id, total_supply, total_actions) where temp.id = erc721_contract.id`
         )
         .transacting(trx);
     }

test/unit/services/evm/erc721.spec.ts

@@ -213,6 +213,10 @@ export default class Erc721Test {
       .first()
       .throwIfNotFound();
     expect(erc721Contract.total_supply).toEqual('1');
+    expect(erc721Contract.total_actions).toEqual({
+      [ERC721_ACTION.TRANSFER]: 2,
+      [ERC721_ACTION.APPROVAL]: 1,
+    });
     const erc721HolderStats = _.keyBy(
       await Erc721HolderStatistic.query(),
       'owner'

test/unit/services/evm/erc721_handler.spec.ts

@@ -223,6 +223,7 @@ export default class Erc721HandlerTest {
 
   @Test('test process')
   async testProcess() {
+    const totalTransfer = 5;
     const holders = [
       '0x38828FA9766dE6eb49011fCC970ed1beFE15974a',
       '0x27CF763feEE58f14C2c69663EA8De784FCF2Dbbb',
@@ -236,6 +237,11 @@ export default class Erc721HandlerTest {
       track: true,
       last_updated_height: 0,
       total_supply: '3',
+      total_actions: {
+        [ERC721_ACTION.TRANSFER]: totalTransfer,
+        [ERC721_ACTION.APPROVAL]: 1,
+        [ERC721_ACTION.APPROVAL_FOR_ALL]: 2,
+      },
     });
     const initErc721Tokens = {};
     const initErc721HolderStats = {};
@@ -323,6 +329,14 @@ export default class Erc721HandlerTest {
 
     // check total supply
     expect(updatedErc721Contract.total_supply).toEqual('6');
+    // check total actions
+    expect(updatedErc721Contract.total_actions).toEqual({
+      [ERC721_ACTION.TRANSFER]: totalTransfer + erc721Activities.length,
+      [ERC721_ACTION.APPROVAL]:
+        initErc721Contract.total_actions[ERC721_ACTION.APPROVAL],
+      [ERC721_ACTION.APPROVAL_FOR_ALL]:
+        initErc721Contract.total_actions[ERC721_ACTION.APPROVAL_FOR_ALL],
+    });
     // check new owner for each token
     const updatedTokens = erc721Handler.erc721Tokens;
     expect(updatedTokens[`${initErc721Contract.address}_0`].owner).toEqual(