refactor: Restructure workers to work with services
This commit is contained in:
@@ -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);
|
||||
|
@@ -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 }
|
||||
);
|
||||
|
@@ -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,
|
||||
};
|
||||
|
||||
|
Reference in New Issue
Block a user