使用 Gorm 在用户表中实现角色过滤功能

“纵有疾风来，人生不言弃”，这句话送给正在学习Golang的朋友们，也希望在阅读本文《使用 Gorm 在用户表中实现角色过滤功能》后，能够真的帮助到大家。我也会在后续的文章中，陆续更新Golang相关的技术文章，有好的建议欢迎大家在评论留言，非常感谢！

我发现理解文档时存在一些问题。我有 go 的 rest api，我尝试创建一个端点，在该端点中我只需要按角色提取用户。我尝试了不同的解决方案，但无法实现我所需要的。这是我所做的，我不知道如何继续，所以我只能按确切的角色来吸引用户。如果有人可以帮助我，我将不胜感激，因为我在文档中找不到更高级的内容。

来自此端点的 json 响应是：

[
    {
        "id": 1,
        "createdat": "2020-12-09t14:40:55.171011+02:00",
        "updatedat": "2020-12-09t14:40:55.175537+02:00",
        "deletedat": null,
        "email": "[email protected]",
        "password": "$2a$14$kn2waonfecaribw0xeajke.okeulcpdhveuk",
        "bearer": "eyjhbgcioijiuzi1niis2lhdci6mtywnjmwnteznn0.j2wbp8eca9tebp6l73qz1ozmo02dwqy9vtyst0fil4c",
        "first_name": "h",
        "last_name": "pro",
        "phone": "353456",
        "salesforce_id": "sfsdddfsdf",
        "webflow_id": "wfwfwfaawfw",
        "roles": null
    },
    {
        "id": 2,
        "createdat": "2020-12-09t14:40:55.171011+02:00",
        "updatedat": "2020-12-09t14:40:55.175537+02:00",
        "deletedat": null,
        "email": "[email protected]",
        "password": "$2waonfecaribw0xeajke.okeulcpdhveuk",
        "bearer": "eyjhbgiis2lhdci6mtywnjmwnteznn0.j2wbp8eca9tebp6l73qz1ozmo02dwqy9vty0fil4c",
        "first_name": "s",
        "last_name": "test",
        "phone": "3556",
        "salesforce_id": "sfsdf",
        "webflow_id": "wfwfwfw",
        "roles": null
    }
]

用户结构：

type user struct {
    gorm.model
    email        string  `json:"email"`
    password     string  `json:"password"`
    token        string  `json:"bearer"`
    firstname    string  `json:"first_name"`
    lastname     string  `json:"last_name"`
    phone        string  `json:"phone"`
    salesforceid string  `json:"salesforce_id"`
    webflowid    string  `json:"webflow_id"`
    roles        []*roles `gorm:"constraint:onupdate:cascade,ondelete:set null;many2many:user_roles;"`
}

角色结构：

type roles struct {
    gorm.model
    name string `json:"name"`
    users []*user `gorm:"many2many:user_roles"`
}

我在 postgre 中创建并适用于我的案例的 sql 查询：

select * from users
join user_roles on users.id = user_roles.user_id
join roles on user_roles.roles_id = roles.id
where user_roles.roles_id = 1

用户结构：

type user struct {
    gorm.model
    email        string  `json:"email"`
    password     string  `json:"password"`
    token        string  `json:"bearer"`
    firstname    string  `json:"first_name"`
    lastname     string  `json:"last_name"`
    phone        string  `json:"phone"`
    salesforceid string  `json:"salesforce_id"`
    webflowid    string  `json:"webflow_id"`
    roles        []roles  `json:"roles" gorm:"constraint:onupdate:cascade,ondelete:set null;many2many:user_roles;"`
}

角色的结构

type roles struct {
    gorm.model
    name string `json:"name"`
    users []*user `gorm:"many2many:user_roles"`
}

端点：

func (h *Handler) GetAllEmployees(c *gin.Context) {
    var users []models.User
    //var roles []models.Roles //the comment here is because I get error message roles is defined but never used
    var roleid int = 1
    //you need to extract roleID from your query string or request body
    
    tx := h.db.DB.
          Preload("Roles").
          Joins("INNER JOIN user_roles ON users.id = user_roles.user.id").
          Joins("INNER JOIN roles ON user_roles.roles_id = roles.id").
          Where("user_roles.roles_id = ?", roleid).
          Find(&users)

    if tx.Error != nil {
      //handle error
    }
    c.JSON(200, users)
}

解决方案

编辑：

根据下面的所有答案，这里需要发生以下事情：

1. 您需要将 roles 结构重命名为 role
2. 您需要为每个对象加载 roles
3. 按特定角色 id 过滤用户

首先，您需要将 roles 结构重命名为 role，并更改对其的引用：

角色结构

type role struct {
    gorm.model
    name string `json:"name"`
    users []user `gorm:"many2many:user_roles"`
}

用户结构

type user struct {
    gorm.model
    email        string  `json:"email"`
    password     string  `json:"password"`
    token        string  `json:"bearer"`
    firstname    string  `json:"first_name"`
    lastname     string  `json:"last_name"`
    phone        string  `json:"phone"`
    salesforceid string  `json:"salesforce_id"`
    webflowid    string  `json:"webflow_id"`
    roles        []role `gorm:"constraint:onupdate:cascade,ondelete:set null;many2many:user_roles;"`
}

接下来，稍微修改一下您的代码。这里有一个不必要的连接：

func (h *Handler) GetAllEmployeesByRoleID(c *gin.Context) {
        var users []models.User
        var roleID int = 1
        //you need to extract roleID from your query string or request body
        
        tx := h.db.DB.
              Preload("Roles").
              Joins("INNER JOIN user_roles ur ON ur.user_id = users.id").
              Where("ur.role_id = ?", roleID).
              Find(&users)

        if tx.Error != nil {
          //handle error
          c.JSON(500, tx.Error)
          return
        }  
        
        c.JSON(200, users)
}

因此，您添加 preload("roles") 来加载用户可以拥有的所有角色，正如我假设的那样，即使您按 roleid 进行过滤，您也希望查看用户可以拥有的所有角色。 p>

您可以使用 joins 和 where 构建您已使用的类似查询。不需要包含 roles 表的 join，因为 user_roles 表中有所需的信息。

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。