mirror of
https://github.com/n8n-io/n8n.git
synced 2024-12-24 20:24:05 -08:00
feat(RabbitMQ Trigger Node): Make message acknowledgement and parallel processing configurable (#3385)
* feat(RabbitMQ Trigger Node): Make message acknowledgement and concurrent processing configurable * ⚡ Make sure that messages do not get executed multiple times * 👕 Fix lint issue * 🐛 Fix issue that for manual executions in "own" mode messages got know acknowledged * ⚡ Increment count now that console.log got removed * ⚡ Improvements * ⚡ Fix default value * ⚡ Improve display name
This commit is contained in:
parent
d7c6833dc3
commit
b851289001
|
@ -20,6 +20,7 @@ import {
|
||||||
IGetExecuteTriggerFunctions,
|
IGetExecuteTriggerFunctions,
|
||||||
INode,
|
INode,
|
||||||
INodeExecutionData,
|
INodeExecutionData,
|
||||||
|
IRun,
|
||||||
IRunExecutionData,
|
IRunExecutionData,
|
||||||
IWorkflowExecuteAdditionalData as IWorkflowExecuteAdditionalDataWorkflow,
|
IWorkflowExecuteAdditionalData as IWorkflowExecuteAdditionalDataWorkflow,
|
||||||
NodeHelpers,
|
NodeHelpers,
|
||||||
|
@ -52,6 +53,9 @@ import config from '../config';
|
||||||
import { User } from './databases/entities/User';
|
import { User } from './databases/entities/User';
|
||||||
import { whereClause } from './WorkflowHelpers';
|
import { whereClause } from './WorkflowHelpers';
|
||||||
import { WorkflowEntity } from './databases/entities/WorkflowEntity';
|
import { WorkflowEntity } from './databases/entities/WorkflowEntity';
|
||||||
|
import * as ActiveExecutions from './ActiveExecutions';
|
||||||
|
|
||||||
|
const activeExecutions = ActiveExecutions.getInstance();
|
||||||
|
|
||||||
const WEBHOOK_PROD_UNREGISTERED_HINT = `The workflow must be active for a production URL to run successfully. You can activate the workflow using the toggle in the top-right of the editor. Note that unlike test URL calls, production URL calls aren't shown on the canvas (only in the executions list)`;
|
const WEBHOOK_PROD_UNREGISTERED_HINT = `The workflow must be active for a production URL to run successfully. You can activate the workflow using the toggle in the top-right of the editor. Note that unlike test URL calls, production URL calls aren't shown on the canvas (only in the executions list)`;
|
||||||
|
|
||||||
|
@ -675,14 +679,31 @@ export class ActiveWorkflowRunner {
|
||||||
returnFunctions.emit = (
|
returnFunctions.emit = (
|
||||||
data: INodeExecutionData[][],
|
data: INodeExecutionData[][],
|
||||||
responsePromise?: IDeferredPromise<IExecuteResponsePromiseData>,
|
responsePromise?: IDeferredPromise<IExecuteResponsePromiseData>,
|
||||||
|
donePromise?: IDeferredPromise<IRun | undefined>,
|
||||||
): void => {
|
): void => {
|
||||||
// eslint-disable-next-line @typescript-eslint/restrict-template-expressions
|
// eslint-disable-next-line @typescript-eslint/restrict-template-expressions
|
||||||
Logger.debug(`Received trigger for workflow "${workflow.name}"`);
|
Logger.debug(`Received trigger for workflow "${workflow.name}"`);
|
||||||
WorkflowHelpers.saveStaticData(workflow);
|
WorkflowHelpers.saveStaticData(workflow);
|
||||||
// eslint-disable-next-line id-denylist
|
// eslint-disable-next-line id-denylist
|
||||||
this.runWorkflow(workflowData, node, data, additionalData, mode, responsePromise).catch(
|
const executePromise = this.runWorkflow(
|
||||||
(error) => console.error(error),
|
workflowData,
|
||||||
|
node,
|
||||||
|
data,
|
||||||
|
additionalData,
|
||||||
|
mode,
|
||||||
|
responsePromise,
|
||||||
);
|
);
|
||||||
|
|
||||||
|
if (donePromise) {
|
||||||
|
executePromise.then((executionId) => {
|
||||||
|
activeExecutions
|
||||||
|
.getPostExecutePromise(executionId)
|
||||||
|
.then(donePromise.resolve)
|
||||||
|
.catch(donePromise.reject);
|
||||||
|
});
|
||||||
|
} else {
|
||||||
|
executePromise.catch(console.error);
|
||||||
|
}
|
||||||
};
|
};
|
||||||
returnFunctions.emitError = async (error: Error): Promise<void> => {
|
returnFunctions.emitError = async (error: Error): Promise<void> => {
|
||||||
await this.activeWorkflows?.remove(workflowData.id.toString());
|
await this.activeWorkflows?.remove(workflowData.id.toString());
|
||||||
|
|
|
@ -627,6 +627,7 @@ export class WorkflowExecute {
|
||||||
|
|
||||||
let currentExecutionTry = '';
|
let currentExecutionTry = '';
|
||||||
let lastExecutionTry = '';
|
let lastExecutionTry = '';
|
||||||
|
let closeFunction: Promise<void> | undefined;
|
||||||
|
|
||||||
return new PCancelable(async (resolve, reject, onCancel) => {
|
return new PCancelable(async (resolve, reject, onCancel) => {
|
||||||
let gotCancel = false;
|
let gotCancel = false;
|
||||||
|
@ -811,7 +812,7 @@ export class WorkflowExecute {
|
||||||
node: executionNode.name,
|
node: executionNode.name,
|
||||||
workflowId: workflow.id,
|
workflowId: workflow.id,
|
||||||
});
|
});
|
||||||
nodeSuccessData = await workflow.runNode(
|
const runNodeData = await workflow.runNode(
|
||||||
executionData.node,
|
executionData.node,
|
||||||
executionData.data,
|
executionData.data,
|
||||||
this.runExecutionData,
|
this.runExecutionData,
|
||||||
|
@ -820,6 +821,14 @@ export class WorkflowExecute {
|
||||||
NodeExecuteFunctions,
|
NodeExecuteFunctions,
|
||||||
this.mode,
|
this.mode,
|
||||||
);
|
);
|
||||||
|
nodeSuccessData = runNodeData.data;
|
||||||
|
|
||||||
|
if (runNodeData.closeFunction) {
|
||||||
|
// Explanation why we do this can be found in n8n-workflow/Workflow.ts -> runNode
|
||||||
|
// eslint-disable-next-line @typescript-eslint/no-unsafe-call
|
||||||
|
closeFunction = runNodeData.closeFunction();
|
||||||
|
}
|
||||||
|
|
||||||
Logger.debug(`Running node "${executionNode.name}" finished successfully`, {
|
Logger.debug(`Running node "${executionNode.name}" finished successfully`, {
|
||||||
node: executionNode.name,
|
node: executionNode.name,
|
||||||
workflowId: workflow.id,
|
workflowId: workflow.id,
|
||||||
|
@ -1033,9 +1042,10 @@ export class WorkflowExecute {
|
||||||
startedAt,
|
startedAt,
|
||||||
workflow,
|
workflow,
|
||||||
new WorkflowOperationError('Workflow has been canceled or timed out!'),
|
new WorkflowOperationError('Workflow has been canceled or timed out!'),
|
||||||
|
closeFunction,
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
return this.processSuccessExecution(startedAt, workflow, executionError);
|
return this.processSuccessExecution(startedAt, workflow, executionError, closeFunction);
|
||||||
})
|
})
|
||||||
.catch(async (error) => {
|
.catch(async (error) => {
|
||||||
const fullRunData = this.getFullRunData(startedAt);
|
const fullRunData = this.getFullRunData(startedAt);
|
||||||
|
@ -1061,6 +1071,20 @@ export class WorkflowExecute {
|
||||||
},
|
},
|
||||||
);
|
);
|
||||||
|
|
||||||
|
if (closeFunction) {
|
||||||
|
try {
|
||||||
|
await closeFunction;
|
||||||
|
} catch (errorClose) {
|
||||||
|
Logger.error(
|
||||||
|
// eslint-disable-next-line @typescript-eslint/no-unsafe-member-access, @typescript-eslint/restrict-template-expressions
|
||||||
|
`There was a problem deactivating trigger of workflow "${workflow.id}": "${errorClose.message}"`,
|
||||||
|
{
|
||||||
|
workflowId: workflow.id,
|
||||||
|
},
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
return fullRunData;
|
return fullRunData;
|
||||||
});
|
});
|
||||||
|
|
||||||
|
@ -1072,6 +1096,7 @@ export class WorkflowExecute {
|
||||||
startedAt: Date,
|
startedAt: Date,
|
||||||
workflow: Workflow,
|
workflow: Workflow,
|
||||||
executionError?: ExecutionError,
|
executionError?: ExecutionError,
|
||||||
|
closeFunction?: Promise<void>,
|
||||||
): Promise<IRun> {
|
): Promise<IRun> {
|
||||||
const fullRunData = this.getFullRunData(startedAt);
|
const fullRunData = this.getFullRunData(startedAt);
|
||||||
|
|
||||||
|
@ -1106,6 +1131,20 @@ export class WorkflowExecute {
|
||||||
|
|
||||||
await this.executeHook('workflowExecuteAfter', [fullRunData, newStaticData]);
|
await this.executeHook('workflowExecuteAfter', [fullRunData, newStaticData]);
|
||||||
|
|
||||||
|
if (closeFunction) {
|
||||||
|
try {
|
||||||
|
await closeFunction;
|
||||||
|
} catch (error) {
|
||||||
|
Logger.error(
|
||||||
|
// eslint-disable-next-line @typescript-eslint/restrict-template-expressions
|
||||||
|
`There was a problem deactivating trigger of workflow "${workflow.id}": "${error.message}"`,
|
||||||
|
{
|
||||||
|
workflowId: workflow.id,
|
||||||
|
},
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
return fullRunData;
|
return fullRunData;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -68,7 +68,7 @@ export const rabbitDefaultOptions: Array<INodePropertyOptions | INodeProperties
|
||||||
],
|
],
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
displayName: 'Auto Delete',
|
displayName: 'Auto Delete Queue',
|
||||||
name: 'autoDelete',
|
name: 'autoDelete',
|
||||||
type: 'boolean',
|
type: 'boolean',
|
||||||
default: false,
|
default: false,
|
||||||
|
|
|
@ -4,9 +4,15 @@ import {
|
||||||
ITriggerFunctions,
|
ITriggerFunctions,
|
||||||
} from 'n8n-workflow';
|
} from 'n8n-workflow';
|
||||||
|
|
||||||
const amqplib = require('amqplib');
|
import * as amqplib from 'amqplib';
|
||||||
|
|
||||||
export async function rabbitmqConnect(this: IExecuteFunctions | ITriggerFunctions, options: IDataObject): Promise<any> { // tslint:disable-line:no-any
|
declare module 'amqplib' {
|
||||||
|
interface Channel {
|
||||||
|
connection: amqplib.Connection;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export async function rabbitmqConnect(this: IExecuteFunctions | ITriggerFunctions, options: IDataObject): Promise<amqplib.Channel> {
|
||||||
const credentials = await this.getCredentials('rabbitmq');
|
const credentials = await this.getCredentials('rabbitmq');
|
||||||
|
|
||||||
const credentialKeys = [
|
const credentialKeys = [
|
||||||
|
@ -44,7 +50,7 @@ export async function rabbitmqConnect(this: IExecuteFunctions | ITriggerFunction
|
||||||
reject(error);
|
reject(error);
|
||||||
});
|
});
|
||||||
|
|
||||||
const channel = await connection.createChannel().catch(console.warn);
|
const channel = await connection.createChannel().catch(console.warn) as amqplib.Channel;
|
||||||
|
|
||||||
if (options.arguments && ((options.arguments as IDataObject).argument! as IDataObject[]).length) {
|
if (options.arguments && ((options.arguments as IDataObject).argument! as IDataObject[]).length) {
|
||||||
const additionalArguments: IDataObject = {};
|
const additionalArguments: IDataObject = {};
|
||||||
|
@ -54,7 +60,6 @@ export async function rabbitmqConnect(this: IExecuteFunctions | ITriggerFunction
|
||||||
options.arguments = additionalArguments;
|
options.arguments = additionalArguments;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
resolve(channel);
|
resolve(channel);
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
reject(error);
|
reject(error);
|
||||||
|
@ -62,7 +67,7 @@ export async function rabbitmqConnect(this: IExecuteFunctions | ITriggerFunction
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
export async function rabbitmqConnectQueue(this: IExecuteFunctions | ITriggerFunctions, queue: string, options: IDataObject): Promise<any> { // tslint:disable-line:no-any
|
export async function rabbitmqConnectQueue(this: IExecuteFunctions | ITriggerFunctions, queue: string, options: IDataObject): Promise<amqplib.Channel> {
|
||||||
const channel = await rabbitmqConnect.call(this, options);
|
const channel = await rabbitmqConnect.call(this, options);
|
||||||
|
|
||||||
return new Promise(async (resolve, reject) => {
|
return new Promise(async (resolve, reject) => {
|
||||||
|
@ -75,7 +80,7 @@ export async function rabbitmqConnectQueue(this: IExecuteFunctions | ITriggerFun
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
export async function rabbitmqConnectExchange(this: IExecuteFunctions | ITriggerFunctions, exchange: string, type: string, options: IDataObject): Promise<any> { // tslint:disable-line:no-any
|
export async function rabbitmqConnectExchange(this: IExecuteFunctions | ITriggerFunctions, exchange: string, type: string, options: IDataObject): Promise<amqplib.Channel> {
|
||||||
const channel = await rabbitmqConnect.call(this, options);
|
const channel = await rabbitmqConnect.call(this, options);
|
||||||
|
|
||||||
return new Promise(async (resolve, reject) => {
|
return new Promise(async (resolve, reject) => {
|
||||||
|
@ -87,3 +92,53 @@ export async function rabbitmqConnectExchange(this: IExecuteFunctions | ITrigger
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
export class MessageTracker {
|
||||||
|
messages: number[] = [];
|
||||||
|
isClosing = false;
|
||||||
|
|
||||||
|
received(message: amqplib.ConsumeMessage) {
|
||||||
|
this.messages.push(message.fields.deliveryTag);
|
||||||
|
}
|
||||||
|
|
||||||
|
answered(message: amqplib.ConsumeMessage) {
|
||||||
|
if (this.messages.length === 0) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const index = this.messages.findIndex(value => value !== message.fields.deliveryTag);
|
||||||
|
this.messages.splice(index);
|
||||||
|
}
|
||||||
|
|
||||||
|
unansweredMessages() {
|
||||||
|
return this.messages.length;
|
||||||
|
}
|
||||||
|
|
||||||
|
async closeChannel(channel: amqplib.Channel, consumerTag: string) {
|
||||||
|
if (this.isClosing) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
this.isClosing = true;
|
||||||
|
|
||||||
|
// Do not accept any new messages
|
||||||
|
await channel.cancel(consumerTag);
|
||||||
|
|
||||||
|
let count = 0;
|
||||||
|
let unansweredMessages = this.unansweredMessages();
|
||||||
|
|
||||||
|
// Give currently executing messages max. 5 minutes to finish before
|
||||||
|
// the channel gets closed. If we would not do that, it would not be possible
|
||||||
|
// to acknowledge messages anymore for which the executions were already running
|
||||||
|
// when for example a new version of the workflow got saved. That would lead to
|
||||||
|
// them getting delivered and processed again.
|
||||||
|
while (unansweredMessages !== 0 && count++ <= 300) {
|
||||||
|
await new Promise((resolve) => {
|
||||||
|
setTimeout(resolve, 1000);
|
||||||
|
});
|
||||||
|
unansweredMessages = this.unansweredMessages();
|
||||||
|
}
|
||||||
|
|
||||||
|
await channel.close();
|
||||||
|
await channel.connection.close();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
|
@ -225,7 +225,7 @@ export class RabbitMQ implements INodeType {
|
||||||
],
|
],
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
displayName: 'Auto Delete',
|
displayName: 'Auto Delete Queue',
|
||||||
name: 'autoDelete',
|
name: 'autoDelete',
|
||||||
type: 'boolean',
|
type: 'boolean',
|
||||||
default: false,
|
default: false,
|
||||||
|
|
|
@ -1,11 +1,14 @@
|
||||||
import {
|
import {
|
||||||
|
createDeferredPromise,
|
||||||
IDataObject,
|
IDataObject,
|
||||||
INodeExecutionData,
|
INodeExecutionData,
|
||||||
INodeProperties,
|
INodeProperties,
|
||||||
INodeType,
|
INodeType,
|
||||||
INodeTypeDescription,
|
INodeTypeDescription,
|
||||||
|
IRun,
|
||||||
ITriggerFunctions,
|
ITriggerFunctions,
|
||||||
ITriggerResponse,
|
ITriggerResponse,
|
||||||
|
LoggerProxy as Logger,
|
||||||
} from 'n8n-workflow';
|
} from 'n8n-workflow';
|
||||||
|
|
||||||
import {
|
import {
|
||||||
|
@ -13,9 +16,12 @@ import {
|
||||||
} from './DefaultOptions';
|
} from './DefaultOptions';
|
||||||
|
|
||||||
import {
|
import {
|
||||||
|
MessageTracker,
|
||||||
rabbitmqConnectQueue,
|
rabbitmqConnectQueue,
|
||||||
} from './GenericFunctions';
|
} from './GenericFunctions';
|
||||||
|
|
||||||
|
import * as amqplib from 'amqplib';
|
||||||
|
|
||||||
export class RabbitMQTrigger implements INodeType {
|
export class RabbitMQTrigger implements INodeType {
|
||||||
description: INodeTypeDescription = {
|
description: INodeTypeDescription = {
|
||||||
displayName: 'RabbitMQ Trigger',
|
displayName: 'RabbitMQ Trigger',
|
||||||
|
@ -42,7 +48,7 @@ export class RabbitMQTrigger implements INodeType {
|
||||||
type: 'string',
|
type: 'string',
|
||||||
default: '',
|
default: '',
|
||||||
placeholder: 'queue-name',
|
placeholder: 'queue-name',
|
||||||
description: 'Name of the queue to publish to',
|
description: 'The name of the queue to read from',
|
||||||
},
|
},
|
||||||
|
|
||||||
{
|
{
|
||||||
|
@ -59,6 +65,30 @@ export class RabbitMQTrigger implements INodeType {
|
||||||
default: false,
|
default: false,
|
||||||
description: 'Saves the content as binary',
|
description: 'Saves the content as binary',
|
||||||
},
|
},
|
||||||
|
{
|
||||||
|
displayName: 'Delete from queue when',
|
||||||
|
name: 'acknowledge',
|
||||||
|
type: 'options',
|
||||||
|
options: [
|
||||||
|
{
|
||||||
|
name: 'Execution finishes',
|
||||||
|
value: 'executionFinishes',
|
||||||
|
description: 'After the workflow execution finished. No matter if the execution was successful or not.',
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: 'Execution finishes successfully',
|
||||||
|
value: 'executionFinishesSuccessfully',
|
||||||
|
description: 'After the workflow execution finished successfully',
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: 'Immediately',
|
||||||
|
value: 'immediately',
|
||||||
|
description: 'As soon as the message got received',
|
||||||
|
},
|
||||||
|
],
|
||||||
|
default: 'immediately',
|
||||||
|
description: 'When to acknowledge the message',
|
||||||
|
},
|
||||||
{
|
{
|
||||||
displayName: 'JSON Parse Body',
|
displayName: 'JSON Parse Body',
|
||||||
name: 'jsonParseBody',
|
name: 'jsonParseBody',
|
||||||
|
@ -87,6 +117,21 @@ export class RabbitMQTrigger implements INodeType {
|
||||||
default: false,
|
default: false,
|
||||||
description: 'Returns only the content property',
|
description: 'Returns only the content property',
|
||||||
},
|
},
|
||||||
|
// eslint-disable-next-line n8n-nodes-base/node-param-default-missing
|
||||||
|
{
|
||||||
|
displayName: 'Parallel message processing limit',
|
||||||
|
name: 'parallelMessages',
|
||||||
|
type: 'number',
|
||||||
|
default: -1,
|
||||||
|
displayOptions: {
|
||||||
|
hide: {
|
||||||
|
acknowledge: [
|
||||||
|
'immediately',
|
||||||
|
],
|
||||||
|
},
|
||||||
|
},
|
||||||
|
description: 'Max number of executions at a time. Use -1 for no limit.',
|
||||||
|
},
|
||||||
...rabbitDefaultOptions,
|
...rabbitDefaultOptions,
|
||||||
].sort((a, b) => {
|
].sort((a, b) => {
|
||||||
if ((a as INodeProperties).displayName.toLowerCase() < (b as INodeProperties).displayName.toLowerCase()) { return -1; }
|
if ((a as INodeProperties).displayName.toLowerCase() < (b as INodeProperties).displayName.toLowerCase()) { return -1; }
|
||||||
|
@ -106,9 +151,42 @@ export class RabbitMQTrigger implements INodeType {
|
||||||
|
|
||||||
const self = this;
|
const self = this;
|
||||||
|
|
||||||
|
let parallelMessages = (options.parallelMessages !== undefined && options.parallelMessages !== -1) ? parseInt(options.parallelMessages as string, 10) : -1;
|
||||||
|
|
||||||
|
if (parallelMessages === 0 || parallelMessages < -1) {
|
||||||
|
throw new Error('Parallel message processing limit must be greater than zero (or -1 for no limit)');
|
||||||
|
}
|
||||||
|
|
||||||
|
if (this.getMode() === 'manual') {
|
||||||
|
// Do only catch a single message when executing manually, else messages will leak
|
||||||
|
parallelMessages = 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
let acknowledgeMode = options.acknowledge ? options.acknowledge : 'immediately';
|
||||||
|
|
||||||
|
if (parallelMessages !== -1 && acknowledgeMode === 'immediately') {
|
||||||
|
// If parallel message limit is set, then the default mode is "executionFinishes"
|
||||||
|
// unless acknowledgeMode got set specifically. Be aware that the mode "immediately"
|
||||||
|
// can not be supported in this case.
|
||||||
|
acknowledgeMode = 'executionFinishes';
|
||||||
|
}
|
||||||
|
|
||||||
|
const messageTracker = new MessageTracker();
|
||||||
|
let consumerTag: string;
|
||||||
|
|
||||||
const startConsumer = async () => {
|
const startConsumer = async () => {
|
||||||
await channel.consume(queue, async (message: IDataObject) => {
|
if (parallelMessages !== -1) {
|
||||||
|
channel.prefetch(parallelMessages);
|
||||||
|
}
|
||||||
|
|
||||||
|
const consumerInfo = await channel.consume(queue, async (message) => {
|
||||||
if (message !== null) {
|
if (message !== null) {
|
||||||
|
|
||||||
|
try {
|
||||||
|
if (acknowledgeMode !== 'immediately') {
|
||||||
|
messageTracker.received(message);
|
||||||
|
}
|
||||||
|
|
||||||
let content: IDataObject | string = message!.content!.toString();
|
let content: IDataObject | string = message!.content!.toString();
|
||||||
|
|
||||||
const item: INodeExecutionData = {
|
const item: INodeExecutionData = {
|
||||||
|
@ -120,8 +198,8 @@ export class RabbitMQTrigger implements INodeType {
|
||||||
data: await this.helpers.prepareBinaryData(message.content),
|
data: await this.helpers.prepareBinaryData(message.content),
|
||||||
};
|
};
|
||||||
|
|
||||||
item.json = message;
|
item.json = message as unknown as IDataObject;
|
||||||
message.content = undefined;
|
message.content = undefined as unknown as Buffer;
|
||||||
} else {
|
} else {
|
||||||
if (options.jsonParseBody === true) {
|
if (options.jsonParseBody === true) {
|
||||||
content = JSON.parse(content as string);
|
content = JSON.parse(content as string);
|
||||||
|
@ -129,19 +207,61 @@ export class RabbitMQTrigger implements INodeType {
|
||||||
if (options.onlyContent === true) {
|
if (options.onlyContent === true) {
|
||||||
item.json = content as IDataObject;
|
item.json = content as IDataObject;
|
||||||
} else {
|
} else {
|
||||||
message.content = content;
|
message.content = content as unknown as Buffer;
|
||||||
item.json = message;
|
item.json = message as unknown as IDataObject;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
let responsePromise = undefined;
|
||||||
|
if (acknowledgeMode !== 'immediately') {
|
||||||
|
responsePromise = await createDeferredPromise<IRun>();
|
||||||
|
}
|
||||||
|
|
||||||
self.emit([
|
self.emit([
|
||||||
[
|
[
|
||||||
item,
|
item,
|
||||||
],
|
],
|
||||||
]);
|
], undefined, responsePromise);
|
||||||
|
|
||||||
|
if (responsePromise) {
|
||||||
|
// Acknowledge message after the execution finished
|
||||||
|
await responsePromise
|
||||||
|
.promise()
|
||||||
|
.then(async (data: IRun) => {
|
||||||
|
if (data.data.resultData.error) {
|
||||||
|
// The execution did fail
|
||||||
|
if (acknowledgeMode === 'executionFinishesSuccessfully') {
|
||||||
|
channel.nack(message);
|
||||||
|
messageTracker.answered(message);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
channel.ack(message);
|
||||||
|
messageTracker.answered(message);
|
||||||
|
});
|
||||||
|
} else {
|
||||||
|
// Acknowledge message directly
|
||||||
channel.ack(message);
|
channel.ack(message);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
} catch (error) {
|
||||||
|
const workflow = this.getWorkflow();
|
||||||
|
const node = this.getNode();
|
||||||
|
if (acknowledgeMode !== 'immediately') {
|
||||||
|
messageTracker.answered(message);
|
||||||
|
}
|
||||||
|
|
||||||
|
Logger.error(`There was a problem with the RabbitMQ Trigger node "${node.name}" in workflow "${workflow.id}": "${error.message}"`,
|
||||||
|
{
|
||||||
|
node: node.name,
|
||||||
|
workflowId: workflow.id,
|
||||||
|
},
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
});
|
});
|
||||||
|
consumerTag = consumerInfo.consumerTag;
|
||||||
};
|
};
|
||||||
|
|
||||||
startConsumer();
|
startConsumer();
|
||||||
|
@ -149,23 +269,23 @@ export class RabbitMQTrigger implements INodeType {
|
||||||
// The "closeFunction" function gets called by n8n whenever
|
// The "closeFunction" function gets called by n8n whenever
|
||||||
// the workflow gets deactivated and can so clean up.
|
// the workflow gets deactivated and can so clean up.
|
||||||
async function closeFunction() {
|
async function closeFunction() {
|
||||||
await channel.close();
|
|
||||||
await channel.connection.close();
|
|
||||||
}
|
|
||||||
|
|
||||||
// The "manualTriggerFunction" function gets called by n8n
|
try {
|
||||||
// when a user is in the workflow editor and starts the
|
return messageTracker.closeChannel(channel, consumerTag);
|
||||||
// workflow manually. So the function has to make sure that
|
} catch(error) {
|
||||||
// the emit() gets called with similar data like when it
|
const workflow = self.getWorkflow();
|
||||||
// would trigger by itself so that the user knows what data
|
const node = self.getNode();
|
||||||
// to expect.
|
Logger.error(`There was a problem closing the RabbitMQ Trigger node connection "${node.name}" in workflow "${workflow.id}": "${error.message}"`,
|
||||||
async function manualTriggerFunction() {
|
{
|
||||||
startConsumer();
|
node: node.name,
|
||||||
|
workflowId: workflow.id,
|
||||||
|
},
|
||||||
|
);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return {
|
return {
|
||||||
closeFunction,
|
closeFunction,
|
||||||
manualTriggerFunction,
|
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -685,6 +685,7 @@
|
||||||
]
|
]
|
||||||
},
|
},
|
||||||
"devDependencies": {
|
"devDependencies": {
|
||||||
|
"@types/amqplib": "^0.8.2",
|
||||||
"@types/aws4": "^1.5.1",
|
"@types/aws4": "^1.5.1",
|
||||||
"@types/basic-auth": "^1.1.2",
|
"@types/basic-auth": "^1.1.2",
|
||||||
"@types/cheerio": "^0.22.15",
|
"@types/cheerio": "^0.22.15",
|
||||||
|
|
|
@ -350,6 +350,10 @@ export interface IGetExecuteTriggerFunctions {
|
||||||
): ITriggerFunctions;
|
): ITriggerFunctions;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
export interface IRunNodeResponse {
|
||||||
|
data: INodeExecutionData[][] | null | undefined;
|
||||||
|
closeFunction?: () => Promise<void>;
|
||||||
|
}
|
||||||
export interface IGetExecuteFunctions {
|
export interface IGetExecuteFunctions {
|
||||||
(
|
(
|
||||||
workflow: Workflow,
|
workflow: Workflow,
|
||||||
|
@ -690,6 +694,7 @@ export interface ITriggerFunctions {
|
||||||
emit(
|
emit(
|
||||||
data: INodeExecutionData[][],
|
data: INodeExecutionData[][],
|
||||||
responsePromise?: IDeferredPromise<IExecuteResponsePromiseData>,
|
responsePromise?: IDeferredPromise<IExecuteResponsePromiseData>,
|
||||||
|
donePromise?: IDeferredPromise<IRun>,
|
||||||
): void;
|
): void;
|
||||||
emitError(error: Error, responsePromise?: IDeferredPromise<IExecuteResponsePromiseData>): void;
|
emitError(error: Error, responsePromise?: IDeferredPromise<IExecuteResponsePromiseData>): void;
|
||||||
getCredentials(type: string): Promise<ICredentialDataDecryptedObject>;
|
getCredentials(type: string): Promise<ICredentialDataDecryptedObject>;
|
||||||
|
|
|
@ -46,7 +46,14 @@ import {
|
||||||
WorkflowExecuteMode,
|
WorkflowExecuteMode,
|
||||||
} from '.';
|
} from '.';
|
||||||
|
|
||||||
import { IConnection, IDataObject, IConnectedNode, IObservableObject } from './Interfaces';
|
import {
|
||||||
|
IConnection,
|
||||||
|
IDataObject,
|
||||||
|
IConnectedNode,
|
||||||
|
IObservableObject,
|
||||||
|
IRun,
|
||||||
|
IRunNodeResponse,
|
||||||
|
} from './Interfaces';
|
||||||
|
|
||||||
function dedupe<T>(arr: T[]): T[] {
|
function dedupe<T>(arr: T[]): T[] {
|
||||||
return [...new Set(arr)];
|
return [...new Set(arr)];
|
||||||
|
@ -1040,6 +1047,7 @@ export class Workflow {
|
||||||
(
|
(
|
||||||
data: INodeExecutionData[][],
|
data: INodeExecutionData[][],
|
||||||
responsePromise?: IDeferredPromise<IExecuteResponsePromiseData>,
|
responsePromise?: IDeferredPromise<IExecuteResponsePromiseData>,
|
||||||
|
donePromise?: IDeferredPromise<IRun>,
|
||||||
) => {
|
) => {
|
||||||
additionalData.hooks!.hookFunctions.sendResponse = [
|
additionalData.hooks!.hookFunctions.sendResponse = [
|
||||||
async (response: IExecuteResponsePromiseData): Promise<void> => {
|
async (response: IExecuteResponsePromiseData): Promise<void> => {
|
||||||
|
@ -1049,6 +1057,14 @@ export class Workflow {
|
||||||
},
|
},
|
||||||
];
|
];
|
||||||
|
|
||||||
|
if (donePromise) {
|
||||||
|
additionalData.hooks!.hookFunctions.workflowExecuteAfter?.unshift(
|
||||||
|
async (runData: IRun): Promise<void> => {
|
||||||
|
return donePromise.resolve(runData);
|
||||||
|
},
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
resolveEmit(data);
|
resolveEmit(data);
|
||||||
}
|
}
|
||||||
)(resolve);
|
)(resolve);
|
||||||
|
@ -1159,18 +1175,18 @@ export class Workflow {
|
||||||
additionalData: IWorkflowExecuteAdditionalData,
|
additionalData: IWorkflowExecuteAdditionalData,
|
||||||
nodeExecuteFunctions: INodeExecuteFunctions,
|
nodeExecuteFunctions: INodeExecuteFunctions,
|
||||||
mode: WorkflowExecuteMode,
|
mode: WorkflowExecuteMode,
|
||||||
): Promise<INodeExecutionData[][] | null | undefined> {
|
): Promise<IRunNodeResponse> {
|
||||||
if (node.disabled === true) {
|
if (node.disabled === true) {
|
||||||
// If node is disabled simply pass the data through
|
// If node is disabled simply pass the data through
|
||||||
// return NodeRunHelpers.
|
// return NodeRunHelpers.
|
||||||
if (inputData.hasOwnProperty('main') && inputData.main.length > 0) {
|
if (inputData.hasOwnProperty('main') && inputData.main.length > 0) {
|
||||||
// If the node is disabled simply return the data from the first main input
|
// If the node is disabled simply return the data from the first main input
|
||||||
if (inputData.main[0] === null) {
|
if (inputData.main[0] === null) {
|
||||||
return undefined;
|
return { data: undefined };
|
||||||
}
|
}
|
||||||
return [inputData.main[0]];
|
return { data: [inputData.main[0]] };
|
||||||
}
|
}
|
||||||
return undefined;
|
return { data: undefined };
|
||||||
}
|
}
|
||||||
|
|
||||||
const nodeType = this.nodeTypes.getByNameAndVersion(node.type, node.typeVersion);
|
const nodeType = this.nodeTypes.getByNameAndVersion(node.type, node.typeVersion);
|
||||||
|
@ -1195,7 +1211,7 @@ export class Workflow {
|
||||||
|
|
||||||
if (connectionInputData.length === 0) {
|
if (connectionInputData.length === 0) {
|
||||||
// No data for node so return
|
// No data for node so return
|
||||||
return undefined;
|
return { data: undefined };
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -1245,7 +1261,7 @@ export class Workflow {
|
||||||
}
|
}
|
||||||
|
|
||||||
if (returnPromises.length === 0) {
|
if (returnPromises.length === 0) {
|
||||||
return null;
|
return { data: null };
|
||||||
}
|
}
|
||||||
|
|
||||||
let promiseResults;
|
let promiseResults;
|
||||||
|
@ -1256,7 +1272,7 @@ export class Workflow {
|
||||||
}
|
}
|
||||||
|
|
||||||
if (promiseResults) {
|
if (promiseResults) {
|
||||||
return [promiseResults];
|
return { data: [promiseResults] };
|
||||||
}
|
}
|
||||||
} else if (nodeType.execute) {
|
} else if (nodeType.execute) {
|
||||||
const thisArgs = nodeExecuteFunctions.getExecuteFunctions(
|
const thisArgs = nodeExecuteFunctions.getExecuteFunctions(
|
||||||
|
@ -1269,7 +1285,7 @@ export class Workflow {
|
||||||
additionalData,
|
additionalData,
|
||||||
mode,
|
mode,
|
||||||
);
|
);
|
||||||
return nodeType.execute.call(thisArgs);
|
return { data: await nodeType.execute.call(thisArgs) };
|
||||||
} else if (nodeType.poll) {
|
} else if (nodeType.poll) {
|
||||||
if (mode === 'manual') {
|
if (mode === 'manual') {
|
||||||
// In manual mode run the poll function
|
// In manual mode run the poll function
|
||||||
|
@ -1280,10 +1296,10 @@ export class Workflow {
|
||||||
mode,
|
mode,
|
||||||
'manual',
|
'manual',
|
||||||
);
|
);
|
||||||
return nodeType.poll.call(thisArgs);
|
return { data: await nodeType.poll.call(thisArgs) };
|
||||||
}
|
}
|
||||||
// In any other mode pass data through as it already contains the result of the poll
|
// In any other mode pass data through as it already contains the result of the poll
|
||||||
return inputData.main as INodeExecutionData[][];
|
return { data: inputData.main as INodeExecutionData[][] };
|
||||||
} else if (nodeType.trigger) {
|
} else if (nodeType.trigger) {
|
||||||
if (mode === 'manual') {
|
if (mode === 'manual') {
|
||||||
// In manual mode start the trigger
|
// In manual mode start the trigger
|
||||||
|
@ -1296,7 +1312,7 @@ export class Workflow {
|
||||||
);
|
);
|
||||||
|
|
||||||
if (triggerResponse === undefined) {
|
if (triggerResponse === undefined) {
|
||||||
return null;
|
return { data: null };
|
||||||
}
|
}
|
||||||
|
|
||||||
if (triggerResponse.manualTriggerFunction !== undefined) {
|
if (triggerResponse.manualTriggerFunction !== undefined) {
|
||||||
|
@ -1306,22 +1322,27 @@ export class Workflow {
|
||||||
|
|
||||||
const response = await triggerResponse.manualTriggerResponse!;
|
const response = await triggerResponse.manualTriggerResponse!;
|
||||||
|
|
||||||
// And then close it again after it did execute
|
let closeFunction;
|
||||||
if (triggerResponse.closeFunction) {
|
if (triggerResponse.closeFunction) {
|
||||||
await triggerResponse.closeFunction();
|
// In manual mode we return the trigger closeFunction. That allows it to be called directly
|
||||||
|
// but we do not have to wait for it to finish. That is important for things like queue-nodes.
|
||||||
|
// There the full close will may be delayed till a message gets acknowledged after the execution.
|
||||||
|
// If we would not be able to wait for it to close would it cause problems with "own" mode as the
|
||||||
|
// process would be killed directly after it and so the acknowledge would not have been finished yet.
|
||||||
|
closeFunction = triggerResponse.closeFunction;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (response.length === 0) {
|
if (response.length === 0) {
|
||||||
return null;
|
return { data: null, closeFunction };
|
||||||
}
|
}
|
||||||
|
|
||||||
return response;
|
return { data: response, closeFunction };
|
||||||
}
|
}
|
||||||
// For trigger nodes in any mode except "manual" do we simply pass the data through
|
// For trigger nodes in any mode except "manual" do we simply pass the data through
|
||||||
return inputData.main as INodeExecutionData[][];
|
return { data: inputData.main as INodeExecutionData[][] };
|
||||||
} else if (nodeType.webhook) {
|
} else if (nodeType.webhook) {
|
||||||
// For webhook nodes always simply pass the data through
|
// For webhook nodes always simply pass the data through
|
||||||
return inputData.main as INodeExecutionData[][];
|
return { data: inputData.main as INodeExecutionData[][] };
|
||||||
} else {
|
} else {
|
||||||
// For nodes which have routing information on properties
|
// For nodes which have routing information on properties
|
||||||
|
|
||||||
|
@ -1334,9 +1355,11 @@ export class Workflow {
|
||||||
mode,
|
mode,
|
||||||
);
|
);
|
||||||
|
|
||||||
return routingNode.runNode(inputData, runIndex, nodeType, nodeExecuteFunctions);
|
return {
|
||||||
|
data: await routingNode.runNode(inputData, runIndex, nodeType, nodeExecuteFunctions),
|
||||||
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
return null;
|
return { data: null };
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue