hospital-admin-api/api/service/user.go

package service

import (
	"crypto/md5"
	"encoding/hex"
	"errors"
	"github.com/gin-gonic/gin"
	"github.com/google/uuid"
	"hospital-admin-api/api/dao"
	"hospital-admin-api/api/model"
	"hospital-admin-api/api/requests"
	"hospital-admin-api/global"
	"strconv"
)

type UserService struct {
}

// AddUser 新增用户
func (r *UserService) AddUser(c *gin.Context, AddUserRequest requests.AddUser) (bool, error) {
	// 当前登陆用户id
	loginUserId := c.GetInt64("UserId")
	if loginUserId == 0 {
		return false, errors.New("新增失败")
	}

	loginRoleId := c.GetInt64("RoleId")
	if loginRoleId == 0 {
		return false, errors.New("新增失败")
	}

	// 获取当前登陆用户角色数据
	adminRoleDao := dao.AdminRoleDao{}
	adminRole, err := adminRoleDao.GetAdminRoleFirstById(loginRoleId)
	if err != nil || adminRole == nil {
		return false, errors.New("非法操作")
	}

	if adminRole.IsAdmin == 0 {
		return false, errors.New("您当前为普通用户，无法添加用户")
	}

	adminUserDao := dao.AdminUserDao{}
	AdminDeptDao := dao.AdminDeptDao{}
	AdminPostDao := dao.AdminPostDao{}

	// 检测账号名
	AdminUser, err := adminUserDao.GetAdminUserFirstByAccess(AddUserRequest.Access)
	if AdminUser != nil {
		return false, errors.New("账号名重复")
	}

	// 检测手机号
	AdminUser, err = adminUserDao.GetAdminUserFirstByPhone(AddUserRequest.Phone)
	if AdminUser != nil {
		return false, errors.New("手机号重复")
	}

	// 检测角色
	roleID, err := strconv.ParseInt(AddUserRequest.RoleID, 10, 64)
	if err != nil {
		return false, errors.New("角色错误")
	}

	adminRole, err = adminRoleDao.GetAdminRoleFirstById(roleID)
	if err != nil || adminRole == nil {
		return false, errors.New("角色错误")
	}

	// 检测部门
	deptID, err := strconv.ParseInt(AddUserRequest.DeptID, 10, 64)
	if err != nil {
		return false, errors.New("部门错误")
	}

	adminDept, err := AdminDeptDao.GetAdminDeptFirstById(deptID)
	if err != nil || adminDept == nil {
		return false, errors.New("部门错误")
	}

	// 检测岗位
	postID, err := strconv.ParseInt(AddUserRequest.PostID, 10, 64)
	if err != nil {
		return false, errors.New("岗位错误")
	}

	adminPost, err := AdminPostDao.GetAdminPostFirstById(postID)
	if err != nil || adminPost == nil {
		return false, errors.New("岗位错误")
	}

	// 处理密码
	salt := uuid.New().String()
	password := md5.Sum([]byte(AddUserRequest.Password + salt))
	// 将哈希值转换为16进制字符串
	passwordString := hex.EncodeToString(password[:])

	// 开始事务
	tx := global.Db.Begin()
	defer func() {
		if r := recover(); r != nil {
			tx.Rollback()
		}
	}()

	// 新增用户
	AdminUserModel := &model.AdminUser{
		Access:     AddUserRequest.Access,
		Password:   passwordString,
		Salt:       salt,
		Status:     AddUserRequest.Status,
		IsDeleted:  AddUserRequest.IsDeleted,
		IsDisabled: AddUserRequest.IsDisabled,
		NickName:   AddUserRequest.NickName,
		Phone:      AddUserRequest.Phone,
		Avatar:     AddUserRequest.Avatar,
		Sex:        AddUserRequest.Sex,
		Email:      AddUserRequest.Email,
		RoleID:     roleID,
		DeptID:     deptID,
		PostID:     postID,
		CreateBy:   loginUserId,
	}

	adminUser, _ := adminUserDao.AddAdminUser(tx, AdminUserModel)
	if adminUser == nil {
		tx.Rollback()
		return false, errors.New("新增失败")
	}

	tx.Commit()

	return true, nil
}

// DeleteUser 删除用户-批量
func (r *UserService) DeleteUser(c *gin.Context, DeleteUserRequest requests.DeleteUser) (bool, error) {
	// 获取当前登陆用户数据
	userId := c.GetInt64("UserId")
	if userId == 0 {
		return false, errors.New("数据错误")
	}

	AdminUserDao := dao.AdminUserDao{}
	adminUser, err := AdminUserDao.GetAdminUserFirstById(userId)
	if err != nil || adminUser == nil {
		return false, errors.New("用户数据错误")
	}

	// 获取当前登陆用户角色数据
	roleId := c.GetInt64("RoleId")
	if roleId == 0 {
		return false, errors.New("数据错误")
	}

	AdminRoleDao := dao.AdminRoleDao{}
	adminRole, err := AdminRoleDao.GetAdminRoleFirstById(roleId)
	if err != nil || adminRole == nil {
		return false, errors.New("非法操作")
	}

	if adminRole.IsAdmin == 0 {
		return false, errors.New("暂无权限，请联系管理员删除")
	}

	// 开始事务
	tx := global.Db.Begin()
	defer func() {
		if r := recover(); r != nil {
			tx.Rollback()
		}
	}()

	for _, v := range DeleteUserRequest.UserIds {
		// 将字符串转换为int64类型
		v, err := strconv.ParseInt(v, 10, 64)
		if err != nil {
			return false, errors.New("菜单错误")
		}

		if v == adminUser.UserID {
			tx.Rollback()
			return false, errors.New("非法操作，请勿删除自己")
		}

		// 检测删除用户
		adminUser, err := AdminUserDao.GetAdminUserFirstById(userId)
		if err != nil || adminUser == nil {
			tx.Rollback()
			return false, errors.New("删除失败")
		}

		// 删除用户
		err = AdminUserDao.DeleteAdminUserById(tx, v)
		if err != nil {
			tx.Rollback()
			return false, errors.New("删除失败")
		}
	}

	tx.Commit()

	return true, nil
}