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.
72 lines
1.8 KiB
72 lines
1.8 KiB
package repository |
|
|
|
import ( |
|
"gofaster/internal/auth/model" |
|
baseModel "gofaster/internal/shared/model" |
|
"gofaster/internal/shared/repository" |
|
|
|
"gorm.io/gorm" |
|
) |
|
|
|
type RoleRepo struct { |
|
repository.BaseRepo |
|
} |
|
|
|
func NewRoleRepo(db *gorm.DB) *RoleRepo { |
|
return &RoleRepo{BaseRepo: *repository.NewBaseRepo(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 |
|
} |
|
|
|
// internal/repository/role_repo.go |
|
func (r *RoleRepo) FindByName(name string) (*model.Role, error) { |
|
var role model.Role |
|
if err := r.DB().Where("name = ?", name).First(&role).Error; err != nil { |
|
return nil, err |
|
} |
|
return &role, nil |
|
} |
|
|
|
func (r *RoleRepo) AssignPermissions(roleID uint, permissions []model.Permission) error { |
|
var permissionIDs []uint |
|
for _, p := range permissions { |
|
permissionIDs = append(permissionIDs, p.ID) |
|
} |
|
return r.DB().Model(&model.Role{BaseModel: baseModel.BaseModel{ID: roleID}}).Association("Permissions").Replace(permissions) |
|
}
|
|
|