refactor: Restructure workers to work with services

This commit is contained in:
Faruk AYDIN
2022-10-14 22:33:26 +02:00
parent 628f872180
commit 237ab48d33
10 changed files with 81 additions and 114 deletions

View File

@@ -1,14 +1,9 @@
import { Worker } from 'bullmq';
import redisConfig from '../config/redis';
import Flow from '../models/flow';
import logger from '../helpers/logger';
import globalVariable from '../helpers/global-variable';
import { IGlobalVariable } from '@automatisch/types';
import Execution from '../models/execution';
import Processor from '../services/processor';
import ExecutionStep from '../models/execution-step';
import Step from '../models/step';
import actionQueue from '../queues/action';
import { processAction } from '../services/action';
type JobData = {
flowId: string;
@@ -19,64 +14,21 @@ type JobData = {
export const worker = new Worker(
'action',
async (job) => {
const { flowId, stepId, executionId } = job.data as JobData;
const step = await Step.query().findById(stepId).throwIfNotFound();
const execution = await Execution.query()
.findById(executionId)
.throwIfNotFound();
const $ = await globalVariable({
flow: await Flow.query().findById(flowId).throwIfNotFound(),
app: await step.getApp(),
step: step,
connection: await step.$relatedQuery('connection'),
execution: execution,
});
const priorExecutionSteps = await ExecutionStep.query().where({
execution_id: $.execution.id,
});
const computedParameters = Processor.computeParameters(
$.step.parameters,
priorExecutionSteps
const { stepId, flowId, executionId } = await processAction(
job.data as JobData
);
const actionCommand = await step.getActionCommand();
const step = await Step.query().findById(stepId).throwIfNotFound();
const nextStep = await step.getNextStep();
$.step.parameters = computedParameters;
const actionDataItem = await actionCommand.run($);
if (!nextStep) return;
await execution.$relatedQuery('executionSteps').insertAndFetch({
stepId: $.step.id,
status: 'success',
dataIn: computedParameters,
dataOut: actionDataItem.data.raw,
});
// TODO: Add until step id logic here!
// TODO: Change job name for the action data item!
const jobName = `${$.step.appKey}-sample`;
if (!$.nextStep.id) return;
const nextStep = await Step.query()
.findById($.nextStep.id)
.throwIfNotFound();
console.log('hello world');
const variable = await globalVariable({
flow: await Flow.query().findById($.flow.id),
app: await nextStep.getApp(),
step: nextStep,
connection: await nextStep.$relatedQuery('connection'),
execution: execution,
});
const jobName = `${executionId}-${nextStep.id}`;
const jobPayload = {
$: variable,
flowId,
executionId,
stepId: nextStep.id,
};
await actionQueue.add(jobName, jobPayload);

View File

@@ -1,25 +1,43 @@
import { Worker } from 'bullmq';
import redisConfig from '../config/redis';
import Flow from '../models/flow';
import logger from '../helpers/logger';
import globalVariable from '../helpers/global-variable';
import triggerQueue from '../queues/trigger';
import { processFlow } from '../services/flow';
import Flow from '../models/flow';
export const worker = new Worker(
'flow',
async (job) => {
const flow = await Flow.query().findById(job.data.flowId).throwIfNotFound();
const { flowId } = job.data;
const flow = await Flow.query().findById(flowId).throwIfNotFound();
const triggerStep = await flow.getTriggerStep();
const triggerCommand = await triggerStep.getTriggerCommand();
const $ = await globalVariable({
flow,
connection: await triggerStep.$relatedQuery('connection'),
app: await triggerStep.getApp(),
step: triggerStep,
});
const { data, error } = await processFlow({ flowId });
await triggerCommand.run($);
for (const triggerDataItem of data) {
const jobName = `${triggerStep.id}-${triggerDataItem.meta.internalId}`;
const jobPayload = {
flowId,
stepId: triggerStep.id,
triggerDataItem,
};
await triggerQueue.add(jobName, jobPayload);
}
if (error) {
const jobName = `${triggerStep.id}-error`;
const jobPayload = {
flowId,
stepId: triggerStep.id,
error,
};
await triggerQueue.add(jobName, jobPayload);
}
},
{ connection: redisConfig }
);

View File

@@ -1,46 +1,34 @@
import { Worker } from 'bullmq';
import redisConfig from '../config/redis';
import Flow from '../models/flow';
import logger from '../helpers/logger';
import globalVariable from '../helpers/global-variable';
import { ITriggerDataItem, IGlobalVariable } from '@automatisch/types';
import Execution from '../models/execution';
import { IJSONObject, ITriggerDataItem } from '@automatisch/types';
import actionQueue from '../queues/action';
import Step from '../models/step';
import { processTrigger } from '../services/trigger';
type JobData = {
$: IGlobalVariable;
triggerDataItem: ITriggerDataItem;
flowId: string;
stepId: string;
triggerDataItem?: ITriggerDataItem;
error?: IJSONObject;
};
export const worker = new Worker(
'trigger',
async (job) => {
const { $, triggerDataItem } = job.data as JobData;
const { flowId, executionId, stepId, executionStep } = await processTrigger(
job.data as JobData
);
// check if we already process this trigger data item or not!
if (executionStep.isFailed) return;
const execution = await Execution.query().insert({
flowId: $.flow.id,
// TODO: Check the testRun logic and adjust following line!
testRun: true,
internalId: triggerDataItem.meta.internalId,
});
await execution.$relatedQuery('executionSteps').insertAndFetch({
stepId: $.step.id,
status: 'success',
dataIn: $.step.parameters,
dataOut: triggerDataItem.raw,
});
const jobName = `${$.step.appKey}-${triggerDataItem.meta.internalId}`;
const nextStep = await Step.query().findById($.nextStep.id);
const step = await Step.query().findById(stepId).throwIfNotFound();
const nextStep = await step.getNextStep();
const jobName = `${executionId}-${nextStep.id}`;
const jobPayload = {
flowId: $.flow.id,
executionId: execution.id,
flowId,
executionId,
stepId: nextStep.id,
};