NodeBB/src/database/postgres/helpers.js

'use strict';

var helpers = module.exports;

helpers.valueToString = function (value) {
	return String(value);
};

helpers.removeDuplicateValues = function (values) {
	var others = Array.prototype.slice.call(arguments, 1);
	for (var i = 0; i < values.length; i++) {
		if (values.lastIndexOf(values[i]) !== i) {
			values.splice(i, 1);
			for (var j = 0; j < others.length; j++) {
				others[j].splice(i, 1);
			}
			i -= 1;
		}
	}
};

helpers.ensureLegacyObjectType = async function (db, key, type) {
	await db.query({
		name: 'ensureLegacyObjectTypeBefore',
		text: `
DELETE FROM "legacy_object"
 WHERE "expireAt" IS NOT NULL
   AND "expireAt" <= CURRENT_TIMESTAMP`,
	});

	await db.query({
		name: 'ensureLegacyObjectType1',
		text: `
INSERT INTO "legacy_object" ("_key", "type")
VALUES ($1::TEXT, $2::TEXT::LEGACY_OBJECT_TYPE)
    ON CONFLICT
    DO NOTHING`,
		values: [key, type],
	});

	const res = await db.query({
		name: 'ensureLegacyObjectType2',
		text: `
SELECT "type"
  FROM "legacy_object_live"
 WHERE "_key" = $1::TEXT`,
		values: [key],
	});

	if (res.rows[0].type !== type) {
		throw new Error('database: cannot insert ' + JSON.stringify(key) + ' as ' + type + ' because it already exists as ' + res.rows[0].type);
	}
};

helpers.ensureLegacyObjectsType = async function (db, keys, type) {
	await db.query({
		name: 'ensureLegacyObjectTypeBefore',
		text: `
DELETE FROM "legacy_object"
 WHERE "expireAt" IS NOT NULL
   AND "expireAt" <= CURRENT_TIMESTAMP`,
	});

	await db.query({
		name: 'ensureLegacyObjectsType1',
		text: `
INSERT INTO "legacy_object" ("_key", "type")
SELECT k, $2::TEXT::LEGACY_OBJECT_TYPE
  FROM UNNEST($1::TEXT[]) k
    ON CONFLICT
    DO NOTHING`,
		values: [keys, type],
	});

	const res = await db.query({
		name: 'ensureLegacyObjectsType2',
		text: `
SELECT "_key", "type"
  FROM "legacy_object_live"
 WHERE "_key" = ANY($1::TEXT[])`,
		values: [keys],
	});

	var invalid = res.rows.filter(r => r.type !== type);

	if (invalid.length) {
		const parts = invalid.map(r => JSON.stringify(r._key) + ' is ' + r.type);
		throw new Error('database: cannot insert multiple objects as ' + type + ' because they already exist: ' + parts.join(', '));
	}

	var missing = keys.filter(function (k) {
		return !res.rows.some(r => r._key === k);
	});

	if (missing.length) {
		throw new Error('database: failed to insert keys for objects: ' + JSON.stringify(missing));
	}
};

helpers.noop = function () {};