104 lines
3.0 KiB
Go
104 lines
3.0 KiB
Go
package mysqldao
|
|
|
|
import (
|
|
"context"
|
|
|
|
"git.lzcwy.cn/zhengchiliu/zt-closet-manage-backend/global"
|
|
"git.lzcwy.cn/zhengchiliu/zt-closet-manage-backend/internal/mysqlmodel"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// AdminDao 管理员账号数据访问
|
|
type AdminDao struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
// NewAdminDao 创建管理员 DAO
|
|
func NewAdminDao() *AdminDao {
|
|
return &AdminDao{db: global.MysqlEngine}
|
|
}
|
|
|
|
// AdminGetByID 按主键 ID 查询管理员
|
|
func (d *AdminDao) AdminGetByID(ctx context.Context, id int64) (*mysqlmodel.AdminUser, error) {
|
|
ctx, span := global.Tracer.Start(ctx, "mysqldao.AdminGetByID")
|
|
defer span.End()
|
|
|
|
var admin mysqlmodel.AdminUser
|
|
if err := d.db.WithContext(ctx).First(&admin, id).Error; err != nil {
|
|
span.RecordError(err)
|
|
return nil, err
|
|
}
|
|
return &admin, nil
|
|
}
|
|
|
|
// AdminGetByUsername 按登录账号查询管理员（含软删除）
|
|
func (d *AdminDao) AdminGetByUsername(ctx context.Context, username string) (*mysqlmodel.AdminUser, error) {
|
|
ctx, span := global.Tracer.Start(ctx, "mysqldao.AdminGetByUsername")
|
|
defer span.End()
|
|
|
|
var admin mysqlmodel.AdminUser
|
|
if err := d.db.WithContext(ctx).Where("username = ?", username).First(&admin).Error; err != nil {
|
|
span.RecordError(err)
|
|
return nil, err
|
|
}
|
|
return &admin, nil
|
|
}
|
|
|
|
// AdminList 分页查询管理员列表（status 为 0 不筛选）
|
|
func (d *AdminDao) AdminList(ctx context.Context, status int8, page, pageSize int) ([]mysqlmodel.AdminUser, int64, error) {
|
|
ctx, span := global.Tracer.Start(ctx, "mysqldao.AdminList")
|
|
defer span.End()
|
|
|
|
var list []mysqlmodel.AdminUser
|
|
var total int64
|
|
q := d.db.WithContext(ctx).Model(&mysqlmodel.AdminUser{})
|
|
if status != 0 {
|
|
q = q.Where("status = ?", status)
|
|
}
|
|
if err := q.Count(&total).Error; err != nil {
|
|
span.RecordError(err)
|
|
return nil, 0, err
|
|
}
|
|
if err := q.Offset((page - 1) * pageSize).Limit(pageSize).Order("id DESC").Find(&list).Error; err != nil {
|
|
span.RecordError(err)
|
|
return nil, 0, err
|
|
}
|
|
return list, total, nil
|
|
}
|
|
|
|
// AdminCreate 创建管理员账号
|
|
func (d *AdminDao) AdminCreate(ctx context.Context, admin *mysqlmodel.AdminUser) error {
|
|
ctx, span := global.Tracer.Start(ctx, "mysqldao.AdminCreate")
|
|
defer span.End()
|
|
|
|
if err := d.db.WithContext(ctx).Create(admin).Error; err != nil {
|
|
span.RecordError(err)
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// AdminUpdate 按主键更新管理员字段
|
|
func (d *AdminDao) AdminUpdate(ctx context.Context, id int64, updates map[string]any) error {
|
|
ctx, span := global.Tracer.Start(ctx, "mysqldao.AdminUpdate")
|
|
defer span.End()
|
|
|
|
if err := d.db.WithContext(ctx).Model(&mysqlmodel.AdminUser{}).Where("id = ?", id).Updates(updates).Error; err != nil {
|
|
span.RecordError(err)
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// AdminSoftDelete 软删除管理员账号
|
|
func (d *AdminDao) AdminSoftDelete(ctx context.Context, id int64) error {
|
|
ctx, span := global.Tracer.Start(ctx, "mysqldao.AdminSoftDelete")
|
|
defer span.End()
|
|
|
|
if err := d.db.WithContext(ctx).Where("id = ?", id).Delete(&mysqlmodel.AdminUser{}).Error; err != nil {
|
|
span.RecordError(err)
|
|
return err
|
|
}
|
|
return nil
|
|
}
|