Sequelize: seed with associations
The bulkInsert
method returns a promise that is resolved with an ID of the first inserted item. We can use this information to insert videos. It can look like this:
function getId( firstId, items, needly ) {
for ( let i = 0; i < items.length; i++ ) {
if ( items[i].title === needly ) {
return firstId + i;
}
}
return null;
}
exports.up = async ( queryInterface, Sequelize ) => {
const courses = [
{
title: 'Course 1',
description: '...',
},
{
title: 'Course 2',
description: '...',
},
{
title: 'Course 3',
description: '...',
},
{
title: 'Course 4',
description: '...',
},
{
title: 'Course 5',
description: '...',
},
];
const firstId = await queryInterface.bulkInsert( 'courses', courses, {} );
const course2Id = getId( firstId, courses, 'Course 2' );
const course5Id = getId( firstId, courses, 'Course 5' );
return queryInterface.bulkInsert( 'categories', [
{ title: 'Video 1', description: '...', courseId: course2Id },
{ title: 'Video 2', description: '...', courseId: course2Id },
{ title: 'Video 3', description: '...', courseId: course5Id },
{ title: 'Video 4', description: '...', courseId: course5Id },
{ title: 'Video 5', description: '...', courseId: course5Id },
], {} );
};
exports.down = async ( queryInterface ) => {
await queryInterface.bulkDelete( 'videos', null, {} );
await queryInterface.bulkDelete( 'courses', null, {} );
}
When passing { returning: true }
in the options field of bulkInsert
it will return the created objects.
let createdOjects = await queryInterface.bulkInsert("table_name", data_to_be_inserted, { returning: true });
Also, you may pass an array with the fields you are interested in, e.g. the ID { returning: ['id'] }
and this will return an array of IDs of the created objects
let createdIds = await queryInterface.bulkInsert("table_name", data_to_be_inserted, { returning: ['id'] });
You can loop through the returned objects/ids and insert the nested objects using bulkInsert
as well.
Sample code:
module.exports = {
up: async (queryInterface) => {
let courses = ["..."]
let videos = ["..."]
let videoIds = await queryInterface.bulkInsert("courses", courses, { returning: ["id"] });
//add courseId to each video object -- depends on your scheme
await queryInterface.bulkInsert("videos", videos);
},
down: async (queryInterface) => {
await queryInterface.bulkDelete("videos", null, {});
await queryInterface.bulkDelete("courses", null, {});
},
};
You can use Sequelize's queryInterface
to drop down to raw SQL in order to insert model instances that require associations. In your case, the easiest way would to create one seeder for courses and videos. (One note: I don't know how you are defining your primary and foreign key so I am making an assumption that the videos table has a field course_id
.)
module.exports = {
up: async (queryInterface) => {
await queryInterface.bulkInsert('courses', [
{title: 'Course 1', description: 'description 1', id: 1}
{title: 'Course 2', description: 'description 2', id: 2}
], {});
const courses = await queryInterface.sequelize.query(
`SELECT id from COURSES;`
);
const courseRows = courses[0];
return await queryInterface.bulkInsert('videos', [
{title: 'Movie 1', description: '...', id: '1', course_id: courseRows[0].id}
{title: 'Movie 2', description: '...', id: '2', course_id: courseRows[0].id},
{title: 'Movie 3', description: '...', id: '3', course_id: courseRows[0].id},
], {});
},
down: async (queryInterface) => {
await queryInterface.bulkDelete('videos', null, {});
await queryInterface.bulkDelete('courses', null, {});
}
};