使用Gorm UpdateColumns触发钩子时获取错误vote值的解决方法？

问题原因 #

当使用UpdateColumns传入map进行批量字段更新时，Gorm不会将数据库中更新后的字段值同步到当前的模型实例（即钩子中的p *Post），因此钩子中访问的p.Upvote和p.Downvote是实例的初始零值，而非数据库中的最新值。

解决方案 #

方案1：在钩子中重新查询最新记录 #

在AfterUpdate钩子内，通过事务重新查询目标记录，获取最新的upvote和downvote值后再计算并更新VoteScore：

func (p *Post) AfterUpdate(tx *gorm.DB) error {
    // 重新查询当前记录的最新数据
    var latestPost Post
    if err := tx.Where("id = ?", p.ID).First(&latestPost).Error; err != nil {
        return err
    }

    // 计算并更新VoteScore
    p.VoteScore = int(latestPost.Upvote) - int(latestPost.Downvote)
    return tx.Model(p).UpdateColumn("vote_score", p.VoteScore).Error
}

注意：确保Post结构体包含ID字段（用于查询定位）。

方案2：改用模型实例进行更新 #

如果能提前获取到upvote和downvote的最新值，构造包含最新值的Post实例，使用Updates方法更新，此时Gorm会将更新后的值同步到实例中，钩子即可直接获取正确数值：

// 构造包含最新值的Post实例
updatedPost := &Post{
    ID:        content.id,
    Upvote:    newUpvoteVal, // 替换为实际最新upvote值
    Downvote:  newDownvoteVal, // 替换为实际最新downvote值
}

// 使用实例更新
query := tx.Model(content.model).Updates(updatedPost)
if err := query.Error; err != nil {
    tx.Rollback()
    return err
}

钩子可以简化为：

func (p *Post) AfterUpdate(tx *gorm.DB) error {
    p.VoteScore = int(p.Upvote) - int(p.Downvote)
    return tx.Model(p).UpdateColumn("vote_score", p.VoteScore).Error
}

方案3：使用PostgreSQL触发器（推荐） #

直接在数据库层面创建触发器，当upvote或downvote字段更新时自动计算vote_score，无需依赖应用层钩子，性能更优且更可靠：

-- 创建计算vote_score的函数
CREATE OR REPLACE FUNCTION update_vote_score()
RETURNS TRIGGER AS $$
BEGIN
    NEW.vote_score := NEW.upvote - NEW.downvote;
    RETURN NEW;
END;
$$ LANGUAGE plpgsql;

-- 为posts表创建BEFORE UPDATE触发器
CREATE TRIGGER trigger_update_vote_score
BEFORE UPDATE OF upvote, downvote ON posts
FOR EACH ROW EXECUTE FUNCTION update_vote_score();

此后每次更新upvote或downvote时，数据库会自动计算并更新vote_score字段，应用层无需额外处理。

内容的提问来源于stack exchange，提问作者Matthew Trent