Using new bulk insert utility for releases, media and actors.

This commit is contained in:
DebaucheryLibrarian
2020-08-14 23:21:53 +02:00
parent e996a45bf5
commit 50c5f921f5
11 changed files with 40 additions and 13 deletions

29
src/utils/bulk-insert.js Normal file
View File

@@ -0,0 +1,29 @@
'use strict';
const knex = require('../knex');
const chunk = require('./chunk');
async function bulkUpsert(table, items, conflict, update = true, chunkSize) {
const updated = (conflict === false && ':query ON CONFLICT DO NOTHING RETURNING *;')
|| (conflict && update && `
:query ON CONFLICT (${conflict})
DO UPDATE SET ${(update === true
? Object.keys(items[0]) // derive updating props
: update)
.reduce((acc, prop, index) => `${acc}${index > 0 ? ',' : ''}\n${prop} = ${conflict.includes(prop) ? table : 'EXCLUDED'}.${prop}`, '')}
RETURNING *;
`);
return knex.transaction(async (transaction) => {
const queries = chunk(items, chunkSize)
.map(chunkItems => knex.raw(updated || ':query RETURNING *;', {
query: knex(table).insert(chunkItems).transacting(transaction),
}));
const responses = await Promise.all(queries);
return responses.flat().map(response => response.rows).flat();
});
}
module.exports = bulkUpsert;