github/directus
1
1
Fork 0
mirror of https://github.com/directus/directus.git synced 2026-04-25 03:00:53 -04:00
Code Issues Packages Projects Releases Wiki Activity

Update seeder to chunk

Browse Source
This commit is contained in:
rijkvanzanten
2020-07-29 15:03:04 -04:00
parent 4509f34457
commit 5894f5dbd4
2 changed files with 28 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
api/src/database/run-seed.ts
View File

@@ -2,8 +2,8 @@ import Knex, { ColumnBuilder } from 'knex';
import fse from 'fs-extra';
import path from 'path';
import yaml from 'js-yaml';
import { types } from '../types';
import { isObject, merge } from 'lodash';
import { types, Item } from '../types';
import { isObject, merge, reduce } from 'lodash';
type SeedData = {
tables?: {
@@ -106,8 +106,30 @@ export default async function runSeed(knex: Knex, seed: string) {
return merge({}, defaults, row);
});
await transaction(table).insert(dataWithDefaults);
const chunks = chunkArray(dataWithDefaults, 25);
for (const chunk of chunks) {
await transaction(table).insert(chunk);
}
}
}
});
}
function chunkArray(array: Item[], chunkSize: number) {
return reduce(
array,
(result: Item[][], value) => {
let lastChunk: Item[] = result[result.length - 1];
if (lastChunk.length < chunkSize) {
lastChunk.push(value);
} else {
result.push([value]);
}
return result;
},
[[]]
);
}