运维单位绑定账号

龚焕茏

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("删除成功");
        }

 ycl-server/src/main/java/com/ycl/system/mapper/SysRoleMapper.java

@@ -1,5 +1,6 @@
package com.ycl.system.mapper;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.ycl.system.entity.SysRole;
import org.apache.ibatis.annotations.Mapper;

@@ -107,4 +108,11 @@
     * @return 结果
     */
    public int deleteRoleByIds(Long[] roleIds);

    /**
     * 根据角色编码查询角色
     * @param roleKey 角色编码
     * @return 角色
     */
    SysRole selectRoleByKey(String roleKey);
}

 ycl-server/src/main/resources/mapper/system/SysRoleMapper.xml

@@ -92,8 +92,12 @@
        <include refid="selectRoleVo"/>
         where r.role_key=#{roleKey} and r.del_flag = '0' limit 1
    </select>
	
     <insert id="insertRole" parameterType="com.ycl.system.entity.SysRole" useGeneratedKeys="true" keyProperty="roleId">

    <select id="selectRoleByKey" resultType="com.ycl.system.entity.SysRole">
        select * from sys_role where role_key = #{roleKey} and del_flag = '0' limit 1
    </select>

    <insert id="insertRole" parameterType="com.ycl.system.entity.SysRole" useGeneratedKeys="true" keyProperty="roleId">
         insert into sys_role(
             <if test="roleId != null and roleId != 0">role_id,</if>
             <if test="roleName != null and roleName != ''">role_name,</if>