获取 Golang gorm 关联数据的另一种方法

在使用 GORM 查询关联数据时，可以通过实现自定义数据类型并将其用于查询来获取关联数据。自定义数据类型必须实现 scanner 和 valuer 接口，以便 GORM 知道如何接收/保存到数据库中。此方法可以避免使用预加载，从而提高性能。

我是 gorm 和 golang 的新手，我正在处理一个查询来获取用户与对话中的用户的对话。我的查询按预期工作，但我的问题是它没有带来用户的关系。

这是我的对话模型：

type conversation struct {
    id       uint   `gorm:"primarykey"`
    users    []user `gorm:"many2many:user_has_conversations;"`
    messages []conversationmessage
}

这是我对数据库的请求：

var conversations []model.conversation

result := r.db.debug().scopes(
        pagination.paginate,
        scope.orderbyiddesctable(conversationstable),
    ).joins(
        "inner join `user_has_conversations` on `user_has_conversations`.`conversation_id` = `conversations`.`id`",
    ).joins(
        "inner join `users` on `users`.`id` = `user_has_conversations`.`user_id` and `users`.`id` = ?",
        userid,
    ).find(&conversations)

这个查询工作正常，可以正确地带来用户的对话，但不会给我返回关系中的用户，并且当您查看它运行的查询时，不会带来用户字段。

执行的查询：

select `conversations`.`id` 
from `conversations` 
inner join `user_has_conversations` on `user_has_conversations`.`conversation_id` = `conversations`.`id` 
inner join `users` on `users`.`id` = `user_has_conversations`.`user_id` and `users`.`id` = 1 
order by `conversations`.`id` desc limit 10;

在查询中可以看到查询是正确的，但是没有带有关系字段的select。

如果我也将预加载与内部联接一起使用，确实会给我提供正确的解决方案，但当我已经联接对用户进行预加载以再次获取用户时，我认为这对性能没有好处。我希望能够通过连接来获取它，而不是使用预加载。

这个示例有效，但如果有办法的话，我想在没有预加载的情况下使用它。

result := r.db.Debug().Scopes(
        pagination.Paginate,
        scope.OrderByIDDescTable(conversationsTable),
    ).Preload("Users").Joins(
        "INNER JOIN `user_has_conversations` ON `user_has_conversations`.`conversation_id` = `conversations`.`id`",
    ).Joins(
        "INNER JOIN `users` ON `users`.`id` = `user_has_conversations`.`user_id` AND `users`.`id` = ?",
        UserID,
    ).Find(&conversations)

解决方案

您可以通过为 gorm 实现自定义数据类型并在查询中使用它来从数据库接收 users 切片（如下）来实现这一点。

自定义的数据类型必须实现 scanner 和 valuer 接口，以便 gorm 知道如何接收/保存到数据库中。 https://gorm.io/docs/data_types.html

type users []user

func (u users) value() (driver.value, error) {
    return json.marshal(u)
}

func (u *users) scan(src interface{}) error {
    if bytes, ok := src.([]byte); ok {
        return json.unmarshal(bytes, u)
    }
    return errors.new(fmt.sprint("failed to unmarshal json from db", src))
}

所以你的 conversation 类型应该更新如下：

type conversation struct {
    id       uint  `gorm:"primarykey"`
    users    users `gorm:"->"`
    messages []conversationmessage
}

为了能够查询对话中所有参与的用户，我们需要使用 jsonb_agg 和 jsonb_build_object 构建 json 对象

var conversations []model.Conversation

result := r.db.Debug().Scopes(
        pagination.Paginate,
        scope.OrderByIDDescTable(conversationsTable),
    ).
    Select(conversationsTable + ".*," +
    "jsonb_agg(
        distinct jsonb_build_object(
            'id',   u.id,
            'name', u.name,
            //...
        )
    ) as users").
    Joins("INNER JOIN `user_has_conversations` ON `user_has_conversations`.`conversation_id` = `conversations`.`id`").
    Joins("INNER JOIN `users` as u ON `u`.`id` = `user_has_conversations`.`user_id` AND `u`.`id` = ?", UserID).
    Find(&conversations)

好了，本文到此结束，带大家了解了《获取 Golang gorm 关联数据的另一种方法》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！