运维单位绑定账号

龚焕茏

2024-04-11 db0b87b36c44a5a9bc32a915769024f0c9db7c05

 ycl-server/src/main/java/com/ycl/platform/service/impl/YwUnitServiceImpl.java

@@ -1,7 +1,9 @@
package com.ycl.platform.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;

import java.util.Collections;
import java.util.HashMap;
import java.util.List;

@@ -12,10 +14,17 @@
import com.ycl.platform.domain.vo.YwUnitVO;
import com.ycl.system.Result;
import com.ycl.platform.mapper.YwUnitMapper;
import com.ycl.system.domain.SysUserRole;
import com.ycl.system.entity.SysRole;
import com.ycl.system.entity.SysUser;
import com.ycl.system.mapper.SysRoleMapper;
import com.ycl.system.mapper.SysUserMapper;
import com.ycl.system.mapper.SysUserRoleMapper;
import com.ycl.system.page.PageUtil;
import com.ycl.platform.service.YwUnitService;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.ycl.utils.DateUtils;
import com.ycl.utils.SecurityUtils;
import org.springframework.stereotype.Service;
import org.springframework.beans.BeanUtils;
import org.springframework.util.Assert;
@@ -37,6 +46,9 @@
public class YwUnitServiceImpl extends ServiceImpl<YwUnitMapper, YwUnit> implements YwUnitService {

    private final YwUnitMapper ywUnitMapper;
    private final SysUserMapper sysUserMapper;
    private final SysUserRoleMapper sysUserRoleMapper;
    private final SysRoleMapper sysRoleMapper;

    /**
     * 添加
@@ -45,9 +57,24 @@
     */
    @Override
    public Result add(YwUnitForm form) {
        // todo 增加对应账号
        YwUnit entity = YwUnitForm.getEntityByForm(form, null);
        if(baseMapper.insert(entity) > 0) {
            // 新增运维人员账号
            SysUser sysUser = new SysUser();
            sysUser.setDeptId(null);
            sysUser.setUserName(entity.getUnitAdminAccount());
            sysUser.setNickName(entity.getUnitContact());
            sysUser.setPhonenumber(entity.getUnitContactPhone());
            sysUser.setPassword(SecurityUtils.encryptPassword("123456"));
            sysUserMapper.insertUser(sysUser);
            // 获取运维人员角色
            SysRole sysRole = sysRoleMapper.selectRoleByKey("yw_user");
            if (Objects.isNull(sysRole)) { return Result.error("运维角色不存在"); }
            // 绑定账号角色关系
            SysUserRole sysUserRole = new SysUserRole();
            sysUserRole.setRoleId(sysRole.getRoleId());
            sysUserRole.setUserId(sysUser.getUserId());
            sysUserRoleMapper.batchUserRole(Collections.singletonList(sysUserRole));
            return Result.ok("添加成功");
        }
        return Result.error("添加失败");
@@ -60,13 +87,19 @@
     */
    @Override
    public Result update(YwUnitForm form) {
        // todo 修改对应账号
        YwUnit entity = baseMapper.selectById(form.getId());

        // 为空抛IllegalArgumentException，做全局异常处理
        Assert.notNull(entity, "记录不存在");
        BeanUtils.copyProperties(form, entity);
        if (baseMapper.updateById(entity) > 0) {
            // 修改运维单位账号信息
            SysUser sysUser = sysUserMapper.selectUserByUserName(entity.getUnitAdminAccount());
            if (Objects.nonNull(sysUser)) {
                sysUser.setPhonenumber(entity.getUnitContactPhone());
                sysUser.setNickName(entity.getUnitContact());
                sysUserMapper.updateUser(sysUser);
            }
            return Result.ok("修改成功");
        }
        return Result.error("修改失败");
@@ -79,6 +112,14 @@
     */
    @Override
    public Result remove(List<String> ids) {
        ids.forEach(id -> {
            // 删除运维人员账号
            SysUser sysUser = sysUserMapper.selectUserByUserName(baseMapper.selectById(id).getUnitAdminAccount());
            if (Objects.nonNull(sysUser)) {
                sysUserRoleMapper.deleteUserRoleByUserId(sysUser.getUserId());
                sysUserMapper.deleteUserById(sysUser.getUserId());
            }
        });
        if(baseMapper.deleteBatchIds(ids) > 0) {
            return Result.ok("删除成功");
        }
@@ -92,6 +133,12 @@
     */
    @Override
    public Result removeById(String id) {
        // 删除运维人员账号
        SysUser sysUser = sysUserMapper.selectUserByUserName(baseMapper.selectById(id).getUnitAdminAccount());
        if (Objects.nonNull(sysUser)) {
            sysUserRoleMapper.deleteUserRoleByUserId(sysUser.getUserId());
            sysUserMapper.deleteUserById(sysUser.getUserId());
        }
        if(baseMapper.deleteById(id) > 0) {
            return Result.ok("删除成功");
        }