feat: tx list function name fuzzy search

Author: He1DAr

migrations/1727186561690_idx-contract-call-function-name-trgm.js

@@ -0,0 +1,19 @@
+/** @param { import("node-pg-migrate").MigrationBuilder } pgm */
+exports.up = pgm => {
+  pgm.sql('CREATE EXTENSION IF NOT EXISTS pg_trgm;');
+
+  pgm.sql(`
+    CREATE INDEX idx_contract_call_function_name_trgm
+    ON txs
+    USING gin (contract_call_function_name gin_trgm_ops);
+  `);
+};
+
+/** @param { import("node-pg-migrate").MigrationBuilder } pgm */
+exports.down = pgm => {
+  pgm.sql(`
+    DROP INDEX IF EXISTS idx_contract_call_function_name_trgm;
+  `);
+
+  pgm.sql('DROP EXTENSION IF EXISTS pg_trgm;');
+};

src/api/routes/tx.ts

@@ -113,6 +113,12 @@ export const TxRoutes: FastifyPluginAsync<
               examples: [1706745599],
             })
           ),
+          search_term: Type.Optional(
+            Type.String({
+              description: 'Option to search for transactions by a search term',
+              examples: ['swap'],
+            })
+          ),
           contract_id: Type.Optional(
             Type.String({
               description: 'Option to filter results by contract ID',
@@ -178,6 +184,11 @@ export const TxRoutes: FastifyPluginAsync<
         contractId = req.query.contract_id;
       }
 
+      let searchTerm: string | undefined;
+      if (typeof req.query.search_term === 'string') {
+        searchTerm = req.query.search_term;
+      }
+
       const { results: txResults, total } = await fastify.db.getTxList({
         offset,
         limit,
@@ -188,6 +199,7 @@ export const TxRoutes: FastifyPluginAsync<
         startTime: req.query.start_time,
         endTime: req.query.end_time,
         contractId,
+        searchTerm,
         functionName: req.query.function_name,
         nonce: req.query.nonce,
         order: req.query.order,

src/datastore/pg-store.ts

@@ -100,6 +100,8 @@ import { Fragment } from 'postgres';
 
 export const MIGRATIONS_DIR = path.join(REPO_DIR, 'migrations');
 
+const TRGM_SIMILARITY_THRESHOLD = 0.3;
+
 /**
  * This is the main interface between the API and the Postgres database. It contains all methods that
  * query the DB in search for blockchain data to be returned via endpoints or WebSockets/Socket.IO.
@@ -1415,6 +1417,7 @@ export class PgStore extends BasePgStore {
     startTime,
     endTime,
     contractId,
+    searchTerm,
     functionName,
     nonce,
     order,
@@ -1429,6 +1432,7 @@ export class PgStore extends BasePgStore {
     startTime?: number;
     endTime?: number;
     contractId?: string;
+    searchTerm?: string;
     functionName?: string;
     nonce?: number;
     order?: 'desc' | 'asc';
@@ -1467,6 +1471,10 @@ export class PgStore extends BasePgStore {
       const contractIdFilterSql = contractId
         ? sql`AND contract_call_contract_id = ${contractId}`
         : sql``;
+
+      const searchTermFilterSql = searchTerm
+        ? sql`AND similarity(contract_call_function_name, ${searchTerm}) > ${TRGM_SIMILARITY_THRESHOLD}`
+        : sql``;
       const contractFuncFilterSql = functionName
         ? sql`AND contract_call_function_name = ${functionName}`
         : sql``;
@@ -1478,6 +1486,7 @@ export class PgStore extends BasePgStore {
         !startTime &&
         !endTime &&
         !contractId &&
+        !searchTerm &&
         !functionName &&
         !nonce;
 
@@ -1496,6 +1505,7 @@ export class PgStore extends BasePgStore {
         ${startTimeFilterSql}
         ${endTimeFilterSql}
         ${contractIdFilterSql}
+        ${searchTermFilterSql}
         ${contractFuncFilterSql}
         ${nonceFilterSql}
       `;
@@ -1510,6 +1520,7 @@ export class PgStore extends BasePgStore {
         ${startTimeFilterSql}
         ${endTimeFilterSql}
         ${contractIdFilterSql}
+        ${searchTermFilterSql}
         ${contractFuncFilterSql}
         ${nonceFilterSql}
         ${orderBySql}

src/tests/tx-tests.ts

@@ -2322,6 +2322,76 @@ describe('tx tests', () => {
     );
   });
 
+  test('tx list - filter by searchTerm using trigram', async () => {
+    const transferTokenTx = {
+      tx_id: '0x1111',
+      contract_call_function_name: 'transferToken',
+    };
+
+    const stakeTokenTx = {
+      tx_id: '0x2222',
+      contract_call_function_name: 'stakeToken',
+    };
+
+    const burnTokenTx = {
+      tx_id: '0x3333',
+      contract_call_function_name: 'burnToken',
+    }
+
+    const block1 = new TestBlockBuilder({ block_height: 1, index_block_hash: '0x01' })
+      .addTx(transferTokenTx)
+      .build();
+    await db.update(block1);
+
+    const block2 = new TestBlockBuilder({
+      block_height: 2,
+      index_block_hash: '0x02',
+      parent_block_hash: block1.block.block_hash,
+      parent_index_block_hash: block1.block.index_block_hash,
+    })
+      .addTx(stakeTokenTx)
+      .addTx(burnTokenTx)
+      .build();
+    await db.update(block2);
+
+    const searchTerm = 'transfer';
+
+    const txsReq = await supertest(api.server).get(`/extended/v1/tx?search_term=${searchTerm}`);
+    expect(txsReq.status).toBe(200);
+    expect(txsReq.body).toEqual(
+      expect.objectContaining({
+        results: [
+          expect.objectContaining({
+            tx_id: transferTokenTx.tx_id,
+          }),
+        ],
+      })
+    );
+
+    const broadSearchTerm = 'token';
+
+    const txsReqBroad = await supertest(api.server).get(
+      `/extended/v1/tx?search_term=${broadSearchTerm}`
+    );
+    expect(txsReqBroad.status).toBe(200);
+
+    expect(txsReqBroad.body).toEqual(
+      expect.objectContaining({
+        results: [
+          expect.objectContaining({
+            tx_id: burnTokenTx.tx_id,
+          }),
+          expect.objectContaining({
+            tx_id: stakeTokenTx.tx_id,
+          }),
+          expect.objectContaining({
+            tx_id: transferTokenTx.tx_id,
+          }),
+        ],
+      })
+    );
+  });
+
   test('tx list - filter by contract id/name', async () => {
     const testContractAddr = 'ST27W5M8BRKA7C5MZE2R1S1F4XTPHFWFRNHA9M04Y.hello-world';
     const testContractFnName = 'test-contract-fn';