n8n/packages/nodes-base/nodes/MongoDb/MongoDb.node.ts

113 lines
3.1 KiB
TypeScript
Raw Normal View History

import { IExecuteFunctions } from 'n8n-core';
2019-10-15 09:39:00 -07:00
import {
IDataObject,
INodeExecutionData,
INodeType,
INodeTypeDescription
} from 'n8n-workflow';
import { nodeDescription } from './mongo.node.options';
import { MongoClient } from 'mongodb';
import {
getItemCopy,
validateAndResolveMongoCredentials
} from './mongo.node.utils';
2019-10-15 09:39:00 -07:00
export class MongoDb implements INodeType {
description: INodeTypeDescription = nodeDescription;
2019-10-15 09:39:00 -07:00
async execute(this: IExecuteFunctions): Promise<INodeExecutionData[][]> {
const { database, connectionString } = validateAndResolveMongoCredentials(
this.getCredentials('mongoDb')
);
const client: MongoClient = await MongoClient.connect(connectionString, {
useNewUrlParser: true,
useUnifiedTopology: true
});
const mdb = client.db(database as string);
2019-10-15 09:39:00 -07:00
let returnItems = [];
const items = this.getInputData();
const operation = this.getNodeParameter('operation', 0) as string;
2019-10-15 09:39:00 -07:00
if (operation === 'find') {
2019-10-15 09:39:00 -07:00
// ----------------------------------
// find
// ----------------------------------
const queryResult = await mdb
.collection(this.getNodeParameter('collection', 0) as string)
.find(JSON.parse(this.getNodeParameter('query', 0) as string))
2019-10-15 09:39:00 -07:00
.toArray();
returnItems = this.helpers.returnJsonArray(queryResult as IDataObject[]);
} else if (operation === 'insert') {
2019-10-15 09:39:00 -07:00
// ----------------------------------
// insert
// ----------------------------------
2019-10-15 12:14:26 -07:00
// Prepare the data to insert and copy it to be returned
const fields = (this.getNodeParameter('fields', 0) as string)
.split(',')
.map(f => f.trim())
.filter(f => !!f);
2019-10-15 09:39:00 -07:00
2019-10-15 12:14:26 -07:00
const insertItems = getItemCopy(items, fields);
2019-10-15 09:39:00 -07:00
2019-10-15 12:14:26 -07:00
const { insertedIds } = await mdb
.collection(this.getNodeParameter('collection', 0) as string)
.insertMany(insertItems);
2019-10-15 12:14:26 -07:00
// Add the id to the data
for (const i of Object.keys(insertedIds)) {
2019-10-15 12:14:26 -07:00
returnItems.push({
json: {
...insertItems[parseInt(i, 10)],
id: insertedIds[parseInt(i, 10)] as string
}
2019-10-15 12:14:26 -07:00
});
}
} else if (operation === 'update') {
2019-10-15 09:39:00 -07:00
// ----------------------------------
// update
// ----------------------------------
const fields = (this.getNodeParameter('fields', 0) as string)
.split(',')
.map(f => f.trim())
.filter(f => !!f);
let updateKey = this.getNodeParameter('updateKey', 0) as string;
updateKey = updateKey.trim();
if (!fields.includes(updateKey)) {
fields.push(updateKey);
}
2019-10-15 09:39:00 -07:00
2019-10-15 12:14:26 -07:00
// Prepare the data to update and copy it to be returned
const updateItems = getItemCopy(items, fields);
2019-10-15 09:39:00 -07:00
for (const item of updateItems) {
if (item[updateKey] === undefined) {
continue;
}
2019-10-15 09:39:00 -07:00
const filter: { [key: string]: string } = {};
2019-10-15 12:14:26 -07:00
filter[updateKey] = item[updateKey] as string;
2019-10-15 09:39:00 -07:00
2019-10-15 12:14:26 -07:00
await mdb
.collection(this.getNodeParameter('collection', 0) as string)
.updateOne(filter, { $set: item });
2019-10-15 12:14:26 -07:00
}
2019-10-15 09:39:00 -07:00
2019-10-15 12:14:26 -07:00
returnItems = this.helpers.returnJsonArray(updateItems as IDataObject[]);
2019-10-15 09:39:00 -07:00
} else {
throw new Error(`The operation "${operation}" is not supported!`);
}
return this.prepareOutputData(returnItems);
}
}