fix(AMQP Sender Node): Node hangs forever on disconnect (#10026)

This commit is contained in:
Michael Kret
2024-07-12 17:56:40 +03:00
committed by GitHub
parent f78f4ea349
commit 27410ab2af
2 changed files with 95 additions and 53 deletions

View File

@@ -12,6 +12,7 @@ export class Amqp implements ICredentialType {
displayName: 'Hostname', displayName: 'Hostname',
name: 'hostname', name: 'hostname',
type: 'string', type: 'string',
placeholder: 'e.g. localhost',
default: '', default: '',
}, },
{ {
@@ -24,12 +25,14 @@ export class Amqp implements ICredentialType {
displayName: 'User', displayName: 'User',
name: 'username', name: 'username',
type: 'string', type: 'string',
placeholder: 'e.g. guest',
default: '', default: '',
}, },
{ {
displayName: 'Password', displayName: 'Password',
name: 'password', name: 'password',
type: 'string', type: 'string',
placeholder: 'e.g. guest',
typeOptions: { typeOptions: {
password: true, password: true,
}, },
@@ -39,8 +42,9 @@ export class Amqp implements ICredentialType {
displayName: 'Transport Type', displayName: 'Transport Type',
name: 'transportType', name: 'transportType',
type: 'string', type: 'string',
placeholder: 'e.g. tcp',
default: '', default: '',
description: 'Optional Transport Type to use. Either tcp or tls.', hint: 'Optional transport type to use, either tcp or tls',
}, },
]; ];
} }

View File

@@ -1,4 +1,4 @@
import type { Connection, ContainerOptions, Dictionary, EventContext } from 'rhea'; import type { Connection, ContainerOptions, Dictionary, EventContext, Sender } from 'rhea';
import { create_container } from 'rhea'; import { create_container } from 'rhea';
import type { import type {
@@ -14,6 +14,46 @@ import type {
} from 'n8n-workflow'; } from 'n8n-workflow';
import { NodeOperationError } from 'n8n-workflow'; import { NodeOperationError } from 'n8n-workflow';
async function checkIfCredentialsValid(
credentials: IDataObject,
): Promise<INodeCredentialTestResult> {
const connectOptions: ContainerOptions = {
reconnect: false,
host: credentials.hostname as string,
hostname: credentials.hostname as string,
port: credentials.port as number,
username: credentials.username ? (credentials.username as string) : undefined,
password: credentials.password ? (credentials.password as string) : undefined,
transport: credentials.transportType ? (credentials.transportType as string) : undefined,
};
let conn: Connection | undefined = undefined;
try {
const container = create_container();
await new Promise<void>((resolve, reject) => {
container.on('connection_open', function (_context: EventContext) {
resolve();
});
container.on('disconnected', function (context: EventContext) {
reject(context.error ?? new Error('unknown error'));
});
conn = container.connect(connectOptions);
});
} catch (error) {
return {
status: 'Error',
message: (error as Error).message,
};
} finally {
if (conn) (conn as Connection).close();
}
return {
status: 'OK',
message: 'Connection successful!',
};
}
export class Amqp implements INodeType { export class Amqp implements INodeType {
description: INodeTypeDescription = { description: INodeTypeDescription = {
displayName: 'AMQP Sender', displayName: 'AMQP Sender',
@@ -40,7 +80,7 @@ export class Amqp implements INodeType {
name: 'sink', name: 'sink',
type: 'string', type: 'string',
default: '', default: '',
placeholder: 'topic://sourcename.something', placeholder: 'e.g. topic://sourcename.something',
description: 'Name of the queue of topic to publish to', description: 'Name of the queue of topic to publish to',
}, },
// Header Parameters // Header Parameters
@@ -106,49 +146,27 @@ export class Amqp implements INodeType {
credential: ICredentialsDecrypted, credential: ICredentialsDecrypted,
): Promise<INodeCredentialTestResult> { ): Promise<INodeCredentialTestResult> {
const credentials = credential.data as ICredentialDataDecryptedObject; const credentials = credential.data as ICredentialDataDecryptedObject;
const connectOptions: ContainerOptions = { return await checkIfCredentialsValid(credentials);
reconnect: false,
host: credentials.hostname as string,
hostname: credentials.hostname as string,
port: credentials.port as number,
username: credentials.username ? (credentials.username as string) : undefined,
password: credentials.password ? (credentials.password as string) : undefined,
transport: credentials.transportType ? (credentials.transportType as string) : undefined,
};
let conn: Connection | undefined = undefined;
try {
const container = create_container();
await new Promise<void>((resolve, reject) => {
container.on('connection_open', function (_contex: EventContext) {
resolve();
});
container.on('disconnected', function (context: EventContext) {
reject(context.error ?? new Error('unknown error'));
});
conn = container.connect(connectOptions);
});
} catch (error) {
return {
status: 'Error',
message: (error as Error).message,
};
} finally {
if (conn) (conn as Connection).close();
}
return {
status: 'OK',
message: 'Connection successful!',
};
}, },
}, },
}; };
async execute(this: IExecuteFunctions): Promise<INodeExecutionData[][]> { async execute(this: IExecuteFunctions): Promise<INodeExecutionData[][]> {
const container = create_container();
let connection: Connection | undefined = undefined;
let sender: Sender | undefined = undefined;
try { try {
const credentials = await this.getCredentials('amqp'); const credentials = await this.getCredentials('amqp');
// check if credentials are valid to avoid unnecessary reconnects
const credentialsTestResult = await checkIfCredentialsValid(credentials);
if (credentialsTestResult.status === 'Error') {
throw new NodeOperationError(this.getNode(), credentialsTestResult.message, {
description: 'Check your credentials and try again',
});
}
const sink = this.getNodeParameter('sink', 0, '') as string; const sink = this.getNodeParameter('sink', 0, '') as string;
const applicationProperties = this.getNodeParameter('headerParametersJson', 0, {}) as const applicationProperties = this.getNodeParameter('headerParametersJson', 0, {}) as
| string | string
@@ -169,30 +187,50 @@ export class Amqp implements INodeType {
throw new NodeOperationError(this.getNode(), 'Queue or Topic required!'); throw new NodeOperationError(this.getNode(), 'Queue or Topic required!');
} }
const container = create_container();
/* /*
Values are documentet here: https://github.com/amqp/rhea#container Values are documented here: https://github.com/amqp/rhea#container
*/ */
const connectOptions: ContainerOptions = { const connectOptions: ContainerOptions = {
host: credentials.hostname, host: credentials.hostname,
hostname: credentials.hostname, hostname: credentials.hostname,
port: credentials.port, port: credentials.port,
reconnect: containerReconnect,
reconnect_limit: containerReconnectLimit,
username: credentials.username ? credentials.username : undefined, username: credentials.username ? credentials.username : undefined,
password: credentials.password ? credentials.password : undefined, password: credentials.password ? credentials.password : undefined,
transport: credentials.transportType ? credentials.transportType : undefined, transport: credentials.transportType ? credentials.transportType : undefined,
container_id: containerId ? containerId : undefined, container_id: containerId ? containerId : undefined,
id: containerId ? containerId : undefined, id: containerId ? containerId : undefined,
reconnect: containerReconnect,
reconnect_limit: containerReconnectLimit,
}; };
const conn = container.connect(connectOptions);
const sender = conn.open_sender(sink); const node = this.getNode();
const responseData: INodeExecutionData[] = await new Promise((resolve, reject) => {
connection = container.connect(connectOptions);
sender = connection.open_sender(sink);
let limit = containerReconnectLimit;
container.on('disconnected', function (context: EventContext) {
//handling this manually as container, despite reconnect_limit, does reconnect on disconnect
if (limit <= 0) {
connection!.options.reconnect = false;
const error = new NodeOperationError(
node,
((context.error as Error) ?? {}).message ?? 'Disconnected',
{
description: `Check your credentials${options.reconnect ? '' : ', and consider enabling reconnect in the options'}`,
itemIndex: 0,
},
);
reject(error);
}
limit--;
});
const responseData: IDataObject[] = await new Promise((resolve) => {
container.once('sendable', (context: EventContext) => { container.once('sendable', (context: EventContext) => {
const returnData = []; const returnData: INodeExecutionData[] = [];
const items = this.getInputData(); const items = this.getInputData();
for (let i = 0; i < items.length; i++) { for (let i = 0; i < items.length; i++) {
@@ -214,23 +252,23 @@ export class Amqp implements INodeType {
body, body,
}); });
returnData.push({ id: result?.id }); returnData.push({ json: { id: result?.id }, pairedItems: { item: i } });
} }
resolve(returnData); resolve(returnData);
}); });
}); });
sender.close(); return [responseData];
conn.close();
return [this.helpers.returnJsonArray(responseData)];
} catch (error) { } catch (error) {
if (this.continueOnFail(error)) { if (this.continueOnFail(error)) {
return [this.helpers.returnJsonArray({ error: error.message })]; return [[{ json: { error: error.message }, pairedItems: { item: 0 } }]];
} else { } else {
throw error; throw error;
} }
} finally {
if (sender) (sender as Sender).close();
if (connection) (connection as Connection).close();
} }
} }
} }