@awarns/tracing/internal/tasks/traceable-task.js

AwarNS Framework package that eases the tracing and debugging of complex task execution workflows

import { Task } from '@awarns/core/tasks';
import { syncedTracesStore } from '../stores';
import { Trace, TraceResult, TraceType } from '../entities';
import { flatten } from '@awarns/core/utils/serialization';
export class TraceableTask extends Task {
    constructor(innerTask, tracerConfig, tracesStore = syncedTracesStore) {
        // @ts-ignore
        super(innerTask.name, innerTask._taskConfig);
        this.innerTask = innerTask;
        this.tracesStore = tracesStore;
        this.sensibleData = tracerConfig && tracerConfig.outputsSensitiveData;
        // Use task method overrides
        this.checkIfCanRun = () => innerTask.checkIfCanRun();
        this.prepare = () => innerTask.prepare();
        // Override methods that can potentially be used inside onRun with decorator methods
        // @ts-ignore
        innerTask.setCancelFunction = (f) => this.setCancelFunction(f);
        // @ts-ignore
        innerTask.runAgainIn = (seconds, params) => this.runAgainIn(seconds, params);
        // @ts-ignore
        innerTask.remainingTime = () => this.remainingTime();
        // @ts-ignore
        innerTask.log = (message) => this.log(message);
    }
    async onRun(taskParams, invocationEvent) {
        const { id, name } = invocationEvent;
        const trace = new Trace(id, TraceType.TASK, this.name, TraceResult.OK, { invokedBy: name });
        let taskOutcome;
        let execError;
        try {
            // @ts-ignore
            taskOutcome = await this.innerTask.onRun(taskParams, invocationEvent);
            trace.content.emitted = taskOutcome && taskOutcome.eventName ? taskOutcome.eventName : this.outputEventNames[0];
            trace.content.outcome = this.sensibleData || !taskOutcome ? {} : flatten(taskOutcome.result);
        }
        catch (err) {
            execError = err;
            trace.result = TraceResult.ERROR;
            trace.content.message = err.stack ? err.stack : `${err}`;
        }
        trace.content.took = Date.now() - trace.timestamp.getTime();
        await this.tracesStore.insert(trace);
        this.log(`Task trace recorded: ${JSON.stringify(trace)}`);
        if (execError) {
            throw execError;
        }
        return taskOutcome;
    }
}
//# sourceMappingURL=traceable-task.js.map