第一版完成

internal/dao/user_methods.go

@@ -0,0 +1,121 @@
+package dao
+
+import (
+	"context"
+	"cms-api/internal/model"
+)
+
+// GetByAccount 根据账号获取用户
+func (dao *UserDao) GetByAccount(ctx context.Context, account string) (*model.User, error) {
+	var user *model.User
+	err := dao.Ctx(ctx).Where(dao.Columns().Account, account).Scan(&user)
+	if err != nil {
+		return nil, err
+	}
+	return user, nil
+}
+
+// GetByEmail 根据邮箱获取用户
+func (dao *UserDao) GetByEmail(ctx context.Context, email string) (*model.User, error) {
+	var user *model.User
+	err := dao.Ctx(ctx).Where(dao.Columns().Email, email).Scan(&user)
+	if err != nil {
+		return nil, err
+	}
+	return user, nil
+}
+
+// GetById 根据ID获取用户
+func (dao *UserDao) GetById(ctx context.Context, id int) (*model.User, error) {
+	var user *model.User
+	err := dao.Ctx(ctx).Where(dao.Columns().Id, id).Scan(&user)
+	if err != nil {
+		return nil, err
+	}
+	return user, nil
+}
+
+// List 获取用户列表
+func (dao *UserDao) List(ctx context.Context, req *model.UserListRequest) ([]*model.User, int, error) {
+	var (
+		users []*model.User
+		total int
+		db    = dao.Ctx(ctx)
+	)
+
+	// 添加查询条件
+	if req.RoleId > 0 {
+		db = db.Where(dao.Columns().RoleId, req.RoleId)
+	}
+	if req.Status >= 0 {
+		db = db.Where(dao.Columns().Status, req.Status)
+	}
+	if req.Keyword != "" {
+		db = db.Where(dao.Columns().Account+" LIKE ? OR "+dao.Columns().NickName+" LIKE ? OR "+dao.Columns().Email+" LIKE ?", 
+			"%"+req.Keyword+"%", "%"+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(&users)
+	if err != nil {
+		return nil, 0, err
+	}
+
+	return users, total, nil
+}
+
+// Create 创建用户
+func (dao *UserDao) Create(ctx context.Context, user *model.User) (int64, error) {
+	result, err := dao.Ctx(ctx).Data(user).Insert()
+	if err != nil {
+		return 0, err
+	}
+	id, err := result.LastInsertId()
+	if err != nil {
+		return 0, err
+	}
+	return id, nil
+}
+
+// Update 更新用户
+func (dao *UserDao) 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 *UserDao) Delete(ctx context.Context, id int) error {
+	_, err := dao.Ctx(ctx).Where(dao.Columns().Id, id).Delete()
+	return err
+}
+
+// UpdateStatus 更新用户状态
+func (dao *UserDao) 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
+}
+
+// GetCount 获取用户总数
+func (dao *UserDao) GetCount(ctx context.Context) (int, error) {
+	count, err := dao.Ctx(ctx).Count()
+	return count, err
+}
+
+// GetCountByStatus 根据状态获取用户数
+func (dao *UserDao) GetCountByStatus(ctx context.Context, status int) (int, error) {
+	count, err := dao.Ctx(ctx).Where(dao.Columns().Status, status).Count()
+	return count, err
+}