traxxx/src/utils/upsert.js

'use strict';

const knex = require('../knex');
const logger = require('../logger')(__filename);

async function upsert(table, items, identifier = ['id'], _knex) {
	const identifiers = Array.isArray(identifier) ? identifier : [identifier];

	const duplicates = await knex(table).whereIn(identifiers, items.map((item) => identifiers.map((identifierX) => item[identifierX])));
	const duplicatesByIdentifiers = duplicates.reduce((acc, duplicate) => {
		const duplicateIdentifier = identifiers.map((identifierX) => duplicate[identifierX]).toString();

		return { ...acc, [duplicateIdentifier]: duplicate };
	}, {});

	const { insert, update } = items.reduce((acc, item) => {
		const itemIdentifier = identifiers.map((identifierX) => item[identifierX]).toString();

		if (duplicatesByIdentifiers[itemIdentifier]) {
			acc.update.push(item);
			return acc;
		}

		acc.insert.push(item);
		return acc;
	}, {
		insert: [],
		update: [],
	});

	if (knex) {
		logger.debug(`${table}: Inserting ${insert.length}`);
		logger.debug(`${table}: Updating ${update.length}`);

		const [inserted, updated] = await Promise.all([
			insert.length > 0 ? knex(table).returning('*').insert(insert) : [],
			update.length > 0 ? knex.transaction(async (trx) => Promise.all(update.map((item) => {
				const clause = identifiers.reduce((acc, identifierX) => ({ ...acc, [identifierX]: item[identifierX] }), {});

				return trx
					.where(clause)
					.update(item)
					.into(table)
					.returning('*');
			}))) : [],
		]);

		return {
			inserted: Array.isArray(inserted) ? inserted : [],
			updated: updated.reduce((acc, updatedItems) => acc.concat(updatedItems), []),
		};
	}

	return { insert, update };
}

module.exports = upsert;