Merge pull request #596 from automatisch/refactor/background-jobs

refactor: Extract processor job into separate background jobs
This commit is contained in:
Ömer Faruk Aydın
2022-10-14 22:36:06 +02:00
committed by GitHub
24 changed files with 568 additions and 297 deletions

View File

@@ -33,7 +33,6 @@ export default {
async run($: IGlobalVariable) { async run($: IGlobalVariable) {
return await searchTweets($, { return await searchTweets($, {
searchTerm: $.step.parameters.searchTerm as string, searchTerm: $.step.parameters.searchTerm as string,
lastInternalId: $.flow.lastInternalId,
}); });
}, },

View File

@@ -25,7 +25,7 @@ const searchTweets = async (
do { do {
const params: IJSONObject = { const params: IJSONObject = {
query: options.searchTerm, query: options.searchTerm,
since_id: options.lastInternalId, since_id: $.execution.testRun ? null : $.flow.lastInternalId,
pagination_token: response?.data?.meta?.next_token, pagination_token: response?.data?.meta?.next_token,
}; };
@@ -62,7 +62,7 @@ const searchTweets = async (
tweets.data.push(dataItem); tweets.data.push(dataItem);
}); });
} }
} while (response.data.meta.next_token && options.lastInternalId); } while (response.data.meta.next_token && !$.execution.testRun);
tweets.data.sort((tweet, nextTweet) => { tweets.data.sort((tweet, nextTweet) => {
return (tweet.raw.id as number) - (nextTweet.raw.id as number); return (tweet.raw.id as number) - (nextTweet.raw.id as number);

View File

@@ -1,7 +1,5 @@
import Context from '../../types/express/context'; import Context from '../../types/express/context';
import Processor from '../../services/processor'; import testRun from '../../services/test-run';
// eslint-disable-next-line @typescript-eslint/no-unused-vars
import processorQueue from '../../queues/processor';
type Params = { type Params = {
input: { input: {
@@ -14,26 +12,18 @@ const executeFlow = async (
params: Params, params: Params,
context: Context context: Context
) => { ) => {
const { stepId } = params.input;
const { executionStep } = await testRun({ stepId });
const untilStep = await context.currentUser const untilStep = await context.currentUser
.$relatedQuery('steps') .$relatedQuery('steps')
.withGraphFetched('connection') .findById(stepId);
.findOne({
'steps.id': params.input.stepId,
})
.throwIfNotFound();
const flow = await untilStep.$relatedQuery('flow');
const executionStep = await new Processor(flow, {
untilStep,
testRun: true,
}).run();
await untilStep.$query().patch({ await untilStep.$query().patch({
status: 'completed', status: 'completed',
}); });
if (executionStep.errorDetails) { if (executionStep.isFailed) {
throw new Error(JSON.stringify(executionStep.errorDetails)); throw new Error(JSON.stringify(executionStep.errorDetails));
} }

View File

@@ -1,5 +1,5 @@
import Context from '../../types/express/context'; import Context from '../../types/express/context';
import processorQueue from '../../queues/processor'; import flowQueue from '../../queues/flow';
type Params = { type Params = {
input: { input: {
@@ -8,7 +8,7 @@ type Params = {
}; };
}; };
const JOB_NAME = 'processorJob'; const JOB_NAME = 'flow';
const EVERY_15_MINUTES_CRON = '*/15 * * * *'; const EVERY_15_MINUTES_CRON = '*/15 * * * *';
const updateFlowStatus = async ( const updateFlowStatus = async (
@@ -32,7 +32,7 @@ const updateFlowStatus = async (
}); });
const triggerStep = await flow.getTriggerStep(); const triggerStep = await flow.getTriggerStep();
const trigger = await triggerStep.getTrigger(); const trigger = await triggerStep.getTriggerCommand();
const interval = trigger.getInterval?.(triggerStep.parameters); const interval = trigger.getInterval?.(triggerStep.parameters);
const repeatOptions = { const repeatOptions = {
cron: interval || EVERY_15_MINUTES_CRON, cron: interval || EVERY_15_MINUTES_CRON,
@@ -43,8 +43,10 @@ const updateFlowStatus = async (
published_at: new Date().toISOString(), published_at: new Date().toISOString(),
}); });
await processorQueue.add( const jobName = `${JOB_NAME}-${flow.id}`;
JOB_NAME,
await flowQueue.add(
jobName,
{ flowId: flow.id }, { flowId: flow.id },
{ {
repeat: repeatOptions, repeat: repeatOptions,
@@ -52,10 +54,10 @@ const updateFlowStatus = async (
} }
); );
} else { } else {
const repeatableJobs = await processorQueue.getRepeatableJobs(); const repeatableJobs = await flowQueue.getRepeatableJobs();
const job = repeatableJobs.find((job) => job.id === flow.id); const job = repeatableJobs.find((job) => job.id === flow.id);
await processorQueue.removeRepeatableByKey(job.key); await flowQueue.removeRepeatableByKey(job.key);
} }
return flow; return flow;

View File

@@ -0,0 +1,43 @@
import Step from '../models/step';
import ExecutionStep from '../models/execution-step';
import get from 'lodash.get';
const variableRegExp = /({{step\.[\da-zA-Z-]+(?:\.[\da-zA-Z-]+)+}})/g;
export default function computeParameters(
parameters: Step['parameters'],
executionSteps: ExecutionStep[]
): Step['parameters'] {
const entries = Object.entries(parameters);
return entries.reduce((result, [key, value]: [string, unknown]) => {
if (typeof value === 'string') {
const parts = value.split(variableRegExp);
const computedValue = parts
.map((part: string) => {
const isVariable = part.match(variableRegExp);
if (isVariable) {
const stepIdAndKeyPath = part.replace(/{{step.|}}/g, '') as string;
const [stepId, ...keyPaths] = stepIdAndKeyPath.split('.');
const keyPath = keyPaths.join('.');
const executionStep = executionSteps.find((executionStep) => {
return executionStep.stepId === stepId;
});
const data = executionStep?.dataOut;
const dataValue = get(data, keyPath);
return dataValue;
}
return part;
})
.join('');
return {
...result,
[key]: computedValue,
};
}
return result;
}, {});
}

View File

@@ -1,13 +1,19 @@
import { ExpressAdapter } from '@bull-board/express'; import { ExpressAdapter } from '@bull-board/express';
import { createBullBoard } from '@bull-board/api'; import { createBullBoard } from '@bull-board/api';
import { BullMQAdapter } from '@bull-board/api/bullMQAdapter'; import { BullMQAdapter } from '@bull-board/api/bullMQAdapter';
import processorQueue from '../queues/processor'; import flowQueue from '../queues/flow';
import triggerQueue from '../queues/trigger';
import actionQueue from '../queues/action';
const serverAdapter = new ExpressAdapter(); const serverAdapter = new ExpressAdapter();
const createBullBoardHandler = async (serverAdapter: ExpressAdapter) => { const createBullBoardHandler = async (serverAdapter: ExpressAdapter) => {
createBullBoard({ createBullBoard({
queues: [new BullMQAdapter(processorQueue)], queues: [
new BullMQAdapter(flowQueue),
new BullMQAdapter(triggerQueue),
new BullMQAdapter(actionQueue),
],
serverAdapter: serverAdapter, serverAdapter: serverAdapter,
}); });
}; };

View File

@@ -2,6 +2,7 @@ import createHttpClient from './http-client';
import Connection from '../models/connection'; import Connection from '../models/connection';
import Flow from '../models/flow'; import Flow from '../models/flow';
import Step from '../models/step'; import Step from '../models/step';
import Execution from '../models/execution';
import { IJSONObject, IApp, IGlobalVariable } from '@automatisch/types'; import { IJSONObject, IApp, IGlobalVariable } from '@automatisch/types';
type GlobalVariableOptions = { type GlobalVariableOptions = {
@@ -9,16 +10,21 @@ type GlobalVariableOptions = {
app: IApp; app: IApp;
flow?: Flow; flow?: Flow;
step?: Step; step?: Step;
execution?: Execution;
testRun?: boolean;
}; };
const globalVariable = async ( const globalVariable = async (
options: GlobalVariableOptions options: GlobalVariableOptions
): Promise<IGlobalVariable> => { ): Promise<IGlobalVariable> => {
const { connection, app, flow, step } = options; const { connection, app, flow, step, execution, testRun = false } = options;
const lastInternalId = await flow?.lastInternalId(); const lastInternalId = await flow?.lastInternalId();
return { const trigger = await step?.getTriggerCommand();
const nextStep = await step?.getNextStep();
const variable: IGlobalVariable = {
auth: { auth: {
set: async (args: IJSONObject) => { set: async (args: IJSONObject) => {
if (connection) { if (connection) {
@@ -37,12 +43,36 @@ const globalVariable = async (
app: app, app: app,
http: createHttpClient({ baseURL: app.baseUrl }), http: createHttpClient({ baseURL: app.baseUrl }),
flow: { flow: {
id: flow?.id,
lastInternalId, lastInternalId,
}, },
step: { step: {
id: step?.id,
appKey: step?.appKey,
parameters: step?.parameters || {}, parameters: step?.parameters || {},
}, },
nextStep: {
id: nextStep?.id,
appKey: nextStep?.appKey,
parameters: nextStep?.parameters || {},
},
execution: {
id: execution?.id,
testRun,
},
}; };
if (trigger && trigger.dedupeStrategy === 'unique') {
const lastInternalIds = await flow?.lastInternalIds();
const isAlreadyProcessed = (internalId: string) => {
return lastInternalIds?.includes(internalId);
};
variable.flow.isAlreadyProcessed = isAlreadyProcessed;
}
return variable;
}; };
export default globalVariable; export default globalVariable;

View File

@@ -50,6 +50,10 @@ class ExecutionStep extends Base {
}, },
}); });
get isFailed() {
return this.status === 'failure';
}
async $afterInsert(queryContext: QueryContext) { async $afterInsert(queryContext: QueryContext) {
await super.$afterInsert(queryContext); await super.$afterInsert(queryContext);
Telemetry.executionStepCreated(this); Telemetry.executionStepCreated(this);

View File

@@ -12,6 +12,7 @@ class Flow extends Base {
active: boolean; active: boolean;
steps: Step[]; steps: Step[];
published_at: string; published_at: string;
executions?: Execution[];
static tableName = 'flows'; static tableName = 'flows';
@@ -57,6 +58,15 @@ class Flow extends Base {
return lastExecution ? (lastExecution as Execution).internalId : null; return lastExecution ? (lastExecution as Execution).internalId : null;
} }
async lastInternalIds(itemCount = 50) {
const lastExecutions = await this.$relatedQuery('executions')
.select('internal_id')
.orderBy('created_at', 'desc')
.limit(itemCount);
return lastExecutions.map((execution) => execution.internalId);
}
async $beforeUpdate( async $beforeUpdate(
opt: ModelOptions, opt: ModelOptions,
queryContext: QueryContext queryContext: QueryContext

View File

@@ -92,16 +92,43 @@ class Step extends Base {
return this.type === 'trigger'; return this.type === 'trigger';
} }
async getTrigger() { get isAction(): boolean {
if (!this.isTrigger) return null; return this.type === 'action';
}
const { appKey, key } = this; async getApp() {
if (!this.appKey) return null;
return await App.findOneByKey(this.appKey);
}
async getNextStep() {
const flow = await this.$relatedQuery('flow');
return await flow
.$relatedQuery('steps')
.findOne({ position: this.position + 1 });
}
async getTriggerCommand() {
const { appKey, key, isTrigger } = this;
if (!isTrigger || !appKey || !key) return null;
const app = await App.findOneByKey(appKey); const app = await App.findOneByKey(appKey);
const command = app.triggers.find((trigger) => trigger.key === key); const command = app.triggers.find((trigger) => trigger.key === key);
return command; return command;
} }
async getActionCommand() {
const { appKey, key, isAction } = this;
if (!isAction || !appKey || !key) return null;
const app = await App.findOneByKey(appKey);
const command = app.actions.find((action) => action.key === key);
return command;
}
} }
export default Step; export default Step;

View File

@@ -0,0 +1,25 @@
import process from 'process';
import { Queue } from 'bullmq';
import redisConfig from '../config/redis';
import logger from '../helpers/logger';
const CONNECTION_REFUSED = 'ECONNREFUSED';
const redisConnection = {
connection: redisConfig,
};
const actionQueue = new Queue('action', redisConnection);
process.on('SIGTERM', async () => {
await actionQueue.close();
});
actionQueue.on('error', (err) => {
if ((err as any).code === CONNECTION_REFUSED) {
logger.error('Make sure you have installed Redis and it is running.', err);
process.exit();
}
});
export default actionQueue;

View File

@@ -9,18 +9,18 @@ const redisConnection = {
connection: redisConfig, connection: redisConfig,
}; };
const processorQueue = new Queue('processor', redisConnection); const flowQueue = new Queue('flow', redisConnection);
const queueScheduler = new QueueScheduler('processor', redisConnection); const queueScheduler = new QueueScheduler('flow', redisConnection);
process.on('SIGTERM', async () => { process.on('SIGTERM', async () => {
await queueScheduler.close(); await queueScheduler.close();
}); });
processorQueue.on('error', (err) => { flowQueue.on('error', (err) => {
if ((err as any).code === CONNECTION_REFUSED) { if ((err as any).code === CONNECTION_REFUSED) {
logger.error('Make sure you have installed Redis and it is running.', err); logger.error('Make sure you have installed Redis and it is running.', err);
process.exit(); process.exit();
} }
}); });
export default processorQueue; export default flowQueue;

View File

@@ -0,0 +1,25 @@
import process from 'process';
import { Queue } from 'bullmq';
import redisConfig from '../config/redis';
import logger from '../helpers/logger';
const CONNECTION_REFUSED = 'ECONNREFUSED';
const redisConnection = {
connection: redisConfig,
};
const triggerQueue = new Queue('trigger', redisConnection);
process.on('SIGTERM', async () => {
await triggerQueue.close();
});
triggerQueue.on('error', (err) => {
if ((err as any).code === CONNECTION_REFUSED) {
logger.error('Make sure you have installed Redis and it is running.', err);
process.exit();
}
});
export default triggerQueue;

View File

@@ -0,0 +1,55 @@
import Step from '../models/step';
import Flow from '../models/flow';
import Execution from '../models/execution';
import ExecutionStep from '../models/execution-step';
import computeParameters from '../helpers/compute-parameters';
import globalVariable from '../helpers/global-variable';
type ProcessActionOptions = {
flowId: string;
executionId: string;
stepId: string;
};
export const processAction = async (options: ProcessActionOptions) => {
const { flowId, stepId, executionId } = options;
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 = computeParameters(
$.step.parameters,
priorExecutionSteps
);
const actionCommand = await step.getActionCommand();
$.step.parameters = computedParameters;
const actionOutput = await actionCommand.run($);
const executionStep = await execution
.$relatedQuery('executionSteps')
.insertAndFetch({
stepId: $.step.id,
status: actionOutput.error ? 'failure' : 'success',
dataIn: computedParameters,
dataOut: actionOutput.error ? null : actionOutput.data.raw,
errorDetails: actionOutput.error,
});
return { flowId, stepId, executionId, executionStep };
};

View File

@@ -0,0 +1,24 @@
import Flow from '../models/flow';
import globalVariable from '../helpers/global-variable';
type ProcessFlowOptions = {
flowId: string;
testRun?: boolean;
};
export const processFlow = async (options: ProcessFlowOptions) => {
const flow = await Flow.query().findById(options.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,
testRun: options.testRun,
});
return await triggerCommand.run($);
};

View File

@@ -1,227 +0,0 @@
import get from 'lodash.get';
import { IActionOutput } from '@automatisch/types';
import App from '../models/app';
import Flow from '../models/flow';
import Step from '../models/step';
import Execution from '../models/execution';
import ExecutionStep from '../models/execution-step';
import globalVariable from '../helpers/global-variable';
type ExecutionSteps = Record<string, ExecutionStep>;
type ProcessorOptions = {
untilStep?: Step;
testRun?: boolean;
};
class Processor {
flow: Flow;
untilStep?: Step;
testRun?: boolean;
static variableRegExp = /({{step\.[\da-zA-Z-]+(?:\.[\da-zA-Z-]+)+}})/g;
constructor(flow: Flow, processorOptions: ProcessorOptions) {
this.flow = flow;
this.untilStep = processorOptions.untilStep;
this.testRun = processorOptions.testRun;
}
async run() {
const steps = await this.flow
.$relatedQuery('steps')
.withGraphFetched('connection')
.orderBy('position', 'asc');
const triggerStep = steps.find((step) => step.type === 'trigger');
// eslint-disable-next-line @typescript-eslint/no-non-null-assertion
const initialTriggerData = await this.getInitialTriggerData(triggerStep!);
if (!initialTriggerData.error && initialTriggerData.data.length === 0) {
const lastInternalId = await this.flow.lastInternalId();
const executionData: Partial<Execution> = {
flowId: this.flow.id,
testRun: this.testRun,
};
if (lastInternalId) {
executionData.internalId = lastInternalId;
}
await Execution.query().insert(executionData);
return;
}
if (this.testRun && initialTriggerData.data.length > 0) {
initialTriggerData.data = [initialTriggerData.data[0]];
}
const executions: Execution[] = [];
for await (const data of initialTriggerData.data) {
const execution = await Execution.query().insert({
flowId: this.flow.id,
testRun: this.testRun,
internalId: data.meta.internalId as string,
});
executions.push(execution);
let previousExecutionStep: ExecutionStep;
const priorExecutionSteps: ExecutionSteps = {};
let fetchedActionData: IActionOutput = {
data: null,
};
for await (const step of steps) {
if (!step.appKey) continue;
const { appKey, key, type, parameters: rawParameters = {}, id } = step;
const isTrigger = type === 'trigger';
const app = await App.findOneByKey(appKey);
const computedParameters = Processor.computeParameters(
rawParameters,
priorExecutionSteps
);
const clonedStep = Object.assign({}, step);
clonedStep.parameters = computedParameters;
const $ = await globalVariable({
connection: step.connection,
app,
flow: this.flow,
step: clonedStep,
});
if (!isTrigger && key) {
const command = app.actions.find((action) => action.key === key);
fetchedActionData = await command.run($);
}
if (!isTrigger && fetchedActionData.error) {
await execution.$relatedQuery('executionSteps').insertAndFetch({
stepId: id,
status: 'failure',
dataIn: null,
dataOut: computedParameters,
errorDetails: fetchedActionData.error,
});
break;
}
previousExecutionStep = await execution
.$relatedQuery('executionSteps')
.insertAndFetch({
stepId: id,
status: 'success',
dataIn: isTrigger ? rawParameters : computedParameters,
dataOut: isTrigger ? data.raw : fetchedActionData.data.raw,
});
priorExecutionSteps[id] = previousExecutionStep;
if (id === this.untilStep?.id) {
break;
}
}
}
if (initialTriggerData.error) {
const executionWithError = await Execution.query().insert({
flowId: this.flow.id,
testRun: this.testRun,
});
executions.push(executionWithError);
await executionWithError.$relatedQuery('executionSteps').insertAndFetch({
stepId: triggerStep.id,
status: 'failure',
dataIn: triggerStep.parameters,
errorDetails: initialTriggerData.error,
});
}
if (!this.testRun) return;
const lastExecutionStepFromFirstExecution = await executions[0]
.$relatedQuery('executionSteps')
.orderBy('created_at', 'desc')
.first();
return lastExecutionStepFromFirstExecution;
}
async getInitialTriggerData(step: Step) {
if (!step.appKey || !step.key) return null;
const app = await App.findOneByKey(step.appKey);
const $ = await globalVariable({
connection: step.connection,
app,
flow: this.flow,
step,
});
const command = app.triggers.find((trigger) => trigger.key === step.key);
let fetchedData;
if (this.testRun) {
fetchedData = await command.testRun($);
} else {
fetchedData = await command.run($);
}
return fetchedData;
}
static computeParameters(
parameters: Step['parameters'],
executionSteps: ExecutionSteps
): Step['parameters'] {
const entries = Object.entries(parameters);
return entries.reduce((result, [key, value]: [string, unknown]) => {
if (typeof value === 'string') {
const parts = value.split(Processor.variableRegExp);
const computedValue = parts
.map((part: string) => {
const isVariable = part.match(Processor.variableRegExp);
if (isVariable) {
const stepIdAndKeyPath = part.replace(
/{{step.|}}/g,
''
) as string;
const [stepId, ...keyPaths] = stepIdAndKeyPath.split('.');
const keyPath = keyPaths.join('.');
const executionStep = executionSteps[stepId.toString() as string];
const data = executionStep?.dataOut;
const dataValue = get(data, keyPath);
return dataValue;
}
return part;
})
.join('');
return {
...result,
[key]: computedValue,
};
}
return result;
}, {});
}
}
export default Processor;

View File

@@ -0,0 +1,65 @@
import Step from '../models/step';
import { processFlow } from '../services/flow';
import { processTrigger } from '../services/trigger';
import { processAction } from '../services/action';
type TestRunOptions = {
stepId: string;
};
const testRun = async (options: TestRunOptions) => {
const untilStep = await Step.query()
.findById(options.stepId)
.throwIfNotFound();
const flow = await untilStep.$relatedQuery('flow');
const [triggerStep, ...actionSteps] = await flow
.$relatedQuery('steps')
.withGraphFetched('connection')
.orderBy('position', 'asc');
const { data, error: triggerError } = await processFlow({
flowId: flow.id,
testRun: true,
});
if (triggerError) {
const { executionStep: triggerExecutionStepWithError } =
await processTrigger({
flowId: flow.id,
stepId: triggerStep.id,
error: triggerError,
testRun: true,
});
return { executionStep: triggerExecutionStepWithError };
}
const firstTriggerDataItem = data[0];
const { executionId, executionStep: triggerExecutionStep } =
await processTrigger({
flowId: flow.id,
stepId: triggerStep.id,
triggerDataItem: firstTriggerDataItem,
testRun: true,
});
if (triggerStep.id === untilStep.id) {
return { executionStep: triggerExecutionStep };
}
for (const actionStep of actionSteps) {
const { executionStep: actionExecutionStep } = await processAction({
flowId: flow.id,
stepId: actionStep.id,
executionId,
});
if (actionStep.id === untilStep.id || actionExecutionStep.isFailed) {
return { executionStep: actionExecutionStep };
}
}
};
export default testRun;

View File

@@ -0,0 +1,46 @@
import { IJSONObject, ITriggerDataItem } from '@automatisch/types';
import Step from '../models/step';
import Flow from '../models/flow';
import Execution from '../models/execution';
import globalVariable from '../helpers/global-variable';
type ProcessTriggerOptions = {
flowId: string;
stepId: string;
triggerDataItem?: ITriggerDataItem;
error?: IJSONObject;
testRun?: boolean;
};
export const processTrigger = async (options: ProcessTriggerOptions) => {
const { flowId, stepId, triggerDataItem, error, testRun } = options;
const step = await Step.query().findById(stepId).throwIfNotFound();
const $ = await globalVariable({
flow: await Flow.query().findById(flowId).throwIfNotFound(),
app: await step.getApp(),
step: step,
connection: await step.$relatedQuery('connection'),
});
// check if we already process this trigger data item or not!
const execution = await Execution.query().insert({
flowId: $.flow.id,
testRun,
internalId: triggerDataItem?.meta.internalId,
});
const executionStep = await execution
.$relatedQuery('executionSteps')
.insertAndFetch({
stepId: $.step.id,
status: error ? 'failure' : 'success',
dataIn: $.step.parameters,
dataOut: !error ? triggerDataItem.raw : null,
errorDetails: error,
});
return { flowId, stepId, executionId: execution.id, executionStep };
};

View File

@@ -1,5 +1,7 @@
import './config/orm'; import './config/orm';
export { worker } from './workers/processor'; import './workers/flow';
import './workers/trigger';
import './workers/action';
import telemetry from './helpers/telemetry'; import telemetry from './helpers/telemetry';
telemetry.setServiceType('worker'); telemetry.setServiceType('worker');

View File

@@ -0,0 +1,51 @@
import { Worker } from 'bullmq';
import redisConfig from '../config/redis';
import logger from '../helpers/logger';
import Step from '../models/step';
import actionQueue from '../queues/action';
import { processAction } from '../services/action';
type JobData = {
flowId: string;
executionId: string;
stepId: string;
};
export const worker = new Worker(
'action',
async (job) => {
const { stepId, flowId, executionId } = await processAction(
job.data as JobData
);
const step = await Step.query().findById(stepId).throwIfNotFound();
const nextStep = await step.getNextStep();
if (!nextStep) return;
const jobName = `${executionId}-${nextStep.id}`;
const jobPayload = {
flowId,
executionId,
stepId: nextStep.id,
};
await actionQueue.add(jobName, jobPayload);
},
{ connection: redisConfig }
);
worker.on('completed', (job) => {
logger.info(`JOB ID: ${job.id} - FLOW ID: ${job.data.flowId} has started!`);
});
worker.on('failed', (job, err) => {
logger.info(
`JOB ID: ${job.id} - FLOW ID: ${job.data.flowId} has failed22 to start with ${err.message}`
);
});
process.on('SIGTERM', async () => {
await worker.close();
});

View File

@@ -0,0 +1,57 @@
import { Worker } from 'bullmq';
import redisConfig from '../config/redis';
import logger from '../helpers/logger';
import triggerQueue from '../queues/trigger';
import { processFlow } from '../services/flow';
import Flow from '../models/flow';
export const worker = new Worker(
'flow',
async (job) => {
const { flowId } = job.data;
const flow = await Flow.query().findById(flowId).throwIfNotFound();
const triggerStep = await flow.getTriggerStep();
const { data, error } = await processFlow({ flowId });
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 }
);
worker.on('completed', (job) => {
logger.info(`JOB ID: ${job.id} - FLOW ID: ${job.data.flowId} has started!`);
});
worker.on('failed', (job, err) => {
logger.info(
`JOB ID: ${job.id} - FLOW ID: ${job.data.flowId} has failed to start with ${err.message}`
);
});
process.on('SIGTERM', async () => {
await worker.close();
});

View File

@@ -1,30 +0,0 @@
import { Worker } from 'bullmq';
import Processor from '../services/processor';
import redisConfig from '../config/redis';
import Flow from '../models/flow';
import logger from '../helpers/logger';
export const worker = new Worker(
'processor',
async (job) => {
const flow = await Flow.query().findById(job.data.flowId).throwIfNotFound();
const data = await new Processor(flow, { testRun: false }).run();
return data;
},
{ connection: redisConfig }
);
worker.on('completed', (job) => {
logger.info(`JOB ID: ${job.id} - FLOW ID: ${job.data.flowId} has completed!`);
});
worker.on('failed', (job, err) => {
logger.info(
`JOB ID: ${job.id} - FLOW ID: ${job.data.flowId} has failed with ${err.message}`
);
});
process.on('SIGTERM', async () => {
await worker.close();
});

View File

@@ -0,0 +1,52 @@
import { Worker } from 'bullmq';
import redisConfig from '../config/redis';
import logger from '../helpers/logger';
import { IJSONObject, ITriggerDataItem } from '@automatisch/types';
import actionQueue from '../queues/action';
import Step from '../models/step';
import { processTrigger } from '../services/trigger';
type JobData = {
flowId: string;
stepId: string;
triggerDataItem?: ITriggerDataItem;
error?: IJSONObject;
};
export const worker = new Worker(
'trigger',
async (job) => {
const { flowId, executionId, stepId, executionStep } = await processTrigger(
job.data as JobData
);
if (executionStep.isFailed) return;
const step = await Step.query().findById(stepId).throwIfNotFound();
const nextStep = await step.getNextStep();
const jobName = `${executionId}-${nextStep.id}`;
const jobPayload = {
flowId,
executionId,
stepId: nextStep.id,
};
await actionQueue.add(jobName, jobPayload);
},
{ connection: redisConfig }
);
worker.on('completed', (job) => {
logger.info(`JOB ID: ${job.id} - FLOW ID: ${job.data.flowId} has started!`);
});
worker.on('failed', (job, err) => {
logger.info(
`JOB ID: ${job.id} - FLOW ID: ${job.data.flowId} has failed to start with ${err.message}`
);
});
process.on('SIGTERM', async () => {
await worker.close();
});

View File

@@ -204,6 +204,7 @@ export interface ITrigger {
key: string; key: string;
pollInterval: number; pollInterval: number;
description: string; description: string;
dedupeStrategy: 'greatest' | 'unique' | 'last';
substeps: ISubstep[]; substeps: ISubstep[];
getInterval(parameters: IGlobalVariable['step']['parameters']): string; getInterval(parameters: IGlobalVariable['step']['parameters']): string;
run($: IGlobalVariable): Promise<ITriggerOutput>; run($: IGlobalVariable): Promise<ITriggerOutput>;
@@ -252,12 +253,26 @@ export type IGlobalVariable = {
}; };
app: IApp; app: IApp;
http: IHttpClient; http: IHttpClient;
flow: { flow?: {
id: string;
lastInternalId: string; lastInternalId: string;
isAlreadyProcessed?: (internalId: string) => boolean;
}; };
step: { step?: {
id: string;
appKey: string;
parameters: IJSONObject; parameters: IJSONObject;
}; };
nextStep?: {
id: string;
appKey: string;
parameters: IJSONObject;
};
execution?: {
id: string;
testRun: boolean;
}
process?: (triggerDataItem: ITriggerDataItem) => Promise<void>;
}; };
declare module 'axios' { declare module 'axios' {