promote

app/main/api/internal/logic/admin_user/admincreateuserlogic.go

@@ -0,0 +1,88 @@
+package admin_user
+
+import (
+	"context"
+
+	"tyc-server/app/main/api/internal/svc"
+	"tyc-server/app/main/api/internal/types"
+	"tyc-server/app/main/model"
+	"tyc-server/common/xerr"
+	"tyc-server/pkg/lzkit/crypto"
+
+	"github.com/pkg/errors"
+	"github.com/zeromicro/go-zero/core/logx"
+	"github.com/zeromicro/go-zero/core/stores/sqlx"
+)
+
+type AdminCreateUserLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+func NewAdminCreateUserLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminCreateUserLogic {
+	return &AdminCreateUserLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx,
+	}
+}
+
+func (l *AdminCreateUserLogic) AdminCreateUser(req *types.AdminCreateUserReq) (resp *types.AdminCreateUserResp, err error) {
+	// 检查用户名是否已存在
+	exists, err := l.svcCtx.AdminUserModel.FindOneByUsername(l.ctx, req.Username)
+	if err != nil && err != model.ErrNotFound {
+		return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建用户失败: %v", err)
+	}
+	if exists != nil {
+		return nil, errors.Wrapf(xerr.NewErrMsg("用户名已存在"), "创建用户失败")
+	}
+	password, err := crypto.PasswordHash("123456")
+	if err != nil {
+		return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "创建用户失败, 加密密码失败: %v", err)
+	}
+	// 创建用户
+	user := &model.AdminUser{
+		Username: req.Username,
+		Password: password, // 注意：实际应用中需要加密密码
+		RealName: req.RealName,
+		Status:   req.Status,
+	}
+
+	// 使用事务创建用户和关联角色
+	err = l.svcCtx.AdminUserModel.Trans(l.ctx, func(ctx context.Context, session sqlx.Session) error {
+		// 创建用户
+		result, err := l.svcCtx.AdminUserModel.Insert(ctx, session, user)
+		if err != nil {
+			return err
+		}
+		userId, err := result.LastInsertId()
+		if err != nil {
+			return err
+		}
+
+		// 创建用户角色关联
+		if len(req.RoleIds) > 0 {
+			for _, roleId := range req.RoleIds {
+				userRole := &model.AdminUserRole{
+					UserId: userId,
+					RoleId: roleId,
+				}
+				_, err = l.svcCtx.AdminUserRoleModel.Insert(ctx, session, userRole)
+				if err != nil {
+					return err
+				}
+			}
+		}
+
+		return nil
+	})
+
+	if err != nil {
+		return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建用户失败: %v", err)
+	}
+
+	return &types.AdminCreateUserResp{
+		Id: user.Id,
+	}, nil
+}

app/main/api/internal/logic/admin_user/admindeleteuserlogic.go

@@ -0,0 +1,68 @@
+package admin_user
+
+import (
+	"context"
+
+	"tyc-server/app/main/api/internal/svc"
+	"tyc-server/app/main/api/internal/types"
+	"tyc-server/common/xerr"
+
+	"github.com/pkg/errors"
+	"github.com/zeromicro/go-zero/core/logx"
+	"github.com/zeromicro/go-zero/core/stores/sqlx"
+)
+
+type AdminDeleteUserLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+func NewAdminDeleteUserLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminDeleteUserLogic {
+	return &AdminDeleteUserLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx,
+	}
+}
+
+func (l *AdminDeleteUserLogic) AdminDeleteUser(req *types.AdminDeleteUserReq) (resp *types.AdminDeleteUserResp, err error) {
+	// 检查用户是否存在
+	_, err = l.svcCtx.AdminUserModel.FindOne(l.ctx, req.Id)
+	if err != nil {
+		return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "用户不存在: %v", err)
+	}
+
+	// 使用事务删除用户和关联数据
+	err = l.svcCtx.AdminUserModel.Trans(l.ctx, func(ctx context.Context, session sqlx.Session) error {
+		// 删除用户角色关联
+		builder := l.svcCtx.AdminUserRoleModel.SelectBuilder().
+			Where("user_id = ?", req.Id)
+		roles, err := l.svcCtx.AdminUserRoleModel.FindAll(ctx, builder, "id ASC")
+		if err != nil {
+			return err
+		}
+		for _, role := range roles {
+			err = l.svcCtx.AdminUserRoleModel.Delete(ctx, session, role.Id)
+			if err != nil {
+				return err
+			}
+		}
+
+		// 删除用户
+		err = l.svcCtx.AdminUserModel.Delete(ctx, session, req.Id)
+		if err != nil {
+			return err
+		}
+
+		return nil
+	})
+
+	if err != nil {
+		return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "删除用户失败: %v", err)
+	}
+
+	return &types.AdminDeleteUserResp{
+		Success: true,
+	}, nil
+}

app/main/api/internal/logic/admin_user/admingetuserdetaillogic.go

@@ -0,0 +1,88 @@
+package admin_user
+
+import (
+	"context"
+	"errors"
+	"sync"
+
+	"tyc-server/app/main/api/internal/svc"
+	"tyc-server/app/main/api/internal/types"
+	"tyc-server/app/main/model"
+
+	"github.com/samber/lo"
+	"github.com/zeromicro/go-zero/core/logx"
+	"github.com/zeromicro/go-zero/core/mr"
+)
+
+type AdminGetUserDetailLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+func NewAdminGetUserDetailLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetUserDetailLogic {
+	return &AdminGetUserDetailLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx,
+	}
+}
+
+func (l *AdminGetUserDetailLogic) AdminGetUserDetail(req *types.AdminGetUserDetailReq) (resp *types.AdminGetUserDetailResp, err error) {
+	// 使用MapReduceVoid并发获取用户信息和角色ID
+	var user *model.AdminUser
+	var roleIds []int64
+	var mutex sync.Mutex
+	var wg sync.WaitGroup
+
+	mr.MapReduceVoid(func(source chan<- interface{}) {
+		source <- 1 // 获取用户信息
+		source <- 2 // 获取角色ID
+	}, func(item interface{}, writer mr.Writer[interface{}], cancel func(error)) {
+		taskType := item.(int)
+		wg.Add(1)
+		defer wg.Done()
+
+		if taskType == 1 {
+			result, err := l.svcCtx.AdminUserModel.FindOne(l.ctx, req.Id)
+			if err != nil {
+				cancel(err)
+				return
+			}
+			mutex.Lock()
+			user = result
+			mutex.Unlock()
+		} else if taskType == 2 {
+			builder := l.svcCtx.AdminUserRoleModel.SelectBuilder().
+				Where("user_id = ?", req.Id)
+			roles, err := l.svcCtx.AdminUserRoleModel.FindAll(l.ctx, builder, "id ASC")
+			if err != nil {
+				cancel(err)
+				return
+			}
+			mutex.Lock()
+			roleIds = lo.Map(roles, func(item *model.AdminUserRole, _ int) int64 {
+				return item.RoleId
+			})
+			mutex.Unlock()
+		}
+	}, func(pipe <-chan interface{}, cancel func(error)) {
+		// 不需要处理pipe中的数据
+	})
+
+	wg.Wait()
+
+	if user == nil {
+		return nil, errors.New("用户不存在")
+	}
+
+	return &types.AdminGetUserDetailResp{
+		Id:         user.Id,
+		Username:   user.Username,
+		RealName:   user.RealName,
+		Status:     user.Status,
+		CreateTime: user.CreateTime.Format("2006-01-02 15:04:05"),
+		UpdateTime: user.UpdateTime.Format("2006-01-02 15:04:05"),
+		RoleIds:    roleIds,
+	}, nil
+}

app/main/api/internal/logic/admin_user/admingetuserlistlogic.go

@@ -0,0 +1,149 @@
+package admin_user
+
+import (
+	"context"
+	"sync"
+
+	"tyc-server/app/main/api/internal/svc"
+	"tyc-server/app/main/api/internal/types"
+	"tyc-server/app/main/model"
+
+	"github.com/samber/lo"
+	"github.com/zeromicro/go-zero/core/logx"
+	"github.com/zeromicro/go-zero/core/mr"
+)
+
+type AdminGetUserListLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+func NewAdminGetUserListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetUserListLogic {
+	return &AdminGetUserListLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx,
+	}
+}
+
+func (l *AdminGetUserListLogic) AdminGetUserList(req *types.AdminGetUserListReq) (resp *types.AdminGetUserListResp, err error) {
+	resp = &types.AdminGetUserListResp{
+		Items: make([]types.UserListItem, 0),
+		Total: 0,
+	}
+
+	// 构建查询条件
+	builder := l.svcCtx.AdminUserModel.SelectBuilder().
+		Where("del_state = ?", 0)
+	if len(req.Username) > 0 {
+		builder = builder.Where("username LIKE ?", "%"+req.Username+"%")
+	}
+	if len(req.RealName) > 0 {
+		builder = builder.Where("real_name LIKE ?", "%"+req.RealName+"%")
+	}
+	if req.Status != -1 {
+		builder = builder.Where("status = ?", req.Status)
+	}
+
+	// 设置分页
+	offset := (req.Page - 1) * req.PageSize
+	builder = builder.OrderBy("id DESC").Limit(uint64(req.PageSize)).Offset(uint64(offset))
+
+	// 使用MapReduceVoid并发获取总数和列表数据
+	var users []*model.AdminUser
+	var total int64
+	var mutex sync.Mutex
+	var wg sync.WaitGroup
+
+	mr.MapReduceVoid(func(source chan<- interface{}) {
+		source <- 1 // 获取用户列表
+		source <- 2 // 获取总数
+	}, func(item interface{}, writer mr.Writer[*model.AdminUser], cancel func(error)) {
+		taskType := item.(int)
+		wg.Add(1)
+		defer wg.Done()
+
+		if taskType == 1 {
+			result, err := l.svcCtx.AdminUserModel.FindAll(l.ctx, builder, "id DESC")
+			if err != nil {
+				cancel(err)
+				return
+			}
+			mutex.Lock()
+			users = result
+			mutex.Unlock()
+		} else if taskType == 2 {
+			countBuilder := l.svcCtx.AdminUserModel.SelectBuilder().
+				Where("del_state = ?", 0)
+			if len(req.Username) > 0 {
+				countBuilder = countBuilder.Where("username LIKE ?", "%"+req.Username+"%")
+			}
+			if len(req.RealName) > 0 {
+				countBuilder = countBuilder.Where("real_name LIKE ?", "%"+req.RealName+"%")
+			}
+			if req.Status != -1 {
+				countBuilder = countBuilder.Where("status = ?", req.Status)
+			}
+
+			count, err := l.svcCtx.AdminUserModel.FindCount(l.ctx, countBuilder, "id")
+			if err != nil {
+				cancel(err)
+				return
+			}
+			mutex.Lock()
+			total = count
+			mutex.Unlock()
+		}
+	}, func(pipe <-chan *model.AdminUser, cancel func(error)) {
+		// 不需要处理pipe中的数据
+	})
+
+	wg.Wait()
+
+	// 并发获取每个用户的角色ID
+	var userItems []types.UserListItem
+	var userItemsMutex sync.Mutex
+
+	mr.MapReduceVoid(func(source chan<- interface{}) {
+		for _, user := range users {
+			source <- user
+		}
+	}, func(item interface{}, writer mr.Writer[[]int64], cancel func(error)) {
+		user := item.(*model.AdminUser)
+
+		// 获取用户关联的角色ID
+		builder := l.svcCtx.AdminUserRoleModel.SelectBuilder().
+			Where("user_id = ?", user.Id)
+		roles, err := l.svcCtx.AdminUserRoleModel.FindAll(l.ctx, builder, "id ASC")
+		if err != nil {
+			cancel(err)
+			return
+		}
+		roleIds := lo.Map(roles, func(item *model.AdminUserRole, _ int) int64 {
+			return item.RoleId
+		})
+
+		writer.Write(roleIds)
+	}, func(pipe <-chan []int64, cancel func(error)) {
+		for _, user := range users {
+			roleIds := <-pipe
+			item := types.UserListItem{
+				Id:         user.Id,
+				Username:   user.Username,
+				RealName:   user.RealName,
+				Status:     user.Status,
+				CreateTime: user.CreateTime.Format("2006-01-02 15:04:05"),
+				RoleIds:    roleIds,
+			}
+			userItemsMutex.Lock()
+			userItems = append(userItems, item)
+			userItemsMutex.Unlock()
+		}
+	})
+
+	resp.Items = userItems
+	resp.Total = total
+
+	return resp, nil
+}

app/main/api/internal/logic/admin_user/adminupdateuserlogic.go

@@ -0,0 +1,141 @@
+package admin_user
+
+import (
+	"context"
+
+	"tyc-server/app/main/api/internal/svc"
+	"tyc-server/app/main/api/internal/types"
+	"tyc-server/app/main/model"
+	"tyc-server/common/xerr"
+
+	"github.com/pkg/errors"
+	"github.com/samber/lo"
+	"github.com/zeromicro/go-zero/core/logx"
+	"github.com/zeromicro/go-zero/core/stores/sqlx"
+)
+
+type AdminUpdateUserLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+func NewAdminUpdateUserLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminUpdateUserLogic {
+	return &AdminUpdateUserLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx,
+	}
+}
+
+func (l *AdminUpdateUserLogic) AdminUpdateUser(req *types.AdminUpdateUserReq) (resp *types.AdminUpdateUserResp, err error) {
+	// 检查用户是否存在
+	user, err := l.svcCtx.AdminUserModel.FindOne(l.ctx, req.Id)
+	if err != nil {
+		return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "用户不存在: %v", err)
+	}
+
+	// 检查用户名是否重复
+	if req.Username != nil && *req.Username != user.Username {
+		exists, err := l.svcCtx.AdminUserModel.FindOneByUsername(l.ctx, *req.Username)
+		if err != nil && err != model.ErrNotFound {
+			return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新用户失败: %v", err)
+		}
+		if exists != nil {
+			return nil, errors.Wrapf(xerr.NewErrMsg("用户名已存在"), "更新用户失败")
+		}
+	}
+
+	// 更新用户信息
+	if req.Username != nil {
+		user.Username = *req.Username
+	}
+	if req.RealName != nil {
+		user.RealName = *req.RealName
+	}
+	if req.Status != nil {
+		user.Status = *req.Status
+	}
+
+	// 使用事务更新用户和关联角色
+	err = l.svcCtx.AdminUserModel.Trans(l.ctx, func(ctx context.Context, session sqlx.Session) error {
+		// 更新用户
+		_, err = l.svcCtx.AdminUserModel.Update(ctx, session, user)
+		if err != nil {
+			return err
+		}
+
+		// 只有当RoleIds不为nil时才更新角色关联
+		if req.RoleIds != nil {
+			// 1. 获取当前关联的角色ID
+			builder := l.svcCtx.AdminUserRoleModel.SelectBuilder().
+				Where("user_id = ?", req.Id)
+			currentRoles, err := l.svcCtx.AdminUserRoleModel.FindAll(ctx, builder, "id ASC")
+			if err != nil {
+				return err
+			}
+
+			// 2. 转换为map便于查找
+			currentRoleMap := make(map[int64]*model.AdminUserRole)
+			for _, role := range currentRoles {
+				currentRoleMap[role.RoleId] = role
+			}
+
+			// 3. 检查新的角色ID是否存在
+			for _, roleId := range req.RoleIds {
+				exists, err := l.svcCtx.AdminRoleModel.FindOne(ctx, roleId)
+				if err != nil || exists == nil {
+					return errors.Wrapf(xerr.NewErrMsg("角色不存在"), "角色ID: %d", roleId)
+				}
+			}
+
+			// 4. 找出需要删除和新增的关联
+			var toDelete []*model.AdminUserRole
+			var toInsert []int64
+
+			// 需要删除的：当前存在但新列表中没有的
+			for roleId, userRole := range currentRoleMap {
+				if !lo.Contains(req.RoleIds, roleId) {
+					toDelete = append(toDelete, userRole)
+				}
+			}
+
+			// 需要新增的：新列表中有但当前不存在的
+			for _, roleId := range req.RoleIds {
+				if _, exists := currentRoleMap[roleId]; !exists {
+					toInsert = append(toInsert, roleId)
+				}
+			}
+
+			// 5. 删除需要移除的关联
+			for _, userRole := range toDelete {
+				err = l.svcCtx.AdminUserRoleModel.Delete(ctx, session, userRole.Id)
+				if err != nil {
+					return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "删除用户角色关联失败: %v", err)
+				}
+			}
+
+			// 6. 添加新的关联
+			for _, roleId := range toInsert {
+				userRole := &model.AdminUserRole{
+					UserId: req.Id,
+					RoleId: roleId,
+				}
+				_, err = l.svcCtx.AdminUserRoleModel.Insert(ctx, session, userRole)
+				if err != nil {
+					return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "添加用户角色关联失败: %v", err)
+				}
+			}
+		}
+
+		return nil
+	})
+
+	if err != nil {
+		return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新用户失败: %v", err)
+	}
+
+	return &types.AdminUpdateUserResp{
+		Success: true,
+	}, nil
+}

app/main/api/internal/logic/admin_user/adminuserinfologic.go

@@ -0,0 +1,67 @@
+package admin_user
+
+import (
+	"context"
+
+	"tyc-server/app/main/api/internal/svc"
+	"tyc-server/app/main/api/internal/types"
+	"tyc-server/common/ctxdata"
+	"tyc-server/common/xerr"
+
+	"github.com/Masterminds/squirrel"
+	"github.com/pkg/errors"
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+type AdminUserInfoLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+func NewAdminUserInfoLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminUserInfoLogic {
+	return &AdminUserInfoLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx,
+	}
+}
+
+func (l *AdminUserInfoLogic) AdminUserInfo(req *types.AdminUserInfoReq) (resp *types.AdminUserInfoResp, err error) {
+	userId, err := ctxdata.GetUidFromCtx(l.ctx)
+	if err != nil {
+		return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户ID失败, %+v", err)
+	}
+
+	user, err := l.svcCtx.AdminUserModel.FindOne(l.ctx, userId)
+	if err != nil {
+		return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户ID信息失败, %+v", err)
+	}
+	// 获取权限
+	adminUserRoleBuilder := l.svcCtx.AdminUserRoleModel.SelectBuilder().Where(squirrel.Eq{"user_id": user.Id})
+	permissions, err := l.svcCtx.AdminUserRoleModel.FindAll(l.ctx, adminUserRoleBuilder, "role_id DESC")
+	if err != nil {
+		return nil, errors.Wrapf(xerr.NewErrMsg("获取权限失败"), "用户登录, 获取权限失败, 用户名: %s", user.Username)
+	}
+
+	// 获取角色ID数组
+	roleIds := make([]int64, 0)
+	for _, permission := range permissions {
+		roleIds = append(roleIds, permission.RoleId)
+	}
+
+	// 获取角色名称
+	roles := make([]string, 0)
+	for _, roleId := range roleIds {
+		role, err := l.svcCtx.AdminRoleModel.FindOne(l.ctx, roleId)
+		if err != nil {
+			continue
+		}
+		roles = append(roles, role.RoleCode)
+	}
+	return &types.AdminUserInfoResp{
+		Username: user.Username,
+		RealName: user.RealName,
+		Roles:    roles,
+	}, nil
+}