blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
xiangpei
2024-08-16 983b100e79431dea9a471168dae451d2f7399b76 
 ycl-server/src/main/java/com/ycl/platform/service/impl/YwUnitServiceImpl.java


@@ -1,39 +1,36 @@


package com.ycl.platform.service.impl;




import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;


import com.baomidou.mybatisplus.core.metadata.IPage;


import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;


import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;




import java.util.Collections;


import java.util.HashMap;


import java.util.List;




import com.ycl.platform.base.BaseSelect;


import com.ycl.platform.domain.entity.YwUnit;


import com.ycl.platform.domain.form.YwUnitForm;


import com.ycl.platform.domain.query.YwUnitQuery;


import com.ycl.platform.domain.vo.YwUnitVO;


import com.ycl.system.Result;


import com.ycl.platform.mapper.YwUnitMapper;


import com.ycl.platform.service.YwUnitService;


import com.ycl.system.Result;


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.model.LoginUser;


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;




import java.util.Objects;


import java.util.stream.Collectors;


import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;


import lombok.RequiredArgsConstructor;


import org.springframework.beans.BeanUtils;


import org.springframework.stereotype.Service;


import org.springframework.transaction.annotation.Transactional;


import org.springframework.util.Assert;


import org.springframework.util.StringUtils;




import java.util.*;


import java.util.stream.Collectors;




/**


 * 运维单位 服务实现类


@@ -56,8 +53,12 @@


     * @return


     */


    @Override


    @Transactional(rollbackFor = Exception.class)


    public Result add(YwUnitForm form) {


        YwUnit entity = YwUnitForm.getEntityByForm(form, null);


        Date now = new Date();


        entity.setCreateTime(now);


        entity.setUpdateTime(now);


        if(baseMapper.insert(entity) > 0) {


            // 新增运维人员账号


            SysUser sysUser = new SysUser();


@@ -86,12 +87,14 @@


     * @return


     */


    @Override


    @Transactional(rollbackFor = Exception.class)


    public Result update(YwUnitForm form) {


        YwUnit entity = baseMapper.selectById(form.getId());




        // 为空抛IllegalArgumentException，做全局异常处理


        Assert.notNull(entity, "记录不存在");


        BeanUtils.copyProperties(form, entity);


        Date now = new Date();


        entity.setUpdateTime(now);


        if (baseMapper.updateById(entity) > 0) {


            // 修改运维单位账号信息


            SysUser sysUser = sysUserMapper.selectUserByUserName(entity.getUnitAdminAccount());


@@ -111,15 +114,16 @@


     * @return


     */


    @Override


    @Transactional(rollbackFor = Exception.class)


    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());


            }


        });


        List<Long> userIds = sysUserMapper.selectUserIdByNames(


                baseMapper.selectList(


                        new LambdaQueryWrapper<YwUnit>().in(YwUnit::getId, ids))


                        .stream().map(YwUnit::getUnitAdminAccount).toList());


        if (!userIds.isEmpty()) {


            sysUserMapper.deleteUserByIds(userIds.toArray(new Long[0]));


            sysUserRoleMapper.deleteUserRoleByUserIds(userIds);


        }


        if(baseMapper.deleteBatchIds(ids) > 0) {


            return Result.ok("删除成功");


        }


@@ -207,6 +211,33 @@


    }




    @Override


    public Result workList() {


        List<YwUnitVO> entities = baseMapper.workList();


        return Result.ok().data(entities);


    }




    @Override


    public void setUnitInfo(LoginUser loginUser) {


        // 单位管理员查询


        YwUnit unit = new LambdaQueryChainWrapper<>(baseMapper)


                .eq(YwUnit::getUnitAdminAccount, loginUser.getUsername())


                .one();


        if (Objects.nonNull(unit)) {


            loginUser.setUnitAdmin(Boolean.TRUE);


            loginUser.setUnitId(unit.getId());


        } else {


            unit = baseMapper.getByUserId(loginUser.getUserId());


            if (Objects.nonNull(unit)) {


                loginUser.setUnitAdmin(Boolean.FALSE);


                loginUser.setUnitId(unit.getId());


            }


            else {


                loginUser.setUnitAdmin(Boolean.FALSE);


            }


        }


    }




    @Override


    public List<YwUnitVO> export(YwUnitQuery query) {


        // 分页条件查询你


        IPage<YwUnit> page = new LambdaQueryChainWrapper<>(baseMapper)


@@ -226,4 +257,9 @@


                .collect(Collectors.toList());


        return vos;


    }




    @Override


    public List<Map<String, Object>> ywUnitCount() {


        return baseMapper.ywUnitCount();


    }


}