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