mirror of
https://github.com/Abdulazizzn/n8n-enterprise-unlocked.git
synced 2025-12-17 18:12:04 +00:00
feat(Postgres Node): Batching warning for executeQuery operation insert query (#14287)
This commit is contained in:
@@ -1,4 +1,4 @@
|
||||
import type { IDataObject, INode } from 'n8n-workflow';
|
||||
import type { IDataObject, IExecuteFunctions, INode } from 'n8n-workflow';
|
||||
import { NodeOperationError } from 'n8n-workflow';
|
||||
import pgPromise from 'pg-promise';
|
||||
|
||||
@@ -18,6 +18,7 @@ import {
|
||||
convertValuesToJsonWithPgp,
|
||||
hasJsonDataTypeInSchema,
|
||||
evaluateExpression,
|
||||
addExecutionHints,
|
||||
} from '../../v2/helpers/utils';
|
||||
|
||||
const node: INode = {
|
||||
@@ -535,3 +536,38 @@ describe('Test PostgresV2, convertArraysToPostgresFormat', () => {
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
describe('Test PostgresV2, addExecutionHints', () => {
|
||||
it('should add batching insert hint to executeQuery operation', () => {
|
||||
const context = {
|
||||
getNodeParameter: (parameterName: string) => {
|
||||
if (parameterName === 'options.queryBatching') {
|
||||
return 'single';
|
||||
}
|
||||
if (parameterName === 'query') {
|
||||
return 'INSERT INTO my_test_table VALUES (`{{ $json.name }}`)';
|
||||
}
|
||||
},
|
||||
addExecutionHints: jest.fn(),
|
||||
} as unknown as IExecuteFunctions;
|
||||
|
||||
addExecutionHints(context, [{ json: {} }, { json: {} }], 'executeQuery', false);
|
||||
expect(context.addExecutionHints).toHaveBeenCalledWith({
|
||||
message:
|
||||
"Inserts were batched for performance. If you need to preserve item matching, consider changing 'Query batching' to 'Independent' in the options.",
|
||||
location: 'outputPane',
|
||||
});
|
||||
});
|
||||
it('should add run per item hint to select operation', () => {
|
||||
const context = {
|
||||
addExecutionHints: jest.fn(),
|
||||
} as unknown as IExecuteFunctions;
|
||||
|
||||
addExecutionHints(context, [{ json: {} }, { json: {} }], 'select', false);
|
||||
expect(context.addExecutionHints).toHaveBeenCalledWith({
|
||||
location: 'outputPane',
|
||||
message:
|
||||
"This node ran 2 times, once for each input item. To run for the first item only, enable 'execute once' in the node settings",
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
@@ -52,7 +52,7 @@ export const optionsCollection: INodeProperties = {
|
||||
description: 'A single query for all incoming items',
|
||||
},
|
||||
{
|
||||
name: 'Independently',
|
||||
name: 'Independent',
|
||||
value: 'independently',
|
||||
description: 'Execute one query per incoming item of the run',
|
||||
},
|
||||
|
||||
@@ -5,7 +5,7 @@ import * as database from './database/Database.resource';
|
||||
import type { PostgresType } from './node.type';
|
||||
import { configurePostgres } from '../../transport';
|
||||
import type { PostgresNodeCredentials, PostgresNodeOptions } from '../helpers/interfaces';
|
||||
import { configureQueryRunner } from '../helpers/utils';
|
||||
import { addExecutionHints, configureQueryRunner } from '../helpers/utils';
|
||||
|
||||
export async function router(this: IExecuteFunctions): Promise<INodeExecutionData[][]> {
|
||||
let returnData: INodeExecutionData[] = [];
|
||||
@@ -53,12 +53,7 @@ export async function router(this: IExecuteFunctions): Promise<INodeExecutionDat
|
||||
);
|
||||
}
|
||||
|
||||
if (operation === 'select' && items.length > 1 && !node.executeOnce) {
|
||||
this.addExecutionHints({
|
||||
message: `This node ran ${items.length} times, once for each input item. To run for the first item only, enable 'execute once' in the node settings`,
|
||||
location: 'outputPane',
|
||||
});
|
||||
}
|
||||
addExecutionHints(this, items, operation, node.executeOnce);
|
||||
|
||||
return [returnData];
|
||||
}
|
||||
|
||||
@@ -616,3 +616,30 @@ export const convertArraysToPostgresFormat = (
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
export function addExecutionHints(
|
||||
context: IExecuteFunctions,
|
||||
items: INodeExecutionData[],
|
||||
operation: string,
|
||||
executeOnce: boolean | undefined,
|
||||
) {
|
||||
if (operation === 'select' && items.length > 1 && !executeOnce) {
|
||||
context.addExecutionHints({
|
||||
message: `This node ran ${items.length} times, once for each input item. To run for the first item only, enable 'execute once' in the node settings`,
|
||||
location: 'outputPane',
|
||||
});
|
||||
}
|
||||
|
||||
if (
|
||||
operation === 'executeQuery' &&
|
||||
items.length > 1 &&
|
||||
(context.getNodeParameter('options.queryBatching', 0, 'single') as string) === 'single' &&
|
||||
(context.getNodeParameter('query', 0, '') as string).toLowerCase().startsWith('insert')
|
||||
) {
|
||||
context.addExecutionHints({
|
||||
message:
|
||||
"Inserts were batched for performance. If you need to preserve item matching, consider changing 'Query batching' to 'Independent' in the options.",
|
||||
location: 'outputPane',
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user