当前位置：首页 > 文章列表 > Golang > Go问答 > 使用 Gorm 进行多对多关系的数据查询

使用 Gorm 进行多对多关系的数据查询

来源：stackoverflow 2024-03-25 15:57:36 0浏览收藏

**摘要：** 本文探讨了使用 Gorm 进行多对多关系数据查询的方法。它着重于解决查询电影的问题，这些电影在导演、类别、类型和编剧的多对多关系中具有特定的 ID。文章提供了解决方案，涉及使用原始 SQL 或 Gorm 查询构建器，并解释了如何处理多个 ID 和多个关系的情况。

问题内容

我有一个 movie 模型，该模型在 gorm 中定义了很多多对多关系。现在我必须创建一个 api 来查询在这些关系中具有 id 的所有电影。

让我们把它分成小块：

电影模型：

type movie struct {
    gorm.model
    titleid           string     `gorm:"size:50;not null" json:"title_id"`
    originalname      string     `gorm:"size:250;not null" json:"original_name"`
    categories        []category `gorm:"null;many2many:movie_categories;" json:"categories"`
    types             []type     `gorm:"null;many2many:movie_types;" json:"types"`
    directors         []cast     `gorm:"null;many2many:movie_directors;" json:"directors"`
    writers           []cast     `gorm:"null;many2many:movie_writers;" json:"writers"`
}

我的查询功能：

func (u *Movie) FindMoviesByDirectors(db *gorm.DB, directors string) (*[]Movie, error) {
    var err error
    movies := []Movie{}
    err = db.Set("gorm:auto_preload", true).Where("directors = ?", directors).Find(&movies).Error
    if err != nil {
        return &[]Movie{}, err
    }
    return &movies, err
}

如何通过关系中的 id 查询电影？（在我的例子中，查询所有 id 为 1 的电影作为 director。）
如何扩展它来查询具有多个 id 和多个关系的电影？（例如，查询所有具有 [1,2] id 的电影以获取 directors 和 [3,4] 的 types。）

解决方案

问题回复

您可以查询引用的多对多关系中的外键引用 id（参见下面的示例）
您可以使用 in 运算符一次查询多个值。 gorm 支持此参见 https://gorm.io/docs/query.html

可能的解决方案

为此，您的原始 sql 可能如下所示：

select 
    m.* 
from
    movies m
left join movie_categories mc on mc.movie_id = m.id
left join movie_types mt on mt.movie_id = m.id
left join movie_directors md on md.movie_id = m.id
left join movie_writers mw on mw.movie_id = m.id
where
  mc.category_id in (?) or
  mt.type_id in (?) or
  md.director_id in (?) or
  mw.writer_id in (?)

注意。如果电影在特定关系中没有记录，我会使用左连接。 where 条件将确保返回所需的记录。此外，or 意味着可以满足这些标准中的任何一个。如果您希望满足所有这些条件，您可以使用 and 而不是 or。

以下可能解决方案的假设

假设

上面的sql存储在变量sql中
所需的类别 id 存储在数组 categories 中，例如。 []int64{1,2,3}
所需的类型 id 存储在数组 types 中，例如。 []int64{1,2,3}
所需的导演 id 存储在数组 directors 中，例如。 []int64{1,2,3}
所需的作者 id 存储在数组 writers 中，例如。 []int64{1,2,3}

您可以根据需要修改函数签名以根据需要接受这些参数。

可能的解决方案 1

gorm 允许您使用 raw-sql

例如。

result := db.raw(sql,categories,types,directors,writers).scan(&movies)
err := result.error

可能的解决方案2

gorm 允许您使用其 query builder 构建自己的 joins，如果您想优化在数据库上运行的查询，这可能会很有用。例如，如果您的 directors 数组为空，但 categories 不为空，则您只能查询 categories 关系。此外，您可以使用构建器对象微调查询

例如。

var query = db.Table("movies") //or you could use db.Model(&Movie{})

query = query.Select("movies.*")

// begin querying relations
if( len(categories)>0 ){
    query = query.Joins("LEFT JOIN movie_categories on movie_categories.movie_id = movies.id");
}

if( len(categories)>0 ){
    query = query.Where("movie_categories.category_id IN (?)",categories);
}
//end querying relations
//you could repeat this for each relation

//With the Gorm query builder object store in query you can make other modifications before executing the query

result := query.Scan(&movies)
err := result.Error