@sentio/sdk 2.59.5-rc.1 → 2.60.0-rc.2

package/src/fuel/fuel-plugin.ts

@@ -1,12 +1,11 @@
 import { errorString, GLOBAL_CONFIG, mergeProcessResults, Plugin, PluginManager, USER_PROCESSOR } from '@sentio/runtime'
 import { PartitionHandlerManager } from '../core/index.js'
+import { HandlerRegister } from '../core/handler-register.js'
 import {
   ContractConfig,
-  Data_FuelBlock,
-  Data_FuelReceipt,
-  Data_FuelTransaction,
   DataBinding,
   HandlerType,
+  InitResponse,
   ProcessConfigResponse,
   ProcessResult,
   ProcessStreamResponse_Partitions,
@@ -20,30 +19,26 @@ import { FuelProcessorState } from './types.js'
 import { FuelProcessor } from './fuel-processor.js'
 import { FuelGlobalProcessor } from './global-processor.js'
 
-interface Handlers {
-  transactionHandlers: ((trace: Data_FuelTransaction) => Promise<ProcessResult>)[]
-  blockHandlers: ((block: Data_FuelBlock) => Promise<ProcessResult>)[]
-  receiptHandlers: ((log: Data_FuelReceipt) => Promise<ProcessResult>)[]
-}
-
 export class FuelPlugin extends Plugin {
   name: string = 'FuelPlugin'
-  handlers: Handlers = {
-    transactionHandlers: [],
-    blockHandlers: [],
-    receiptHandlers: []
-  }
-
+  handlerRegister = new HandlerRegister()
   partitionManager = new PartitionHandlerManager()
 
-  async configure(config: ProcessConfigResponse) {
-    const handlers: Handlers = {
-      transactionHandlers: [],
-      blockHandlers: [],
-      receiptHandlers: []
+  async init(config: InitResponse) {
+    for (const fuelProcessor of FuelProcessorState.INSTANCE.getValues()) {
+      const chainId = fuelProcessor.config.chainId
+      config.chainIds.push(chainId)
     }
+  }
+
+  async configure(config: ProcessConfigResponse, forChainId?: string) {
+    this.handlerRegister.clear(forChainId as any)
 
     for (const processor of FuelProcessorState.INSTANCE.getValues()) {
+      const chainId = processor.config.chainId
+      if (forChainId !== undefined && forChainId !== chainId.toString()) {
+        continue
+      }
       const processorConfig = processor.config
       const contractConfig = ContractConfig.fromPartial({
         processorType: USER_PROCESSOR,
@@ -57,7 +52,7 @@ export class FuelPlugin extends Plugin {
         endBlock: processorConfig.endBlock
       })
       for (const txHandler of processor.txHandlers) {
-        const handlerId = handlers.transactionHandlers.push(txHandler.handler) - 1
+        const handlerId = this.handlerRegister.register(txHandler.handler, chainId)
         this.partitionManager.registerPartitionHandler(
           HandlerType.FUEL_TRANSACTION,
           handlerId,
@@ -90,7 +85,7 @@ export class FuelPlugin extends Plugin {
       }
 
       for (const receiptHandler of processor.receiptHandlers ?? []) {
-        const handlerId = handlers.receiptHandlers.push(receiptHandler.handler) - 1
+        const handlerId = this.handlerRegister.register(receiptHandler.handler, chainId)
         this.partitionManager.registerPartitionHandler(
           HandlerType.FUEL_RECEIPT,
           handlerId,
@@ -107,7 +102,7 @@ export class FuelPlugin extends Plugin {
       }
 
       for (const blockHandler of processor.blockHandlers) {
-        const handlerId = handlers.blockHandlers.push(blockHandler.handler) - 1
+        const handlerId = this.handlerRegister.register(blockHandler.handler, chainId)
         this.partitionManager.registerPartitionHandler(HandlerType.FUEL_BLOCK, handlerId, blockHandler.partitionHandler)
         contractConfig.intervalConfigs.push({
           slot: 0,
@@ -123,8 +118,6 @@ export class FuelPlugin extends Plugin {
 
       config.contractConfigs.push(contractConfig)
     }
-
-    this.handlers = handlers
   }
 
   supportedHandlers = [
@@ -212,12 +205,14 @@ export class FuelPlugin extends Plugin {
     const promises: Promise<ProcessResult>[] = []
 
     for (const handlerId of binding.handlerIds) {
-      const promise = this.handlers.receiptHandlers[handlerId](receipt).catch((e) => {
-        throw new ServerError(
-          Status.INTERNAL,
-          'error processing transaction: ' + JSON.stringify(receipt) + '\n' + errorString(e)
-        )
-      })
+      const promise = this.handlerRegister
+        .getHandlerById(handlerId)(receipt)
+        .catch((e: any) => {
+          throw new ServerError(
+            Status.INTERNAL,
+            'error processing transaction: ' + JSON.stringify(receipt) + '\n' + errorString(e)
+          )
+        })
       if (GLOBAL_CONFIG.execution.sequential) {
         await promise
       }
@@ -235,12 +230,14 @@ export class FuelPlugin extends Plugin {
     const promises: Promise<ProcessResult>[] = []
 
     for (const handlerId of binding.handlerIds) {
-      const promise = this.handlers.transactionHandlers[handlerId](fuelTransaction).catch((e) => {
-        throw new ServerError(
-          Status.INTERNAL,
-          'error processing transaction: ' + JSON.stringify(fuelTransaction.transaction) + '\n' + errorString(e)
-        )
-      })
+      const promise = this.handlerRegister
+        .getHandlerById(handlerId)(fuelTransaction)
+        .catch((e: any) => {
+          throw new ServerError(
+            Status.INTERNAL,
+            'error processing transaction: ' + JSON.stringify(fuelTransaction.transaction) + '\n' + errorString(e)
+          )
+        })
       if (GLOBAL_CONFIG.execution.sequential) {
         await promise
       }
@@ -257,13 +254,15 @@ export class FuelPlugin extends Plugin {
 
     const promises: Promise<ProcessResult>[] = []
     for (const handlerId of binding.handlerIds) {
-      const promise = this.handlers.blockHandlers[handlerId](ethBlock).catch((e) => {
-        console.error('error processing block: ', e)
-        throw new ServerError(
-          Status.INTERNAL,
-          'error processing block: ' + ethBlock.block?.height + '\n' + errorString(e)
-        )
-      })
+      const promise = this.handlerRegister
+        .getHandlerById(handlerId)(ethBlock)
+        .catch((e: any) => {
+          console.error('error processing block: ', e)
+          throw new ServerError(
+            Status.INTERNAL,
+            'error processing block: ' + ethBlock.block?.height + '\n' + errorString(e)
+          )
+        })
       if (GLOBAL_CONFIG.execution.sequential) {
         await promise
       }

package/src/solana/solana-plugin.ts

@@ -1,5 +1,12 @@
 import { errorString, mergeProcessResults, Plugin, PluginManager, USER_PROCESSOR } from '@sentio/runtime'
-import { ContractConfig, DataBinding, HandlerType, ProcessConfigResponse, ProcessResult } from '@sentio/protos'
+import {
+  ContractConfig,
+  DataBinding,
+  HandlerType,
+  InitResponse,
+  ProcessConfigResponse,
+  ProcessResult
+} from '@sentio/protos'
 
 import { ServerError, Status } from 'nice-grpc'
 
@@ -9,9 +16,20 @@ import { Instruction as SolInstruction } from '@coral-xyz/anchor'
 export class SolanaPlugin extends Plugin {
   name: string = 'SolanaPlugin'
 
-  async configure(config: ProcessConfigResponse) {
+  async init(config: InitResponse) {
+    for (const solanaProcessor of SolanaProcessorState.INSTANCE.getValues()) {
+      const chainId = solanaProcessor.network
+      config.chainIds.push(chainId)
+    }
+  }
+
+  async configure(config: ProcessConfigResponse, forChainId?: string) {
     // Part 2, prepare solana constractors
     for (const solanaProcessor of SolanaProcessorState.INSTANCE.getValues()) {
+      const chainId = solanaProcessor.network
+      if (forChainId !== undefined && forChainId !== chainId.toString()) {
+        continue
+      }
       const contractConfig = ContractConfig.fromPartial({
         processorType: USER_PROCESSOR,
         contract: {

package/src/stark/starknet-plugin.ts

@@ -1,9 +1,9 @@
 import { errorString, GLOBAL_CONFIG, mergeProcessResults, Plugin, PluginManager, USER_PROCESSOR } from '@sentio/runtime'
 import {
   ContractConfig,
-  Data_StarknetEvent,
   DataBinding,
   HandlerType,
+  InitResponse,
   ProcessConfigResponse,
   ProcessResult,
   StartRequest
@@ -11,25 +11,29 @@ import {
 
 import { ServerError, Status } from 'nice-grpc'
 import { TemplateInstanceState } from '../core/template.js'
+import { HandlerRegister } from '../core/handler-register.js'
 import { StarknetProcessorState } from './starknet-processor.js'
 import { hash } from 'starknet'
 
-interface Handlers {
-  callHandlers: ((trace: Data_StarknetEvent) => Promise<ProcessResult>)[]
-}
-
 export class StarknetPlugin extends Plugin {
   name: string = 'StarknetPlugin'
-  handlers: Handlers = {
-    callHandlers: []
-  }
+  handlerRegister = new HandlerRegister()
 
-  async configure(config: ProcessConfigResponse) {
-    const handlers: Handlers = {
-      callHandlers: []
+  async init(config: InitResponse) {
+    for (const solanaProcessor of StarknetProcessorState.INSTANCE.getValues()) {
+      const chainId = solanaProcessor.config.chainId
+      config.chainIds.push(chainId)
     }
+  }
+
+  async configure(config: ProcessConfigResponse, forChainId?: string) {
+    this.handlerRegister.clear(forChainId as any)
 
     for (const processor of StarknetProcessorState.INSTANCE.getValues()) {
+      const chainId = processor.config.chainId
+      if (forChainId !== undefined && forChainId !== chainId.toString()) {
+        continue
+      }
       await processor.configure()
       const contractConfig = ContractConfig.fromPartial({
         processorType: USER_PROCESSOR,
@@ -43,7 +47,7 @@ export class StarknetPlugin extends Plugin {
         endBlock: processor.config.endBlock
       })
       for (const callHandler of processor.callHandlers) {
-        const handlerId = handlers.callHandlers.push(callHandler.handler) - 1
+        const handlerId = this.handlerRegister.register(callHandler.handler, chainId)
 
         if (callHandler.eventFilter) {
           contractConfig.starknetEventConfigs.push({
@@ -60,8 +64,6 @@ export class StarknetPlugin extends Plugin {
       // Finish up a contract
       config.contractConfigs.push(contractConfig)
     }
-
-    this.handlers = handlers
   }
 
   supportedHandlers = [HandlerType.STARKNET_EVENT]
@@ -91,12 +93,14 @@ export class StarknetPlugin extends Plugin {
     const result = binding.data?.starknetEvents?.result
 
     for (const handlerId of binding.handlerIds) {
-      const promise = this.handlers.callHandlers[handlerId](binding.data?.starknetEvents).catch((e) => {
-        throw new ServerError(
-          Status.INTERNAL,
-          'error processing transaction: ' + JSON.stringify(result) + '\n' + errorString(e)
-        )
-      })
+      const promise = this.handlerRegister
+        .getHandlerById(handlerId)(binding.data?.starknetEvents)
+        .catch((e: any) => {
+          throw new ServerError(
+            Status.INTERNAL,
+            'error processing transaction: ' + JSON.stringify(result) + '\n' + errorString(e)
+          )
+        })
       if (GLOBAL_CONFIG.execution.sequential) {
         await promise
       }

package/src/sui/sui-plugin.ts

@@ -2,12 +2,9 @@ import { errorString, mergeProcessResults, Plugin, PluginManager, USER_PROCESSOR
 import {
   AccountConfig,
   ContractConfig,
-  Data_SuiCall,
-  Data_SuiEvent,
-  Data_SuiObject,
-  Data_SuiObjectChange,
   DataBinding,
   HandlerType,
+  InitResponse,
   MoveCallHandlerConfig,
   MoveEventHandlerConfig,
   MoveResourceChangeConfig,
@@ -19,6 +16,7 @@ import {
 
 import { ServerError, Status } from 'nice-grpc'
 import { PartitionHandlerManager } from '../core/index.js'
+import { HandlerRegister } from '../core/handler-register.js'
 
 import { SuiProcessorState } from './sui-processor.js'
 import { SuiAccountProcessorState, SuiAddressProcessor } from './sui-object-processor.js'
@@ -31,22 +29,9 @@ import {
 import { SuiNetwork } from './network.js'
 import { SuiContext } from './context.js'
 
-interface Handlers {
-  suiEventHandlers: ((event: Data_SuiEvent) => Promise<ProcessResult>)[]
-  suiCallHandlers: ((func: Data_SuiCall) => Promise<ProcessResult>)[]
-  suiObjectHandlers: ((object: Data_SuiObject) => Promise<ProcessResult>)[]
-  suiObjectChangeHandlers: ((object: Data_SuiObjectChange) => Promise<ProcessResult>)[]
-}
-
 export class SuiPlugin extends Plugin {
   name: string = 'SuiPlugin'
-  handlers: Handlers = {
-    suiCallHandlers: [],
-    suiEventHandlers: [],
-    suiObjectHandlers: [],
-    suiObjectChangeHandlers: []
-  }
-
+  handlerRegister = new HandlerRegister()
   partitionManager = new PartitionHandlerManager()
   async start(request: StartRequest): Promise<void> {
     await initCoinList()
@@ -76,14 +61,21 @@ export class SuiPlugin extends Plugin {
     }
   }
 
-  async configure(config: ProcessConfigResponse) {
-    const handlers: Handlers = {
-      suiCallHandlers: [],
-      suiEventHandlers: [],
-      suiObjectHandlers: [],
-      suiObjectChangeHandlers: []
+  async init(config: InitResponse) {
+    for (const state of [SuiProcessorState.INSTANCE, SuiAccountProcessorState.INSTANCE]) {
+      for (const suiProcessor of state.getValues()) {
+        config.chainIds.push(suiProcessor.config.network)
+      }
     }
+  }
+
+  async configure(config: ProcessConfigResponse, forChainId?: string) {
+    this.handlerRegister.clear(forChainId as any)
     for (const suiProcessor of SuiProcessorState.INSTANCE.getValues()) {
+      const chainId = suiProcessor.config.network
+      if (forChainId !== undefined && forChainId !== chainId.toString()) {
+        continue
+      }
       const contractConfig = ContractConfig.fromPartial({
         transactionConfig: [],
         processorType: USER_PROCESSOR,
@@ -97,7 +89,7 @@ export class SuiPlugin extends Plugin {
         endBlock: suiProcessor.config.endCheckpoint
       })
       for (const handler of suiProcessor.eventHandlers) {
-        const handlerId = handlers.suiEventHandlers.push(handler.handler) - 1
+        const handlerId = this.handlerRegister.register(handler.handler, chainId)
         this.partitionManager.registerPartitionHandler(HandlerType.SUI_EVENT, handlerId, handler.partitionHandler)
         const eventHandlerConfig: MoveEventHandlerConfig = {
           filters: handler.filters.map((f) => {
@@ -114,7 +106,7 @@ export class SuiPlugin extends Plugin {
         contractConfig.moveEventConfigs.push(eventHandlerConfig)
       }
       for (const handler of suiProcessor.callHandlers) {
-        const handlerId = handlers.suiCallHandlers.push(handler.handler) - 1
+        const handlerId = this.handlerRegister.register(handler.handler, chainId)
         this.partitionManager.registerPartitionHandler(HandlerType.SUI_CALL, handlerId, handler.partitionHandler)
         const functionHandlerConfig: MoveCallHandlerConfig = {
           filters: handler.filters.map((filter) => {
@@ -135,7 +127,7 @@ export class SuiPlugin extends Plugin {
       }
       // deprecated, use objectType processor instead
       for (const handler of suiProcessor.objectChangeHandlers) {
-        const handlerId = handlers.suiObjectChangeHandlers.push(handler.handler) - 1
+        const handlerId = this.handlerRegister.register(handler.handler, chainId)
         const objectChangeHandler: MoveResourceChangeConfig = {
           type: handler.type,
           handlerId,
@@ -148,6 +140,10 @@ export class SuiPlugin extends Plugin {
     }
 
     for (const processor of SuiAccountProcessorState.INSTANCE.getValues()) {
+      const chainId = processor.getChainId()
+      if (forChainId !== undefined && forChainId !== chainId.toString()) {
+        continue
+      }
       const accountConfig = AccountConfig.fromPartial({
         address: processor.config.address,
         chainId: processor.getChainId(),
@@ -156,7 +152,7 @@ export class SuiPlugin extends Plugin {
       })
 
       for (const handler of processor.objectChangeHandlers) {
-        const handlerId = handlers.suiObjectChangeHandlers.push(handler.handler) - 1
+        const handlerId = this.handlerRegister.register(handler.handler, chainId)
         const objectChangeHandler: MoveResourceChangeConfig = {
           type: handler.type,
           handlerId,
@@ -167,7 +163,7 @@ export class SuiPlugin extends Plugin {
       }
 
       for (const handler of processor.objectHandlers) {
-        const handlerId = handlers.suiObjectHandlers.push(handler.handler) - 1
+        const handlerId = this.handlerRegister.register(handler.handler, chainId)
 
         accountConfig.moveIntervalConfigs.push({
           intervalConfig: {
@@ -188,7 +184,7 @@ export class SuiPlugin extends Plugin {
 
       if (processor instanceof SuiAddressProcessor) {
         for (const handler of processor.callHandlers) {
-          const handlerId = handlers.suiCallHandlers.push(handler.handler) - 1
+          const handlerId = this.handlerRegister.register(handler.handler, chainId)
           const functionHandlerConfig: MoveCallHandlerConfig = {
             filters: handler.filters.map((filter) => {
               return {
@@ -210,7 +206,6 @@ export class SuiPlugin extends Plugin {
 
       config.accountConfigs.push(accountConfig)
     }
-    this.handlers = handlers
   }
 
   supportedHandlers = [
@@ -284,12 +279,14 @@ export class SuiPlugin extends Plugin {
 
     for (const handlerId of binding.handlerIds) {
       promises.push(
-        this.handlers.suiEventHandlers[handlerId](event).catch((e) => {
-          throw new ServerError(
-            Status.INTERNAL,
-            'error processing event: ' + JSON.stringify(event) + '\n' + errorString(e)
-          )
-        })
+        this.handlerRegister
+          .getHandlerById(handlerId)(event)
+          .catch((e: any) => {
+            throw new ServerError(
+              Status.INTERNAL,
+              'error processing event: ' + JSON.stringify(event) + '\n' + errorString(e)
+            )
+          })
       )
     }
     return mergeProcessResults(await Promise.all(promises))
@@ -303,9 +300,14 @@ export class SuiPlugin extends Plugin {
 
     const promises: Promise<ProcessResult>[] = []
     for (const handlerId of binding.handlerIds) {
-      const promise = this.handlers.suiCallHandlers[handlerId](call).catch((e) => {
-        throw new ServerError(Status.INTERNAL, 'error processing call: ' + JSON.stringify(call) + '\n' + errorString(e))
-      })
+      const promise = this.handlerRegister
+        .getHandlerById(handlerId)(call)
+        .catch((e: any) => {
+          throw new ServerError(
+            Status.INTERNAL,
+            'error processing call: ' + JSON.stringify(call) + '\n' + errorString(e)
+          )
+        })
       promises.push(promise)
     }
     return mergeProcessResults(await Promise.all(promises))
@@ -320,12 +322,14 @@ export class SuiPlugin extends Plugin {
     const promises: Promise<ProcessResult>[] = []
     for (const handlerId of binding.handlerIds) {
       promises.push(
-        this.handlers.suiObjectHandlers[handlerId](object).catch((e) => {
-          throw new ServerError(
-            Status.INTERNAL,
-            'error processing object: ' + JSON.stringify(object) + '\n' + errorString(e)
-          )
-        })
+        this.handlerRegister
+          .getHandlerById(handlerId)(object)
+          .catch((e: any) => {
+            throw new ServerError(
+              Status.INTERNAL,
+              'error processing object: ' + JSON.stringify(object) + '\n' + errorString(e)
+            )
+          })
       )
     }
     return mergeProcessResults(await Promise.all(promises))
@@ -340,12 +344,14 @@ export class SuiPlugin extends Plugin {
     const promises: Promise<ProcessResult>[] = []
     for (const handlerId of binding.handlerIds) {
       promises.push(
-        this.handlers.suiObjectChangeHandlers[handlerId](objectChange).catch((e) => {
-          throw new ServerError(
-            Status.INTERNAL,
-            'error processing object change: ' + JSON.stringify(objectChange) + '\n' + errorString(e)
-          )
-        })
+        this.handlerRegister
+          .getHandlerById(handlerId)(objectChange)
+          .catch((e: any) => {
+            throw new ServerError(
+              Status.INTERNAL,
+              'error processing object change: ' + JSON.stringify(objectChange) + '\n' + errorString(e)
+            )
+          })
       )
     }
     return mergeProcessResults(await Promise.all(promises))