@sentio/runtime 2.59.0-rc.31 → 2.59.0-rc.33

package/lib/index.js

@@ -28,7 +28,7 @@ import {
   providerMetrics,
   recordRuntimeInfo,
   timeoutError
-} from "./chunk-X2VTMTYL.js";
+} from "./chunk-QWKQ7U4H.js";
 import {
   Plugin,
   PluginManager

package/lib/processor-runner.js

@@ -44,7 +44,7 @@ import {
   require_lodash,
   require_src,
   trace
-} from "./chunk-X2VTMTYL.js";
+} from "./chunk-QWKQ7U4H.js";
 import {
   ExecutionConfig,
   HandlerType,
@@ -26328,6 +26328,7 @@ var ServiceManager = class extends ProcessorServiceImpl {
     for await (const request3 of requests) {
       try {
         if (request3.binding) {
+          lastBinding = request3.binding;
           process_binding_count.add(1);
           if (request3.binding.handlerType === HandlerType.UNKNOWN) {
             subject.next({
@@ -26337,13 +26338,7 @@ var ServiceManager = class extends ProcessorServiceImpl {
             continue;
           }
           if (this.enablePartition) {
-            PluginManager.INSTANCE.partition(request3.binding).then((partitions) => {
-              subject.next({
-                processId: request3.processId,
-                partitions
-              });
-            });
-            lastBinding = request3.binding;
+            this.doPartition(request3.processId, request3.binding, subject);
           } else {
             this.doProcess(request3.processId, request3.binding, subject);
           }
@@ -26367,10 +26362,33 @@ var ServiceManager = class extends ProcessorServiceImpl {
       }
     }
   }
+  async doPartition(processId, binding, subject) {
+    if (!this.pool) {
+      await this.initPool();
+    }
+    const dbContext = this.contexts.new(processId, subject);
+    const start = Date.now();
+    this.process(binding, processId, dbContext, true).then(async (result) => {
+      console.debug("partition", processId, "finished, took:", Date.now() - start);
+      subject.next({
+        result,
+        processId
+      });
+    }).catch((e) => {
+      dbContext.error(processId, e);
+      process_binding_error.add(1);
+      console.error("partition", processId, "failed, took:", Date.now() - start);
+    }).finally(() => {
+      const cost = Date.now() - start;
+      process_binding_time.add(cost);
+      this.contexts.delete(processId);
+    });
+  }
   doProcess(processId, binding, subject) {
     const dbContext = this.contexts.new(processId, subject);
     const start = Date.now();
-    this.process(binding, dbContext).then(async (result) => {
+    this.process(binding, processId, dbContext, false).then(async (result) => {
+      console.debug("process", processId, "finished, took:", Date.now() - start);
       subject.next({
         result,
         processId
@@ -26378,18 +26396,19 @@ var ServiceManager = class extends ProcessorServiceImpl {
     }).catch((e) => {
       dbContext.error(processId, e);
       process_binding_error.add(1);
+      console.error("process", processId, "failed, took:", Date.now() - start);
     }).finally(() => {
       const cost = Date.now() - start;
       process_binding_time.add(cost);
       this.contexts.delete(processId);
     });
   }
-  async process(request3, dbContext) {
+  async process(request3, processId, dbContext, partition = false) {
     if (!this.pool) {
       await this.initPool();
     }
     return this.pool.run(
-      { request: request3, workerPort: dbContext?.workerPort },
+      { request: request3, workerPort: dbContext?.workerPort, partition, processId },
       { transferList: dbContext?.workerPort ? [dbContext?.workerPort] : [] }
     );
   }
@@ -26418,7 +26437,11 @@ var Contexts = class {
     return this.contexts.get(processId);
   }
   new(processId, subject) {
-    const context2 = new ChannelStoreContext(subject, processId);
+    let context2 = this.get(processId);
+    if (context2) {
+      return context2;
+    }
+    context2 = new ChannelStoreContext(subject, processId);
     this.contexts.set(processId, context2);
     return context2;
   }
@@ -26496,7 +26519,9 @@ var optionDefinitions = [
   { name: "debug", type: Boolean, defaultValue: false },
   { name: "otlp-debug", type: Boolean, defaultValue: false },
   { name: "start-action-server", type: Boolean, defaultValue: false },
-  { name: "worker", type: Number, defaultValue: workerNum }
+  { name: "worker", type: Number, defaultValue: workerNum },
+  { name: "process-timeout", type: Number, defaultValue: 60 }
+  // 60 seconds
 ];
 var options = command_line_args_default(optionDefinitions, { partial: true });
 var logLevel = process.env["LOG_LEVEL"]?.toLowerCase();