getExecution can now return undefined if there is no active execution for that id

This commit is contained in:
Danny Martini 2024-09-19 11:01:35 +02:00
parent 9b0b0fbf57
commit 047b1d13c0
No known key found for this signature in database
2 changed files with 33 additions and 13 deletions

View file

@ -127,14 +127,20 @@ export class ActiveExecutions {
*/ */
attachWorkflowExecution(executionId: string, workflowExecution: PCancelable<IRun>) { attachWorkflowExecution(executionId: string, workflowExecution: PCancelable<IRun>) {
this.getExecution(executionId).workflowExecution = workflowExecution; const execution = this.getExecution(executionId);
if (execution) {
execution.workflowExecution = workflowExecution;
}
} }
attachResponsePromise( attachResponsePromise(
executionId: string, executionId: string,
responsePromise: IDeferredPromise<IExecuteResponsePromiseData>, responsePromise: IDeferredPromise<IExecuteResponsePromiseData>,
): void { ): void {
this.getExecution(executionId).responsePromise = responsePromise; const execution = this.getExecution(executionId);
if (execution) {
execution.responsePromise = responsePromise;
}
} }
resolveResponsePromise(executionId: string, response: IExecuteResponsePromiseData): void { resolveResponsePromise(executionId: string, response: IExecuteResponsePromiseData): void {
@ -145,23 +151,31 @@ export class ActiveExecutions {
/** Cancel the execution promise and reject its post-execution promise. */ /** Cancel the execution promise and reject its post-execution promise. */
stopExecution(executionId: string): void { stopExecution(executionId: string): void {
const execution = this.getExecution(executionId); const execution = this.getExecution(executionId);
if (execution) {
execution.workflowExecution?.cancel(); execution.workflowExecution?.cancel();
execution.postExecutePromise.reject(new ExecutionCancelledError(executionId)); execution.postExecutePromise.reject(new ExecutionCancelledError(executionId));
this.logger.debug('Execution cancelled', { executionId }); this.logger.debug('Execution cancelled', { executionId });
} }
}
/** Resolve the post-execution promise in an execution. */ /** Resolve the post-execution promise in an execution. */
finalizeExecution(executionId: string, fullRunData?: IRun) { finalizeExecution(executionId: string, fullRunData?: IRun) {
const execution = this.getExecution(executionId); const execution = this.getExecution(executionId);
if (execution) {
execution.postExecutePromise.resolve(fullRunData); execution.postExecutePromise.resolve(fullRunData);
this.logger.debug('Execution finalized', { executionId }); this.logger.debug('Execution finalized', { executionId });
} }
}
/** /**
* Returns a promise which will resolve with the data of the execution with the given id * Returns a promise which will resolve with the data of the execution with the given id
*/ */
async getPostExecutePromise(executionId: string): Promise<IRun | undefined> { async getPostExecutePromise(executionId: string): Promise<IRun | undefined> {
return await this.getExecution(executionId).postExecutePromise.promise; const execution = this.getExecution(executionId);
if (execution) {
return await execution.postExecutePromise.promise;
}
return undefined;
} }
/** /**
@ -188,11 +202,14 @@ export class ActiveExecutions {
} }
setStatus(executionId: string, status: ExecutionStatus) { setStatus(executionId: string, status: ExecutionStatus) {
this.getExecution(executionId).status = status; const execution = this.getExecution(executionId);
if (execution) {
execution.status = status;
}
} }
getStatus(executionId: string): ExecutionStatus { getStatus(executionId: string): ExecutionStatus | undefined {
return this.getExecution(executionId).status; return this.getExecution(executionId)?.status;
} }
/** Wait for all active executions to finish */ /** Wait for all active executions to finish */
@ -224,7 +241,7 @@ export class ActiveExecutions {
} }
} }
private getExecution(executionId: string): IExecutingWorkflowData { private getExecution(executionId: string): IExecutingWorkflowData | undefined {
const execution = this.activeExecutions[executionId]; const execution = this.activeExecutions[executionId];
if (!execution) { if (!execution) {
ErrorReporterProxy.error(new ExecutionNotFoundError(executionId)); ErrorReporterProxy.error(new ExecutionNotFoundError(executionId));

View file

@ -334,7 +334,10 @@ export class WorkflowRunner {
if (workflowExecution.isCanceled) { if (workflowExecution.isCanceled) {
fullRunData.finished = false; fullRunData.finished = false;
} }
fullRunData.status = this.activeExecutions.getStatus(executionId); const status = this.activeExecutions.getStatus(executionId);
if (status) {
fullRunData.status = status;
}
this.activeExecutions.finalizeExecution(executionId, fullRunData); this.activeExecutions.finalizeExecution(executionId, fullRunData);
}) })
.catch( .catch(