121 lines
2.9 KiB
Go
121 lines
2.9 KiB
Go
|
package dao
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"cms-api/internal/model"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// GetById 根据ID获取角色
|
|||
|
|
func (dao *RoleDao) GetById(ctx context.Context, id int) (*model.Role, error) {
|
|||
|
|
var role *model.Role
|
|||
|
|
err := dao.Ctx(ctx).Where(dao.Columns().Id, id).Scan(&role)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
return role, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// List 获取角色列表
|
|||
|
|
func (dao *RoleDao) List(ctx context.Context, req *model.RoleListRequest) ([]*model.Role, int, error) {
|
|||
|
|
var (
|
|||
|
|
roles []*model.Role
|
|||
|
|
total int
|
|||
|
|
db = dao.Ctx(ctx)
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// 添加查询条件
|
|||
|
|
if req.Status >= 0 {
|
|||
|
|
db = db.Where(dao.Columns().Status, req.Status)
|
|||
|
|
}
|
|||
|
|
if req.Keyword != "" {
|
|||
|
|
db = db.Where(dao.Columns().Name+" LIKE ? OR "+dao.Columns().Description+" LIKE ?", "%"+req.Keyword+"%", "%"+req.Keyword+"%")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取总数
|
|||
|
|
count, err := db.Count()
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, 0, err
|
|||
|
|
}
|
|||
|
|
total = count
|
|||
|
|
|
|||
|
|
// 分页查询
|
|||
|
|
offset := (req.Page - 1) * req.PageSize
|
|||
|
|
err = db.Order(dao.Columns().CreatedAt + " DESC").
|
|||
|
|
Limit(req.PageSize).
|
|||
|
|
Offset(offset).
|
|||
|
|
Scan(&roles)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, 0, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return roles, total, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetAll 获取所有角色
|
|||
|
|
func (dao *RoleDao) GetAll(ctx context.Context) ([]*model.Role, error) {
|
|||
|
|
var roles []*model.Role
|
|||
|
|
err := dao.Ctx(ctx).Where(dao.Columns().Status, 1).Order(dao.Columns().CreatedAt + " ASC").Scan(&roles)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
return roles, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetByName 根据名称获取角色
|
|||
|
|
func (dao *RoleDao) GetByName(ctx context.Context, name string) (*model.Role, error) {
|
|||
|
|
var role *model.Role
|
|||
|
|
err := dao.Ctx(ctx).Where(dao.Columns().Name, name).Scan(&role)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
return role, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Create 创建角色
|
|||
|
|
func (dao *RoleDao) Create(ctx context.Context, role *model.Role) (int64, error) {
|
|||
|
|
result, err := dao.Ctx(ctx).Data(role).Insert()
|
|||
|
|
if err != nil {
|
|||
|
|
return 0, err
|
|||
|
|
}
|
|||
|
|
id, err := result.LastInsertId()
|
|||
|
|
if err != nil {
|
|||
|
|
return 0, err
|
|||
|
|
}
|
|||
|
|
return id, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Update 更新角色
|
|||
|
|
func (dao *RoleDao) Update(ctx context.Context, id int, data interface{}) error {
|
|||
|
|
_, err := dao.Ctx(ctx).Where(dao.Columns().Id, id).Data(data).Update()
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Delete 删除角色
|
|||
|
|
func (dao *RoleDao) Delete(ctx context.Context, id int) error {
|
|||
|
|
_, err := dao.Ctx(ctx).Where(dao.Columns().Id, id).Delete()
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// UpdateStatus 更新角色状态
|
|||
|
|
func (dao *RoleDao) UpdateStatus(ctx context.Context, id int, status int) error {
|
|||
|
|
_, err := dao.Ctx(ctx).Where(dao.Columns().Id, id).Data(map[string]interface{}{
|
|||
|
|
dao.Columns().Status: status,
|
|||
|
|
}).Update()
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// CheckPermission 检查权限
|
|||
|
|
func (dao *RoleDao) CheckPermission(ctx context.Context, roleId int, permission string) (bool, error) {
|
|||
|
|
var role *model.Role
|
|||
|
|
err := dao.Ctx(ctx).Where(dao.Columns().Id, roleId).Scan(&role)
|
|||
|
|
if err != nil {
|
|||
|
|
return false, err
|
|||
|
|
}
|
|||
|
|
if role == nil {
|
|||
|
|
return false, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 这里应该实现权限检查逻辑
|
|||
|
|
// 暂时返回true，实际项目中需要根据role.Permissions字段进行权限验证
|
|||
|
|
return true, nil
|
|||
|
|
}
|