You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
61 lines
1.5 KiB
61 lines
1.5 KiB
package repository |
|
|
|
import ( |
|
"gofaster/internal/model" |
|
|
|
"gorm.io/gorm" |
|
) |
|
|
|
type RoleRepo struct { |
|
BaseRepo |
|
} |
|
|
|
func NewRoleRepo(db *gorm.DB) *RoleRepo { |
|
return &RoleRepo{BaseRepo{db: db}} |
|
} |
|
|
|
func (r *RoleRepo) Create(role *model.Role) error { |
|
return r.db.Create(role).Error |
|
} |
|
|
|
func (r *RoleRepo) GetByID(id uint) (*model.Role, error) { |
|
var role model.Role |
|
err := r.db.Preload("Permissions").First(&role, id).Error |
|
return &role, err |
|
} |
|
|
|
func (r *RoleRepo) GetByName(name string) (*model.Role, error) { |
|
var role model.Role |
|
err := r.db.Preload("Permissions").Where("name = ?", name).First(&role).Error |
|
return &role, err |
|
} |
|
|
|
func (r *RoleRepo) Update(role *model.Role) error { |
|
return r.db.Save(role).Error |
|
} |
|
|
|
func (r *RoleRepo) Delete(id uint) error { |
|
return r.db.Delete(&model.Role{}, id).Error |
|
} |
|
|
|
func (r *RoleRepo) List(page, pageSize int) ([]model.Role, int64, error) { |
|
var roles []model.Role |
|
var count int64 |
|
|
|
err := r.db.Model(&model.Role{}).Count(&count).Error |
|
if err != nil { |
|
return nil, 0, err |
|
} |
|
|
|
offset := (page - 1) * pageSize |
|
err = r.db.Preload("Permissions").Offset(offset).Limit(pageSize).Find(&roles).Error |
|
return roles, count, err |
|
} |
|
|
|
func (r *RoleRepo) AssignPermissions(roleID uint, permissionIDs []uint) error { |
|
var permissions []model.Permission |
|
if err := r.db.Find(&permissions, permissionIDs).Error; err != nil { |
|
return err |
|
} |
|
return r.db.Model(&model.Role{BaseModel: model.BaseModel{ID: roleID}}).Association("Permissions").Replace(permissions) |
|
}
|
|
|