Sequelize: seed with associations

后端 未结 3 556
南旧
南旧 2021-02-03 10:37

I have 2 models, Courses and Videos, for example. And Courses has many Videos.

// course.js
\'use strict\';

module.exports = (sequelize, DataTypes) => {
  co         


        
相关标签:
3条回答
  • 2021-02-03 11:15

    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, {} );
    }
    
    0 讨论(0)
  • 2021-02-03 11:32

    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, {});
      },
    };
    
    0 讨论(0)
  • 2021-02-03 11:35

    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, {});
      }
    };
    
    0 讨论(0)
提交回复
热议问题