Как фильтровать пользователей по ролям в Gorm?

#go #go-gorm

# #Вперед #go-gorm

Вопрос:

Я обнаружил некоторые проблемы с пониманием документации. У меня есть 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": "$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"`
}
 

SQL-запрос, который я сделал и работает для моего случая в Postgre:

 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(amp;users)

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

Комментарии:

1. в принципе, вы хотите загружать Roles в каждый User объект и фильтровать пользователей по идентификатору роли?

2. @EminLaletovic да, именно. К настоящему времени, как вы можете видеть из json, я получил роли в виде списка, потому что я определяю роли как массив, я думаю.

Ответ №1:

Редактировать:

Основываясь на всех приведенных ниже ответах, здесь должны произойти следующие вещи:

  1. Вам нужно переименовать Roles struct в Role
  2. Вам нужно загрузить Roles для каждого объекта
  3. Фильтровать пользователей по определенному идентификатору роли

Во-первых, вам нужно переименовать 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(amp;users)

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

Итак, вы добавляете Preload("Roles") для загрузки все роли, которые могут быть у пользователя, поскольку я предполагал, что, даже если вы фильтруете по roleID , вы хотите увидеть все роли, которые может иметь пользователь.

Вы можете использовать Joins и Where для создания аналогичного запроса, который вы уже использовали. A JOIN с roles таблицей не требуется, поскольку у вас есть необходимая информация в user_roles таблице.

Комментарии:

1. я редактирую запрос в соответствии со своими потребностями, и ответ равен нулю. Также, возможно, я вас не понял, потому что var roles и var RoleId сказали, что они не используются, и IDE помечает их как ошибку.

2. по какому идентификатору роли вы хотите фильтровать пользователей? вы передаете его этому методу конечной точки или он жестко запрограммирован?

3. У меня есть поле для идентификатора роли в таблице ролей с id = 1, которое я присваиваю пользователю

4. если идентификатор роли равен 1, то жестко закодируйте его. Просто измените его на var roleID int = 1 или roleID := 1 .

5. Теперь я понял. Он по-прежнему выдает мне null в качестве ответа, что для меня очень странно