mirror of
https://github.com/laurent22/joplin.git
synced 2024-12-24 10:27:10 +02:00
57 lines
1.4 KiB
TypeScript
57 lines
1.4 KiB
TypeScript
const Folder = require('lib/models/Folder');
|
|
const Note = require('lib/models/Note');
|
|
|
|
function randomIndex(array:any[]):number {
|
|
return Math.round(Math.random() * (array.length - 1));
|
|
}
|
|
|
|
export default async function populateDatabase(db:any) {
|
|
await db.clearForTesting();
|
|
|
|
const folderCount = 2000;
|
|
const noteCount = 20000;
|
|
|
|
const createdFolderIds:string[] = [];
|
|
const createdNoteIds:string[] = [];
|
|
|
|
for (let i = 0; i < folderCount; i++) {
|
|
const folder:any = {
|
|
title: `folder${i}`,
|
|
};
|
|
|
|
const isRoot = Math.random() <= 0.1 || i === 0;
|
|
|
|
if (!isRoot) {
|
|
const parentIndex = randomIndex(createdFolderIds);
|
|
folder.parent_id = createdFolderIds[parentIndex];
|
|
}
|
|
|
|
const savedFolder = await Folder.save(folder);
|
|
createdFolderIds.push(savedFolder.id);
|
|
|
|
console.info(`Folders: ${i} / ${folderCount}`);
|
|
}
|
|
|
|
let noteBatch = [];
|
|
for (let i = 0; i < noteCount; i++) {
|
|
const note:any = { title: `note${i}`, body: `This is note num. ${i}` };
|
|
const parentIndex = randomIndex(createdFolderIds);
|
|
note.parent_id = createdFolderIds[parentIndex];
|
|
|
|
noteBatch.push(Note.save(note, { dispatchUpdateAction: false }).then((savedNote:any) => {
|
|
createdNoteIds.push(savedNote.id);
|
|
console.info(`Notes: ${i} / ${noteCount}`);
|
|
}));
|
|
|
|
if (noteBatch.length > 1000) {
|
|
await Promise.all(noteBatch);
|
|
noteBatch = [];
|
|
}
|
|
}
|
|
|
|
if (noteBatch.length) {
|
|
await Promise.all(noteBatch);
|
|
noteBatch = [];
|
|
}
|
|
}
|