Как отфильтровать пользователей по ролям в 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": "[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"`
}

Запрос 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(&users)

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

person Community    schedule 09.12.2020    source источник
comment
в основном, вы хотите загрузить Roles в каждый User объект и отфильтровать пользователей по идентификатору роли?   -  person Emin Laletovic    schedule 09.12.2020
comment
@EminLaletovic да, именно так. К настоящему времени, как вы можете видеть из json, у меня есть список ролей как, потому что я определяю роли как массив, я думаю.   -  person    schedule 09.12.2020


Ответы (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(&users)

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

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

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

person Emin Laletovic    schedule 09.12.2020
comment
Я редактирую запрос в соответствии со своими потребностями, и ответ нулевой. Также, возможно, я вас не понял, потому что var роли и var roleID сказали, что они не используются, и IDE помечает их как ошибку. - person ; 09.12.2020
comment
по какому идентификатору роли вы хотите фильтровать пользователей? вы передаете его этому методу конечной точки или это жестко запрограммировано? - person Emin Laletovic; 09.12.2020
comment
У меня есть поле для идентификатора роли в таблице ролей с идентификатором = 1, которое я назначаю пользователю - person ; 09.12.2020
comment
если ID роли равен 1, то жестко запрограммируйте его. Просто измените его на var roleID int = 1 или roleID := 1. - person Emin Laletovic; 09.12.2020
comment
Теперь я понял. Он по-прежнему дает мне нулевой ответ, что для меня очень странно. - person ; 09.12.2020
comment
Я предположил, что это ваш метод конечной точки REST API и что вы хотите передать ему roleID. Если это так, его нужно будет извлечь из gin.Context. - person Emin Laletovic; 09.12.2020
comment
что такое null? Roles внутри объекта User? - person Emin Laletovic; 09.12.2020
comment
вы также можете попробовать определить Roles в объекте User как Roles []Roles json:"roles" gorm:"constraint:OnUpdate:CASCADE,OnDelete:SET NULL;many2many:user_roles;", чтобы увидеть, будет ли он предварительно загружать данные. - person Emin Laletovic; 09.12.2020
comment
Я по-прежнему получаю нулевой ответ, я редактирую свой вопрос, чтобы вы могли видеть текущую версию. Не могли бы вы взглянуть? Я также думал сделать что-то вроде Raw Query, чтобы он работал с моим запросом sql, но я не могу найти, как использовать Joins с Raw - person ; 10.12.2020
comment
Просто для подтверждения, вы имеете в виду поле Roles внутри объекта User, которое все еще является nil? Или пользователи не найдены и переменная users равна nil? - person Emin Laletovic; 10.12.2020
comment
Да, я сослался на роли внутри структуры User, и ответ, который я получил от конечной точки, нулевой. Также у меня есть несколько пользователей в таблице, но только что получил null - person ; 10.12.2020
comment
Я меняю ответ на основе вашей информации, пожалуйста, скажите мне, работает ли он или какую ошибку возвращает, если есть. - person Emin Laletovic; 10.12.2020
comment
Большое спасибо! Возвращает то, что мне нужно :) Принимаю ваш ответ - person ; 10.12.2020
comment
отлично, рад слышать, что работает :) - person Emin Laletovic; 10.12.2020