如何在 Gorm 中按角色过滤用户？

Question

我在理解文档时发现了一些问题。我有 REST API 和 Go，我试图创建一个端点，我只需要按他们的角色提取用户。我尝试了不同的解决方案，但无法实现我所需要的。这是我所做的，我不确定如何继续，所以我只能按确切的角色来接收用户。如果有人能帮助我，我将不胜感激，因为我在文档中找不到更高级的东西。

来自此端点的 JSON 响应是：

[
    {
        "ID": 1,
        "CreatedAt": "2020-12-09T14:40:55.171011+02:00",
        "UpdatedAt": "2020-12-09T14:40:55.175537+02:00",
        "DeletedAt": null,
        "email": "h@go.com",
        "password": "a$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": "s@go.com",
        "password": "wAOnfecAriBW0xeAJke.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)
}

Answer 1

编辑：

根据以下所有答案，此处需要发生以下情况：

您需要将 Roles 结构重命名为 Role
您需要为每个对象加载 Roles
按特定角色 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 进行过滤，您也希望看到用户可以拥有的所有角色有。

您可以使用 Joins 和 Where 构建您已经使用过的类似查询。 JOIN 和 roles table 不是必需的，因为您在 user_roles table.

中有所需的信息

如何在 Gorm 中按角色过滤用户？

How to filter users by role in Gorm?

go

go-gorm