@sentio/runtime 2.61.0-rc.8 → 2.61.0

package/src/plugin.ts

@@ -1,14 +1,14 @@
 import {
   DataBinding,
   HandlerType,
-  InitResponse,
   PreparedData,
   PreprocessResult,
   ProcessConfigResponse,
   ProcessResult,
   ProcessStreamResponse_Partitions,
   ProcessStreamResponse_Partitions_Partition_SysValue,
-  StartRequest
+  StartRequest,
+  UpdateTemplatesRequest
 } from '@sentio/protos'
 import { IStoreContext, StoreContext } from './db-context.js'
 import { AsyncLocalStorage } from 'node:async_hooks'
@@ -60,12 +60,6 @@ export abstract class Plugin {
    * method used by action server only
    */
   shutdownServer() {}
-
-  /**
-   * Initialize the plugin, for service v3.
-   * @param config
-   */
-  async init(config: InitResponse): Promise<void> {}
 }
 
 export class PluginManager {
@@ -90,8 +84,10 @@ export class PluginManager {
     }
   }
 
-  async configure(config: ProcessConfigResponse, forChainId?: string): Promise<void> {
-    await Promise.all(this.plugins.map((plugin) => plugin.configure(config, forChainId)))
+  async configure(config: ProcessConfigResponse): Promise<void> {
+    for (const plugin of this.plugins) {
+      await plugin.configure(config)
+    }
   }
 
   start(start: StartRequest, actionServerPort?: number) {
@@ -149,7 +145,11 @@ export class PluginManager {
     })
   }
 
-  async init(resp: InitResponse) {
-    return Promise.all(this.plugins.map((plugin) => plugin.init(resp)))
+  async updateTemplates(request: UpdateTemplatesRequest) {
+    for (const plugin of this.plugins) {
+      await plugin.start({
+        templateInstances: request.templateInstances
+      })
+    }
   }
 }

package/src/service-v3.ts

@@ -1,20 +1,19 @@
 import {
-  ConfigureHandlersRequest,
-  ConfigureHandlersResponse,
   DataBinding,
   DeepPartial,
   Empty,
   HandlerType,
-  InitResponse,
+  ProcessConfigRequest,
   ProcessConfigResponse,
   ProcessorV3ServiceImplementation,
   ProcessResult,
   ProcessStreamRequest,
   ProcessStreamResponse,
-  ProcessStreamResponseV2,
-  StartRequest
+  ProcessStreamResponseV3,
+  StartRequest,
+  UpdateTemplatesRequest
 } from '@sentio/protos'
-import { CallContext } from 'nice-grpc'
+import { CallContext, ServerError, Status } from 'nice-grpc'
 import { AsyncIterable } from 'ix'
 import { PluginManager } from './plugin.js'
 import { Subject } from 'rxjs'
@@ -26,6 +25,7 @@ import { processMetrics } from './metrics.js'
 import { recordRuntimeInfo } from './service.js'
 import { DataBindingContext } from './db-context.js'
 import { TemplateInstanceState } from './state.js'
+import { freezeGlobalConfig } from './global-config.js'
 
 const { process_binding_count, process_binding_time, process_binding_error } = processMetrics
 
@@ -33,6 +33,7 @@ export class ProcessorServiceImplV3 implements ProcessorV3ServiceImplementation
   readonly enablePartition: boolean
   private readonly loader: () => Promise<any>
   private readonly shutdownHandler?: () => void
+  private started = false
 
   constructor(loader: () => Promise<any>, options?: any, shutdownHandler?: () => void) {
     this.loader = loader
@@ -41,18 +42,37 @@ export class ProcessorServiceImplV3 implements ProcessorV3ServiceImplementation
     this.enablePartition = options?.['enable-partition'] == true
   }
 
-  async init(request: Empty, context: CallContext): Promise<DeepPartial<InitResponse>> {
-    await this.loader()
-    const resp = InitResponse.fromPartial({
-      chainIds: []
-    })
-    await PluginManager.INSTANCE.init(resp)
-    resp.chainIds = Array.from(new Set(resp.chainIds))
-    return resp
+  async start(request: StartRequest, context: CallContext): Promise<Empty> {
+    if (this.started) {
+      return {}
+    }
+
+    freezeGlobalConfig()
+
+    try {
+      await this.loader()
+    } catch (e) {
+      throw new ServerError(Status.INVALID_ARGUMENT, 'Failed to load processor: ' + errorString(e))
+    }
+
+    await PluginManager.INSTANCE.start(request)
+
+    this.started = true
+    return {}
+  }
+
+  async getConfig(request: ProcessConfigRequest, context: CallContext): Promise<ProcessConfigResponse> {
+    if (!this.started) {
+      throw new ServerError(Status.UNAVAILABLE, 'Service Not started.')
+    }
+
+    const newConfig = ProcessConfigResponse.fromPartial({})
+    await PluginManager.INSTANCE.configure(newConfig)
+    return newConfig
   }
 
   async *processBindingsStream(requests: AsyncIterable<ProcessStreamRequest>, context: CallContext) {
-    const subject = new Subject<DeepPartial<ProcessStreamResponseV2>>()
+    const subject = new Subject<DeepPartial<ProcessStreamResponseV3>>()
     this.handleRequests(requests, subject)
       .then(() => {
         subject.complete()
@@ -88,7 +108,7 @@ export class ProcessorServiceImplV3 implements ProcessorV3ServiceImplementation
   async handleRequest(
     request: ProcessStreamRequest,
     lastBinding: DataBinding | undefined,
-    subject: Subject<DeepPartial<ProcessStreamResponseV2>>
+    subject: Subject<DeepPartial<ProcessStreamResponseV3>>
   ) {
     if (request.binding) {
       process_binding_count.add(1)
@@ -141,7 +161,7 @@ export class ProcessorServiceImplV3 implements ProcessorV3ServiceImplementation
   private startProcess(
     processId: number,
     binding: DataBinding,
-    subject: Subject<DeepPartial<ProcessStreamResponseV2>>
+    subject: Subject<DeepPartial<ProcessStreamResponseV3>>
   ) {
     const context = this.contexts.new(processId, subject)
     const start = Date.now()
@@ -194,22 +214,9 @@ export class ProcessorServiceImplV3 implements ProcessorV3ServiceImplementation
       })
   }
 
-  async configureHandlers(
-    request: ConfigureHandlersRequest,
-    context: CallContext
-  ): Promise<DeepPartial<ConfigureHandlersResponse>> {
-    await PluginManager.INSTANCE.start(
-      StartRequest.fromPartial({
-        templateInstances: request.templateInstances
-      })
-    )
-
-    const newConfig = ProcessConfigResponse.fromPartial({})
-    await PluginManager.INSTANCE.configure(newConfig, request.chainId)
-    return {
-      accountConfigs: newConfig.accountConfigs,
-      contractConfigs: newConfig.contractConfigs
-    }
+  async updateTemplates(request: UpdateTemplatesRequest, context: CallContext): Promise<DeepPartial<Empty>> {
+    await PluginManager.INSTANCE.updateTemplates(request)
+    return {}
   }
 }
 
@@ -220,7 +227,7 @@ class Contexts {
     return this.contexts.get(processId)
   }
 
-  new(processId: number, subject: Subject<DeepPartial<ProcessStreamResponseV2>>) {
+  new(processId: number, subject: Subject<DeepPartial<ProcessStreamResponseV3>>) {
     const context = new DataBindingContext(processId, subject)
     this.contexts.set(processId, context)
     return context