mirror of
https://github.com/Abdulazizzn/n8n-enterprise-unlocked.git
synced 2025-12-17 10:02:05 +00:00
refactor(RabbitMQ Trigger Node): Improve type-safety, add tests, and fix issues with manual triggers (#10663)
This commit is contained in:
committed by
GitHub
parent
a5a92ec8b1
commit
e50f0e6a4e
@@ -1,6 +1,5 @@
|
||||
/* eslint-disable n8n-nodes-base/node-filename-against-convention */
|
||||
import * as amqplib from 'amqplib';
|
||||
import type { Options } from 'amqplib';
|
||||
import type * as amqplib from 'amqplib';
|
||||
import type {
|
||||
IExecuteFunctions,
|
||||
ICredentialsDecrypted,
|
||||
@@ -14,8 +13,13 @@ import type {
|
||||
} from 'n8n-workflow';
|
||||
import { NodeApiError, NodeConnectionType, NodeOperationError } from 'n8n-workflow';
|
||||
|
||||
import { rabbitmqConnectExchange, rabbitmqConnectQueue } from './GenericFunctions';
|
||||
import { formatPrivateKey } from '@utils/utilities';
|
||||
import {
|
||||
parsePublishArguments,
|
||||
rabbitmqConnect,
|
||||
rabbitmqConnectExchange,
|
||||
rabbitmqConnectQueue,
|
||||
} from './GenericFunctions';
|
||||
import type { Options, RabbitMQCredentials } from './types';
|
||||
|
||||
export class RabbitMQ implements INodeType {
|
||||
description: INodeTypeDescription = {
|
||||
@@ -363,38 +367,8 @@ export class RabbitMQ implements INodeType {
|
||||
this: ICredentialTestFunctions,
|
||||
credential: ICredentialsDecrypted,
|
||||
): Promise<INodeCredentialTestResult> {
|
||||
const credentials = credential.data as IDataObject;
|
||||
try {
|
||||
const credentialKeys = ['hostname', 'port', 'username', 'password', 'vhost'];
|
||||
|
||||
const credentialData: IDataObject = {};
|
||||
credentialKeys.forEach((key) => {
|
||||
credentialData[key] = credentials[key] === '' ? undefined : credentials[key];
|
||||
});
|
||||
|
||||
const optsData: IDataObject = {};
|
||||
if (credentials.ssl === true) {
|
||||
credentialData.protocol = 'amqps';
|
||||
|
||||
optsData.ca =
|
||||
credentials.ca === ''
|
||||
? undefined
|
||||
: [Buffer.from(formatPrivateKey(credentials.ca as string))];
|
||||
if (credentials.passwordless === true) {
|
||||
optsData.cert =
|
||||
credentials.cert === ''
|
||||
? undefined
|
||||
: Buffer.from(formatPrivateKey(credentials.cert as string));
|
||||
optsData.key =
|
||||
credentials.key === ''
|
||||
? undefined
|
||||
: Buffer.from(formatPrivateKey(credentials.key as string));
|
||||
optsData.passphrase =
|
||||
credentials.passphrase === '' ? undefined : credentials.passphrase;
|
||||
optsData.credentials = amqplib.credentials.external();
|
||||
}
|
||||
}
|
||||
const connection = await amqplib.connect(credentialData, optsData);
|
||||
const connection = await rabbitmqConnect(credential.data as RabbitMQCredentials);
|
||||
await connection.close();
|
||||
} catch (error) {
|
||||
return {
|
||||
@@ -411,7 +385,7 @@ export class RabbitMQ implements INodeType {
|
||||
};
|
||||
|
||||
async execute(this: IExecuteFunctions): Promise<INodeExecutionData[][]> {
|
||||
let channel, options: IDataObject;
|
||||
let channel: amqplib.Channel | undefined;
|
||||
try {
|
||||
const items = this.getInputData();
|
||||
const operation = this.getNodeParameter('operation', 0);
|
||||
@@ -424,7 +398,7 @@ export class RabbitMQ implements INodeType {
|
||||
if (mode === 'queue') {
|
||||
const queue = this.getNodeParameter('queue', 0) as string;
|
||||
|
||||
options = this.getNodeParameter('options', 0, {});
|
||||
const options = this.getNodeParameter('options', 0, {}) as Options;
|
||||
|
||||
channel = await rabbitmqConnectQueue.call(this, queue, options);
|
||||
|
||||
@@ -457,7 +431,7 @@ export class RabbitMQ implements INodeType {
|
||||
queuePromises.push(
|
||||
channel.sendToQueue(queue, Buffer.from(message), {
|
||||
headers,
|
||||
...(options.arguments ? (options.arguments as Options.Publish) : {}),
|
||||
...parsePublishArguments(options),
|
||||
}),
|
||||
);
|
||||
}
|
||||
@@ -492,12 +466,11 @@ export class RabbitMQ implements INodeType {
|
||||
await channel.connection.close();
|
||||
} else if (mode === 'exchange') {
|
||||
const exchange = this.getNodeParameter('exchange', 0) as string;
|
||||
const type = this.getNodeParameter('exchangeType', 0) as string;
|
||||
const routingKey = this.getNodeParameter('routingKey', 0) as string;
|
||||
|
||||
options = this.getNodeParameter('options', 0, {});
|
||||
const options = this.getNodeParameter('options', 0, {}) as Options;
|
||||
|
||||
channel = await rabbitmqConnectExchange.call(this, exchange, type, options);
|
||||
channel = await rabbitmqConnectExchange.call(this, exchange, options);
|
||||
|
||||
const sendInputData = this.getNodeParameter('sendInputData', 0) as boolean;
|
||||
|
||||
@@ -529,7 +502,7 @@ export class RabbitMQ implements INodeType {
|
||||
exchangePromises.push(
|
||||
channel.publish(exchange, routingKey, Buffer.from(message), {
|
||||
headers,
|
||||
...(options.arguments ? (options.arguments as Options.Publish) : {}),
|
||||
...parsePublishArguments(options),
|
||||
}),
|
||||
);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user