mirror of
https://github.com/n8n-io/n8n.git
synced 2024-11-09 22:24:05 -08:00
fix(Google BigQuery Node): Better error messages, transform timestamps (#9255)
Co-authored-by: Jonathan Bennetts <jonathan.bennetts@gmail.com>
This commit is contained in:
parent
e896889394
commit
7ff24f134b
|
@ -13,12 +13,13 @@ export class GoogleBigQuery extends VersionedNodeType {
|
|||
group: ['input'],
|
||||
subtitle: '={{$parameter["operation"] + ": " + $parameter["resource"]}}',
|
||||
description: 'Consume Google BigQuery API',
|
||||
defaultVersion: 2,
|
||||
defaultVersion: 2.1,
|
||||
};
|
||||
|
||||
const nodeVersions: IVersionedNodeType['nodeVersions'] = {
|
||||
1: new GoogleBigQueryV1(baseDescription),
|
||||
2: new GoogleBigQueryV2(baseDescription),
|
||||
2.1: new GoogleBigQueryV2(baseDescription),
|
||||
};
|
||||
|
||||
super(nodeVersions, baseDescription);
|
||||
|
|
|
@ -1,11 +1,14 @@
|
|||
import type { IDataObject, IExecuteFunctions } from 'n8n-workflow';
|
||||
import type { IDataObject, IExecuteFunctions, INode } from 'n8n-workflow';
|
||||
import { constructExecutionMetaData } from 'n8n-core';
|
||||
import { mock } from 'jest-mock-extended';
|
||||
import { prepareOutput } from '../../../v2/helpers/utils';
|
||||
|
||||
describe('Google BigQuery v2 Utils', () => {
|
||||
it('should prepareOutput', () => {
|
||||
const thisArg = mock<IExecuteFunctions>({ helpers: mock({ constructExecutionMetaData }) });
|
||||
const thisArg = mock<IExecuteFunctions>({
|
||||
getNode: () => ({ typeVersion: 2.1 }) as INode,
|
||||
helpers: mock({ constructExecutionMetaData }),
|
||||
});
|
||||
const response: IDataObject = {
|
||||
kind: 'bigquery#getQueryResultsResponse',
|
||||
etag: 'e_tag',
|
||||
|
|
|
@ -249,10 +249,16 @@ export async function execute(this: IExecuteFunctions): Promise<INodeExecutionDa
|
|||
returnData.push(...executionErrorData);
|
||||
continue;
|
||||
}
|
||||
if ((error.message as string).includes('location')) {
|
||||
if ((error.message as string).includes('location') || error.httpCode === '404') {
|
||||
error.description =
|
||||
"Are you sure your table is in that region? You can specify the region using the 'Location' parameter from options.";
|
||||
}
|
||||
|
||||
if (error.httpCode === '403' && error.message.includes('Drive')) {
|
||||
error.description =
|
||||
'If your table(s) pull from a document in Google Drive, make sure that document is shared with your user';
|
||||
}
|
||||
|
||||
throw new NodeOperationError(this.getNode(), error as Error, {
|
||||
itemIndex: i,
|
||||
description: error.description,
|
||||
|
|
|
@ -7,7 +7,7 @@ export const versionDescription: INodeTypeDescription = {
|
|||
name: 'googleBigQuery',
|
||||
icon: 'file:googleBigQuery.svg',
|
||||
group: ['input'],
|
||||
version: 2,
|
||||
version: [2, 2.1],
|
||||
subtitle: '={{$parameter["operation"]}}',
|
||||
description: 'Consume Google BigQuery API',
|
||||
defaults: {
|
||||
|
|
|
@ -1,8 +1,9 @@
|
|||
import { DateTime } from 'luxon';
|
||||
import type { IDataObject, IExecuteFunctions, INodeExecutionData } from 'n8n-workflow';
|
||||
import { jsonParse, NodeOperationError } from 'n8n-workflow';
|
||||
import type { SchemaField, TableRawData, TableSchema } from './interfaces';
|
||||
|
||||
function getFieldValue(schemaField: SchemaField, field: IDataObject) {
|
||||
function getFieldValue(schemaField: SchemaField, field: IDataObject, parseTimestamps = false) {
|
||||
if (schemaField.type === 'RECORD') {
|
||||
// eslint-disable-next-line @typescript-eslint/no-use-before-define
|
||||
return simplify([field.v as TableRawData], schemaField.fields as unknown as SchemaField[]);
|
||||
|
@ -12,6 +13,9 @@ function getFieldValue(schemaField: SchemaField, field: IDataObject) {
|
|||
try {
|
||||
value = jsonParse(value as string);
|
||||
} catch (error) {}
|
||||
} else if (schemaField.type === 'TIMESTAMP' && parseTimestamps) {
|
||||
const dt = DateTime.fromSeconds(Number(value));
|
||||
value = dt.isValid ? dt.toISO() : value;
|
||||
}
|
||||
return value;
|
||||
}
|
||||
|
@ -26,18 +30,27 @@ export function wrapData(data: IDataObject | IDataObject[]): INodeExecutionData[
|
|||
}));
|
||||
}
|
||||
|
||||
export function simplify(data: TableRawData[], schema: SchemaField[], includeSchema = false) {
|
||||
export function simplify(
|
||||
data: TableRawData[],
|
||||
schema: SchemaField[],
|
||||
includeSchema = false,
|
||||
parseTimestamps = false,
|
||||
) {
|
||||
const returnData: IDataObject[] = [];
|
||||
for (const entry of data) {
|
||||
const record: IDataObject = {};
|
||||
|
||||
for (const [index, field] of entry.f.entries()) {
|
||||
if (schema[index].mode !== 'REPEATED') {
|
||||
record[schema[index].name] = getFieldValue(schema[index], field);
|
||||
record[schema[index].name] = getFieldValue(schema[index], field, parseTimestamps);
|
||||
} else {
|
||||
record[schema[index].name] = (field.v as unknown as IDataObject[]).flatMap(
|
||||
(repeatedField) => {
|
||||
return getFieldValue(schema[index], repeatedField as unknown as IDataObject);
|
||||
return getFieldValue(
|
||||
schema[index],
|
||||
repeatedField as unknown as IDataObject,
|
||||
parseTimestamps,
|
||||
);
|
||||
},
|
||||
);
|
||||
}
|
||||
|
@ -68,12 +81,18 @@ export function prepareOutput(
|
|||
responseData = response;
|
||||
} else {
|
||||
const { rows, schema } = response;
|
||||
const parseTimestamps = this.getNode().typeVersion >= 2.1;
|
||||
|
||||
if (rows !== undefined && schema !== undefined) {
|
||||
const fields = (schema as TableSchema).fields;
|
||||
responseData = rows;
|
||||
|
||||
responseData = simplify(responseData as TableRawData[], fields, includeSchema);
|
||||
responseData = simplify(
|
||||
responseData as TableRawData[],
|
||||
fields,
|
||||
includeSchema,
|
||||
parseTimestamps,
|
||||
);
|
||||
} else if (schema && includeSchema) {
|
||||
responseData = { success: true, _schema: schema };
|
||||
} else {
|
||||
|
|
Loading…
Reference in a new issue