是否可以使用 sequelize 在迁移中更新 postgresql 数据库中的现有数据

Question

我正在尝试使用 sequelize 迁移向 model 添加一个字段，并且需要从“向上”迁移中更新数据。 但是，在迁移中调用数据库以更新数据不会完成或引发错误，它们只是挂起。

我正在尝试将registrationType字段添加到我数据库中现有的 model 中。 这个字段不应该是 null，所以我需要添加 'allowNull: false' 属性。 尚未设置registrationType的旧注册应使用 model 中已存在的数据更新为正确的类型。 为此，我需要访问 model 中的 ID 字段，获取链接的 object （注册链接到具有 locationtype 的locationtype ）并使用它来确定registrationType 。 我将代码添加到迁移中，就好像它是更新某些数据的正常数据库操作一样，但是这些调用不会返回或抛出错误。

我不能（也不想）使用默认值，因为应该根据现有数据为每个注册确定该值（并且添加默认值将使 allowNull 属性过时）。 我的方法： - 添加没有 'allowNull' 约束的列（使用 addColumn） - 更新所有现有数据 - 添加 'allowNull' 约束（使用 changeColumn）

"use strict";

/*const db = require("../models");
const Registration = db.Registration;
const Site = db.Site;
const Location = db.Location;
*/
const REGISTATION_MODEL = "Registrations";
module.exports = {
    up: async (queryInterface, Sequelize) => {
        const transaction = await queryInterface.sequelize.transaction();

        try {
            const Registration = await queryInterface.sequelize.import(
                "../models/registration.js"
            );
            const Site = await queryInterface.sequelize.import(
                "../models/site.js"
            );
            const Location = await queryInterface.sequelize.import(
                "../models/location.js"
            );

            await queryInterface.addColumn(
                REGISTATION_MODEL,
                "registrationType",
                {
                    type: Sequelize.STRING
                },
                { transaction }
            );

            console.log(
                " * Column added, going to update existing registrations."
            );
            const registrations = await Registration.findAll();
            console.log(
                `\tFound ${registrations.length} registrations to be updated.`
            );
            for await (const registration of registrations) {
                const site = await Site.findByPk(registration.SiteId);
                const location = await Location.findByPk(site.LocationId);
                await registration.update(
                    {
                        registrationType: location.locationType
                    },
                    { transaction }
                );
            }

            console.log(`\tUpdated ${registrations.length} registrations.`);
            console.log(" * Adding 'allowNull:false' to field.");
            //allowNull: false
            await queryInterface.changeColumn(
                REGISTATION_MODEL,
                "registrationType",
                { type: Sequelize.STRING, allowNull: false },
                { transaction: t }
            );
            await transaction.commit();
        } catch (ex) {
            await transaction.rollback();
            console.error("Something went wrong: ", ex);
        }
    },

    down: (queryInterface, Sequelize) => {
        return queryInterface.removeColumn(
            REGISTATION_MODEL,
            "registrationType"
        );
    }
};

Output：

Loaded configuration file "config/config.json".
Using environment "development".
== 20191107134514-add-registration-types: migrating =======
 * Column added, going to update existing registrations.

然后挂起。

此处显示的代码不会产生任何错误，也不会产生任何 output。 我添加了 console.log 语句，并再次运行迁移，这表明代码在第一次 findAll() 调用时挂起。 谁能告诉我该怎么做？

Answer 1

更新：正如聊天中所讨论的，我们需要将 addColumn、changeColumn 和更新查询分离到单独的文件中。 因为我们不能在同一个事务下运行所有这些，因为它们影响同一个表。

如果您想在迁移函数中await ，您需要使up/down function 异步并使用 await 和异步运行您的事务：

以下是仅用于更新表的迁移文件。 为 addColumn 和 changeColumn 创建单独的迁移。 按顺序运行它们：

1. 添加列
2. 更新表
3. 更改列

迁移表更新：

up: async (queryInterface, Sequelize) => {
    // declare transaction outside try catch so it is available in both
    const transaction = await queryInterface.sequelize.transaction()
    try {
        // No need to use transactions for read operations
        const registrations = await Registration.findAll()

        // using for...of loop which supports awaiting inside it
        for await (const registration of registrations) {
            const site = await Site.findByPk(registration.SiteId)
            const location = await Location.findByPk(site.LocationId)

            // Make sure to await on all sequelize methdos
            await registration.update({ registrationType: location.locationType })
        }

        // Commit transaction if no error occurs
        await transaction.commit()
    } catch (error) {
        // Rollback transaction if error occurs
        await transaction.rollback()
        console.error("Something went wrong: ", ex)
    }
}