@sentio/runtime 2.44.3 → 2.44.4-rc.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@sentio/runtime",
-  "version": "2.44.3",
+  "version": "2.44.4-rc.2",
   "license": "Apache-2.0",
   "type": "module",
   "exports": {

package/src/plugin.ts

@@ -15,7 +15,13 @@ export abstract class Plugin {
   supportedHandlers: HandlerType[] = []
 
   async configure(config: ProcessConfigResponse): Promise<void> {}
-  async start(start: StartRequest): Promise<void> {}
+  async start(start: StartRequest, actionServerPort?: number): Promise<void> {}
+
+  /**
+   * method used by action server only
+   * @param port
+   */
+  async startServer(port?: number): Promise<void> {}
 
   /**
    * @deprecated The method should not be used, use ctx.states instead
@@ -59,8 +65,12 @@ export class PluginManager {
     return Promise.all(this.plugins.map((plugin) => plugin.configure(config)))
   }
 
-  start(start: StartRequest) {
-    return Promise.all(this.plugins.map((plugin) => plugin.start(start)))
+  start(start: StartRequest, actionServerPort?: number) {
+    return Promise.all(this.plugins.map((plugin) => plugin.start(start, actionServerPort)))
+  }
+
+  startServer(port?: number) {
+    return Promise.all(this.plugins.map((plugin) => plugin.startServer(port)))
   }
 
   /**

package/src/processor-runner.ts

@@ -5,7 +5,7 @@ import fs from 'fs-extra'
 
 import { compressionAlgorithms } from '@grpc/grpc-js'
 import commandLineArgs from 'command-line-args'
-import { createServer } from 'nice-grpc'
+import { createServer, Server } from 'nice-grpc'
 import { errorDetailsServerMiddleware } from 'nice-grpc-error-details'
 // import { registry as niceGrpcRegistry } from 'nice-grpc-prometheus'
 import { openTelemetryServerMiddleware } from 'nice-grpc-opentelemetry'
@@ -21,6 +21,8 @@ import { ChainConfig } from './chain-config.js'
 import { setupLogger } from './logger.js'
 
 import { setupOTLP } from './otlp.js'
+import { PluginManager } from './plugin.js'
+import { ProcessConfigResponse } from '@sentio/protos'
 
 // const mergedRegistry = Registry.merge([globalRegistry, niceGrpcRegistry])
 
@@ -39,7 +41,8 @@ const optionDefinitions = [
   { name: 'chainquery-server', type: String, defaultValue: '' },
   { name: 'pricefeed-server', type: String, defaultValue: '' },
   { name: 'log-format', type: String, defaultValue: 'console' },
-  { name: 'debug', type: Boolean, defaultValue: false }
+  { name: 'debug', type: Boolean, defaultValue: false },
+  { name: 'start-action-server', type: Boolean, defaultValue: false }
 ]
 
 const options = commandLineArgs(optionDefinitions, { partial: true })
@@ -84,26 +87,41 @@ for (const [id, config] of Object.entries(chainsConfig)) {
 
 console.debug('Starting Server', options)
 
-const server = createServer({
-  'grpc.max_send_message_length': 384 * 1024 * 1024,
-  'grpc.max_receive_message_length': 384 * 1024 * 1024,
-  'grpc.default_compression_algorithm': compressionAlgorithms.gzip
-})
-  // .use(prometheusServerMiddleware())
-  .use(openTelemetryServerMiddleware())
-  .use(errorDetailsServerMiddleware)
-const baseService = new ProcessorServiceImpl(async () => {
-  const m = await import(options.target)
-  console.debug('Module loaded', m)
-  return m
-}, server.shutdown)
-const service = new FullProcessorServiceImpl(baseService)
+let server: Server
+let baseService: ProcessorServiceImpl
+
+if (options.startActionServer) {
+  const pluginManager = PluginManager.INSTANCE
+  pluginManager
+    .configure(ProcessConfigResponse.create())
+    .then(() => {
+      return pluginManager.startServer(options.port)
+    })
+    .catch((err) => {
+      console.error('Error starting action server', err)
+    })
+} else {
+  server = createServer({
+    'grpc.max_send_message_length': 384 * 1024 * 1024,
+    'grpc.max_receive_message_length': 384 * 1024 * 1024,
+    'grpc.default_compression_algorithm': compressionAlgorithms.gzip
+  })
+    // .use(prometheusServerMiddleware())
+    .use(openTelemetryServerMiddleware())
+    .use(errorDetailsServerMiddleware)
+  baseService = new ProcessorServiceImpl(async () => {
+    const m = await import(options.target)
+    console.debug('Module loaded', m)
+    return m
+  }, server.shutdown)
+  const service = new FullProcessorServiceImpl(baseService)
 
-server.add(ProcessorDefinition, service)
+  server.add(ProcessorDefinition, service)
 
-server.listen('0.0.0.0:' + options.port)
+  server.listen('0.0.0.0:' + options.port)
 
-console.log('Processor Server Started at:', options.port)
+  console.log('Processor Server Started at:', options.port)
+}
 
 const metricsPort = 4040
 const httpServer = http
@@ -154,7 +172,9 @@ process
   })
   .on('uncaughtException', (err) => {
     console.error('Uncaught Exception, please checking if await is properly used', err)
-    baseService.unhandled = err
+    if (baseService) {
+      baseService.unhandled = err
+    }
     // shutdownServers(1)
   })
   .on('unhandledRejection', (reason, p) => {
@@ -163,12 +183,14 @@ process
       return
     }
     console.error('Unhandled Rejection, please checking if await is properly', reason)
-    baseService.unhandled = reason as Error
+    if (baseService) {
+      baseService.unhandled = reason as Error
+    }
     // shutdownServers(1)
   })
 
 function shutdownServers(exitCode: number) {
-  server.forceShutdown()
+  server?.forceShutdown()
   console.log('RPC server shut down')
 
   httpServer.close(function () {