first commit

internal/mysqldao/admin.go

@@ -0,0 +1,103 @@
+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
+}