为了账号安全,请及时绑定邮箱和手机立即绑定

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

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

Go
蓝山帝景 2022-07-18 16:58:20
我发现了一些理解文档的问题。我有 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": "h@go.com",        "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": "s@go.com",        "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 usersJOIN user_roles ON users.id = user_roles.user_idJOIN roles ON user_roles.roles_id = roles.idWHERE user_roles.roles_id = 1
查看完整描述

1 回答

?
拉风的咖菲猫

TA贡献1995条经验 获得超2个赞

编辑:

根据以下所有答案,这里需要发生以下事情:

  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,您也希望查看用户可以拥有的所有角色。


您可以使用JoinsandWhere来构建您已经使用过的类似查询。A JOINwith the rolestable 不是必需的,因为您在user_roles表格中有所需的信息。


查看完整回答
反对 回复 2022-07-18
  • 1 回答
  • 0 关注
  • 121 浏览
慕课专栏
更多

添加回答

举报

0/150
提交
取消
意见反馈 帮助中心 APP下载
官方微信