Dispatching error code handling

Author: patuwwy

packages/adapters/src/docker-instance-adapter.ts

@@ -185,7 +185,7 @@ IComponent {
     }
 
     // eslint-disable-next-line complexity
-    async dispatch(config: InstanceConfig, instancesServerPort: number, instanceId: string, sequenceInfo: SequenceInfo, payload: RunnerConnectInfo): Promise<void> {
+    async dispatch(config: InstanceConfig, instancesServerPort: number, instanceId: string, sequenceInfo: SequenceInfo, payload: RunnerConnectInfo): Promise<ExitCode> {
         if (!(config.type === "docker" && "container" in config)) {
             throw new Error("Docker instance adapter run with invalid runner config");
         }
@@ -241,6 +241,8 @@ IComponent {
         this.resources.containerId = containerId; // doesnt matter
 
         this.logger.trace("Container is running", containerId);
+
+        return 0;
     }
 
     async waitUntilExit(config: InstanceConfig, instanceId:string, _sequenceInfo: SequenceInfo): Promise<number> {

packages/adapters/src/kubernetes-instance-adapter.ts

@@ -95,13 +95,13 @@ IComponent {
             }
         };
     }
-    async dispatch(config: InstanceConfig, instancesServerPort: number, instanceId: string, sequenceInfo: SequenceInfo, payload: RunnerConnectInfo): Promise<void> {
+    async dispatch(config: InstanceConfig, instancesServerPort: number, instanceId: string, sequenceInfo: SequenceInfo, payload: RunnerConnectInfo): Promise<number> {
         if (config.type !== "kubernetes") {
             throw new Error(`Invalid config type for kubernetes adapter: ${config.type}`);
         }
 
         if (this.adapterConfig.quotaName && await this.kubeClient.isPodsLimitReached(this.adapterConfig.quotaName)) {
-            throw Error(RunnerExitCode.PODS_LIMIT_REACHED.toString());
+            return RunnerExitCode.PODS_LIMIT_REACHED;
         }
 
         this.limits = config.limits;
@@ -163,7 +163,7 @@ IComponent {
             // This means runner pod was unable to start. So it went from "Pending" to "Failed" state directly.
             // Return 1 which is Linux exit code for "General Error" since we are not able
             // to determine what happened exactly.
-            return;
+            return RunnerExitCode.UNCAUGHT_EXCEPTION;
         }
 
         this.logger.debug("Copy sequence files to Runner");
@@ -176,6 +176,8 @@ IComponent {
         await this.kubeClient.exec(runnerName, runnerName, ["unpack.sh", "/package"], process.stdout, this.stdErrorStream, compressedStream, 2);
 
         this.logger.debug("Copy command done");
+
+        return 0;
     }
 
     async waitUntilExit(_config: InstanceConfig, instanceId: string, _sequenceInfo: SequenceInfo): Promise<ExitCode> {

packages/adapters/src/process-instance-adapter.ts

@@ -138,7 +138,7 @@ class ProcessInstanceAdapter implements
         return this.waitUntilExit(config, instanceId, sequenceInfo);
     }
 
-    async dispatch(config: InstanceConfig, instancesServerPort: number, instanceId: string, sequenceInfo: SequenceInfo, payload: RunnerConnectInfo): Promise<void> {
+    async dispatch(config: InstanceConfig, instancesServerPort: number, instanceId: string, sequenceInfo: SequenceInfo, payload: RunnerConnectInfo): Promise<ExitCode> {
         if (config.type !== "process") {
             throw new Error("Process instance adapter run with invalid runner config");
         }
@@ -182,6 +182,8 @@ class ProcessInstanceAdapter implements
         this.runnerProcess = runnerProcess;
 
         this.logger.trace("Runner process is running", runnerProcess.pid);
+
+        return 0;
     }
 
     getRunnerInfo(): RunnerConnectInfo["system"] {

packages/host/src/lib/csi-dispatcher.ts

@@ -234,14 +234,20 @@ export class CSIDispatcher extends TypedEmitter<Events> {
 
         this.logger.debug("Dispatching...");
 
-        await instanceAdapter.dispatch(
+        const dispatchResultCode = await instanceAdapter.dispatch(
             instanceConfig,
             this.STHConfig.host.instancesServerPort,
             id,
             sequence,
             payload
         );
 
+        this.logger.warn("Dispatch result code:", dispatchResultCode);
+
+        if (dispatchResultCode !== 0) {
+            throw await mapRunnerExitCode(dispatchResultCode, sequence);
+        }
+
         this.logger.debug("Dispatched.");
         this.logger.debug("Waiting for connection...");
 

packages/types/src/lifecycle-adapters.ts

@@ -46,7 +46,7 @@ export interface ILifeCycleAdapterRun extends ILifeCycleAdapterMain {
       * @param {InstanceConfig} Runner configuration.
       * @returns {ExitCode} Runner exit code.
       */
-    dispatch(config: InstanceConfig, instancesServerPort: number, instanceId: string, sequenceInfo: SequenceInfo, payload: RunnerConnectInfo): Promise<void>;
+    dispatch(config: InstanceConfig, instancesServerPort: number, instanceId: string, sequenceInfo: SequenceInfo, payload: RunnerConnectInfo): Promise<number>;
 
     /**
       * Starts Runner - in essence does `dispatch` and then `waitUntilExit`.