优化GORM多对多关联更新：减少数据库查询次数

问题分析 #

你的初始实现需要多次与数据库交互：先查询所有现有关联，再循环删除不需要的记录，最后循环插入新关联。这种方式在关联数量较多时会产生大量DB请求，性能低下。我们可以通过批量SQL操作+子查询来减少DB交互次数，提升性能。

优化方案 #

首先确保你的MovieGenre中间表模型定义正确（GORM多对多关联会自动生成，但手动定义更便于直接操作）：

type MovieGenre struct {
    MovieID int `gorm:"primaryKey"`
    GenreID int `gorm:"primaryKey"`
}

以下是优化后的实现，通过事务保证原子性，用批量删除和批量插入替代循环操作：

import (
    "gorm.io/gorm"
    "gorm.io/gorm/clause"
)

func UpdateMovieGenres(movieID int, genreIDs []int) error {
    // 开启事务，确保删除和插入操作要么全成功要么全回滚
    tx := db.Begin()
    defer func() {
        if r := recover(); r != nil {
            tx.Rollback()
        }
    }()

    if err := tx.Error; err != nil {
        return err
    }

    // 1. 批量删除不在新分类列表中的关联记录
    if len(genreIDs) > 0 {
        // 直接通过WHERE条件筛选要删除的记录，无需先查询现有关联
        if err := tx.Where("movie_id = ? AND genre_id NOT IN ?", movieID, genreIDs).Delete(&MovieGenre{}).Error; err != nil {
            tx.Rollback()
            return err
        }
    } else {
        // 若新分类列表为空，直接清空该电影所有关联
        if err := tx.Where("movie_id = ?", movieID).Delete(&MovieGenre{}).Error; err != nil {
            tx.Rollback()
            return err
        }
    }

    // 2. 批量插入新分类中未存在的关联记录
    if len(genreIDs) > 0 {
        // 构建待插入的关联记录切片
        var movieGenres []MovieGenre
        for _, gid := range genreIDs {
            movieGenres = append(movieGenres, MovieGenre{MovieID: movieID, GenreID: gid})
        }

        // 使用OnConflict子句忽略已存在的记录，避免重复插入
        if err := tx.Clauses(clause.OnConflict{
            Columns:   []clause.Column{{Name: "movie_id"}, {Name: "genre_id"}}, // 联合主键
            DoNothing: true,
        }).CreateInBatches(movieGenres, len(movieGenres)).Error; err != nil {
            tx.Rollback()
            return err
        }
    }

    return tx.Commit().Error
}

进阶：纯子查询实现插入 #

如果你不想提前构建movieGenres切片，可以直接用SQL子查询完成插入逻辑，进一步减少内存开销：

// 替换上述插入部分的代码
if err := tx.Exec(`
    INSERT INTO movie_genres (movie_id, genre_id)
    SELECT ?, g.id
    FROM (SELECT id FROM genres WHERE id IN ?) g
    WHERE NOT EXISTS (
        SELECT 1 FROM movie_genres WHERE movie_id = ? AND genre_id = g.id
    )
`, movieID, genreIDs, movieID).Error; err != nil {
    tx.Rollback()
    return err
}

优化效果对比 #

• 原实现：1次查询 + N次删除 + M次插入（N为待删除关联数，M为待新增关联数），共1+N+M次DB交互
• 优化后：1次删除 + 1次插入（含事务操作），共3次DB交互，与关联数量无关，性能提升显著

内容的提问来源于stack exchange，提问作者vahid yg