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

@@ -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,
}; };

View File

@@ -23,7 +23,7 @@ const executeFlow = async (
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

@@ -22,10 +22,7 @@ const globalVariable = async (
const lastInternalId = await flow?.lastInternalId(); const lastInternalId = await flow?.lastInternalId();
const trigger = await step?.getTriggerCommand(); const trigger = await step?.getTriggerCommand();
const nextStep = await flow const nextStep = await step?.getNextStep();
?.$relatedQuery('steps')
.where({ position: step.position + 1 })
.first();
const variable: IGlobalVariable = { const variable: IGlobalVariable = {
auth: { auth: {

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

@@ -102,6 +102,14 @@ class Step extends Base {
return await App.findOneByKey(this.appKey); 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() { async getTriggerCommand() {
const { appKey, key, isTrigger } = this; const { appKey, key, isTrigger } = this;
if (!isTrigger || !appKey || !key) return null; if (!isTrigger || !appKey || !key) return null;

View File

@@ -23,16 +23,6 @@ const testRun = async (options: TestRunOptions) => {
testRun: true, testRun: true,
}); });
const firstTriggerDataItem = data[0];
const { executionId, executionStep: triggerExecutionStep } =
await processTrigger({
flowId: flow.id,
stepId: triggerStep.id,
triggerDataItem: firstTriggerDataItem,
testRun: true,
});
if (triggerError) { if (triggerError) {
const { executionStep: triggerExecutionStepWithError } = const { executionStep: triggerExecutionStepWithError } =
await processTrigger({ await processTrigger({
@@ -45,6 +35,16 @@ const testRun = async (options: TestRunOptions) => {
return { executionStep: triggerExecutionStepWithError }; 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) { if (triggerStep.id === untilStep.id) {
return { executionStep: triggerExecutionStep }; return { executionStep: triggerExecutionStep };
} }
@@ -56,7 +56,7 @@ const testRun = async (options: TestRunOptions) => {
executionId, executionId,
}); });
if (actionStep.id === untilStep.id || actionExecutionStep.errorDetails) { if (actionStep.id === untilStep.id || actionExecutionStep.isFailed) {
return { executionStep: actionExecutionStep }; return { executionStep: actionExecutionStep };
} }
} }

View File

@@ -29,7 +29,7 @@ export const processTrigger = async (options: ProcessTriggerOptions) => {
const execution = await Execution.query().insert({ const execution = await Execution.query().insert({
flowId: $.flow.id, flowId: $.flow.id,
testRun, testRun,
internalId: triggerDataItem.meta.internalId, internalId: triggerDataItem?.meta.internalId,
}); });
const executionStep = await execution const executionStep = await execution

View File

@@ -1,14 +1,9 @@
import { Worker } from 'bullmq'; import { Worker } from 'bullmq';
import redisConfig from '../config/redis'; import redisConfig from '../config/redis';
import Flow from '../models/flow';
import logger from '../helpers/logger'; 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 Step from '../models/step';
import actionQueue from '../queues/action'; import actionQueue from '../queues/action';
import { processAction } from '../services/action';
type JobData = { type JobData = {
flowId: string; flowId: string;
@@ -19,64 +14,21 @@ type JobData = {
export const worker = new Worker( export const worker = new Worker(
'action', 'action',
async (job) => { async (job) => {
const { flowId, stepId, executionId } = job.data as JobData; const { stepId, flowId, executionId } = await processAction(
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 actionCommand = await step.getActionCommand(); const step = await Step.query().findById(stepId).throwIfNotFound();
const nextStep = await step.getNextStep();
$.step.parameters = computedParameters; if (!nextStep) return;
const actionDataItem = await actionCommand.run($);
await execution.$relatedQuery('executionSteps').insertAndFetch({ const jobName = `${executionId}-${nextStep.id}`;
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 jobPayload = { const jobPayload = {
$: variable, flowId,
executionId,
stepId: nextStep.id,
}; };
await actionQueue.add(jobName, jobPayload); await actionQueue.add(jobName, jobPayload);

View File

@@ -1,25 +1,43 @@
import { Worker } from 'bullmq'; import { Worker } from 'bullmq';
import redisConfig from '../config/redis'; import redisConfig from '../config/redis';
import Flow from '../models/flow';
import logger from '../helpers/logger'; 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( export const worker = new Worker(
'flow', 'flow',
async (job) => { 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 triggerStep = await flow.getTriggerStep();
const triggerCommand = await triggerStep.getTriggerCommand();
const $ = await globalVariable({ const { data, error } = await processFlow({ flowId });
flow,
connection: await triggerStep.$relatedQuery('connection'),
app: await triggerStep.getApp(),
step: triggerStep,
});
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 } { connection: redisConfig }
); );

View File

@@ -1,46 +1,34 @@
import { Worker } from 'bullmq'; import { Worker } from 'bullmq';
import redisConfig from '../config/redis'; import redisConfig from '../config/redis';
import Flow from '../models/flow';
import logger from '../helpers/logger'; import logger from '../helpers/logger';
import globalVariable from '../helpers/global-variable'; import { IJSONObject, ITriggerDataItem } from '@automatisch/types';
import { ITriggerDataItem, IGlobalVariable } from '@automatisch/types';
import Execution from '../models/execution';
import actionQueue from '../queues/action'; import actionQueue from '../queues/action';
import Step from '../models/step'; import Step from '../models/step';
import { processTrigger } from '../services/trigger';
type JobData = { type JobData = {
$: IGlobalVariable; flowId: string;
triggerDataItem: ITriggerDataItem; stepId: string;
triggerDataItem?: ITriggerDataItem;
error?: IJSONObject;
}; };
export const worker = new Worker( export const worker = new Worker(
'trigger', 'trigger',
async (job) => { 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({ const step = await Step.query().findById(stepId).throwIfNotFound();
flowId: $.flow.id, const nextStep = await step.getNextStep();
// TODO: Check the testRun logic and adjust following line! const jobName = `${executionId}-${nextStep.id}`;
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 jobPayload = { const jobPayload = {
flowId: $.flow.id, flowId,
executionId: execution.id, executionId,
stepId: nextStep.id, stepId: nextStep.id,
}; };