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;
|
|
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.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.transaction.annotation.Transactional;
|
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.util.StringUtils;
|
|
/**
|
* 运维单位 服务实现类
|
*
|
* @author xp
|
* @since 2024-03-04
|
*/
|
@Service
|
@RequiredArgsConstructor
|
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;
|
|
/**
|
* 添加
|
* @param form
|
* @return
|
*/
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public Result add(YwUnitForm form) {
|
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("添加失败");
|
}
|
|
/**
|
* 修改
|
* @param form
|
* @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);
|
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("修改失败");
|
}
|
|
/**
|
* 批量删除
|
* @param ids
|
* @return
|
*/
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public Result remove(List<String> ids) {
|
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("删除成功");
|
}
|
return Result.error("删除失败");
|
}
|
|
/**
|
* id删除
|
* @param id
|
* @return
|
*/
|
@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("删除成功");
|
}
|
return Result.error("删除失败");
|
}
|
|
/**
|
* 分页查询
|
* @param query
|
* @return
|
*/
|
@Override
|
public Result page(YwUnitQuery query) {
|
// 分页条件查询你
|
IPage<YwUnit> page = new LambdaQueryChainWrapper<>(baseMapper)
|
.eq(StringUtils.hasText(query.getUnitCode()), YwUnit::getUnitCode, query.getUnitCode())
|
.like(StringUtils.hasText(query.getUnitName()), YwUnit::getUnitName, query.getUnitName())
|
.between(Objects.nonNull(query.getStart()) && Objects.nonNull(query.getEnd()),
|
YwUnit::getCreateTime,
|
DateUtils.getDayStart(query.getStart()),
|
DateUtils.getDayEnd(query.getEnd()))
|
.orderByDesc(YwUnit::getCreateTime)
|
.page(PageUtil.getPage(query, YwUnit.class));
|
|
List<YwUnitVO> vos = page.getRecords().stream()
|
.map(
|
entity -> YwUnitVO.getVoByEntity(entity, null)
|
)
|
.collect(Collectors.toList());
|
return Result.ok().data(vos).total(page.getTotal());
|
}
|
|
/**
|
* 根据id查找
|
* @param id
|
* @return
|
*/
|
@Override
|
public Result detail(String id) {
|
|
YwUnit entity = baseMapper.selectById(id);
|
Assert.notNull(entity, "记录不存在");
|
YwUnitVO vo = YwUnitVO.getVoByEntity(entity, null);
|
return Result.ok().data(vo);
|
}
|
|
/**
|
* 列表
|
* @return
|
*/
|
@Override
|
public Result all() {
|
List<YwUnit> entities = baseMapper.selectList(null);
|
|
List<BaseSelect> vos = entities.stream()
|
.map(
|
entity -> {
|
BaseSelect baseSelect = new BaseSelect();
|
baseSelect.setId(entity.getId());
|
baseSelect.setValue(entity.getUnitName());
|
return baseSelect;
|
}
|
)
|
.collect(Collectors.toList());
|
return Result.ok().data(vos);
|
}
|
|
@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)
|
.eq(StringUtils.hasText(query.getUnitCode()), YwUnit::getUnitCode, query.getUnitCode())
|
.like(StringUtils.hasText(query.getUnitName()), YwUnit::getUnitName, query.getUnitName())
|
.between(Objects.nonNull(query.getStart()) && Objects.nonNull(query.getEnd()),
|
YwUnit::getCreateTime,
|
DateUtils.getDayStart(query.getStart()),
|
DateUtils.getDayEnd(query.getEnd()))
|
.orderByDesc(YwUnit::getCreateTime)
|
.page(PageUtil.getPage(query, YwUnit.class));
|
|
List<YwUnitVO> vos = page.getRecords().stream()
|
.map(
|
entity -> YwUnitVO.getVoByEntity(entity, null)
|
)
|
.collect(Collectors.toList());
|
return vos;
|
}
|
}
|
