package com.whyc.service;
|
|
import com.baomidou.mybatisplus.core.conditions.Wrapper;
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
|
import com.whyc.dto.Response;
|
import com.whyc.mapper.*;
|
import com.whyc.pojo.db_station.PowerInf;
|
import com.whyc.pojo.db_user.Baojigroup;
|
import com.whyc.pojo.db_user.BaojigroupPower;
|
import com.whyc.pojo.db_user.BaojigroupUsr;
|
import com.whyc.pojo.db_user.User;
|
import com.whyc.util.ActionUtil;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
|
import java.util.ArrayList;
|
import java.util.LinkedHashMap;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.stream.Collectors;
|
|
@Service
|
public class BaojigroupService {
|
@Autowired(required = false)
|
private BaojigroupMapper mapper;
|
|
@Autowired(required = false)
|
private BaojigroupPowerMapper bjPowermapper;
|
|
@Autowired(required = false)
|
private BaojigroupUsrMapper bjUsrmapper;
|
|
@Autowired(required = false)
|
private UserMapper UserMapper;
|
|
@Autowired(required = false)
|
private PowerInfMapper powerInfMapper;
|
|
|
//将内置用户加入到包机组下
|
public void insertNeiZhi(Integer baojiId){
|
List<BaojigroupUsr> list=new ArrayList<>();
|
QueryWrapper wrapper=new QueryWrapper();
|
wrapper.lt("uid",99);
|
List<User> uList=UserMapper.selectList(wrapper);
|
for (User uinf:uList) {
|
BaojigroupUsr baojiusr=new BaojigroupUsr();
|
baojiusr.setUid(uinf.getId());
|
baojiusr.setBaojiGroupId(baojiId);
|
list.add(baojiusr);
|
}
|
bjUsrmapper.insertBatchSomeColumn(list);
|
}
|
//添加包机组
|
@Transactional
|
public Response addBaoji(String baojiName) {
|
//验证包机组名是否重复
|
//将内置用户加入到包机组下
|
QueryWrapper wrapper=new QueryWrapper();
|
wrapper.eq("baoji_name",baojiName);
|
wrapper.last("limit 1");
|
Baojigroup baoji=mapper.selectOne(wrapper);
|
if(baoji!=null){
|
return new Response().set(0,false,"包机组已存在");
|
}else{
|
baoji=new Baojigroup();
|
baoji.setBaojiGroupName(baojiName);
|
int bl=mapper.insert(baoji);
|
if(bl>0){
|
//将内置用户加入到包机组下
|
QueryWrapper wrapper1=new QueryWrapper();
|
wrapper1.eq("baoji_name",baojiName);
|
wrapper1.last("limit 1");
|
baoji=mapper.selectOne(wrapper1);
|
insertNeiZhi(baoji.getBaojiGroupId());
|
}
|
return new Response().set(1,bl>0,"添加包机组成功");
|
}
|
|
}
|
//编辑包机组
|
public Response updateBaoji(Integer id, String baojiName) {
|
UpdateWrapper wrapper=new UpdateWrapper();
|
wrapper.eq("id",id);
|
wrapper.set("baoji_name",baojiName);
|
int bl=mapper.update((Baojigroup) ActionUtil.objeNull,wrapper);
|
return new Response().set(1,bl>0,"编辑包机组成功");
|
}
|
//删除包机组
|
@Transactional
|
public Response delBaoji(Integer id) {
|
//先删除包机组对应人下的记录
|
UpdateWrapper wrapper1=new UpdateWrapper();
|
wrapper1.eq("baoji_id",id);
|
bjUsrmapper.delete(wrapper1);
|
//再删除包机组对应下的机房和锁
|
UpdateWrapper wrapper2=new UpdateWrapper();
|
wrapper2.eq("baoji_id",id);
|
bjPowermapper.delete(wrapper2);
|
//最后删除包机组
|
UpdateWrapper wrapper=new UpdateWrapper();
|
wrapper.eq("id",id);
|
int bl=mapper.delete(wrapper);
|
return new Response().set(1,bl>0,"删除包机组成功");
|
}
|
//查询包机组及包机组对应的用户和对应机房和锁
|
public Response getAllBaojiInf() {
|
QueryWrapper wrapper=new QueryWrapper();
|
wrapper.orderByAsc("id");
|
List<Baojigroup> baojiList=mapper.selectList(wrapper);
|
for (Baojigroup bj:baojiList) {
|
//查询包机组对应的用户信息
|
List<BaojigroupUsr> usrList=bjUsrmapper.getALlUsrByBjId(bj.getBaojiGroupId());
|
bj.setUsrList(usrList);
|
//查询包机组对应的机房和锁的信息
|
List<BaojigroupPower> powerList=bjPowermapper.getALlLockByBjId(bj.getBaojiGroupId());
|
bj.setPowerList(powerList);
|
}
|
return new Response().setII(1,baojiList!=null,baojiList,"查询包机组及包机组对应的用户和对应机房和锁");
|
}
|
//包机组添加机房和锁|移除机房和锁
|
@Transactional
|
public void updateStationAndLockList(List<BaojigroupPower> BaojigroupPowerList, int operationFlag) {
|
if(operationFlag==1){
|
bjPowermapper.insertBatchSomeColumn(BaojigroupPowerList);
|
}else{
|
bjPowermapper.deleteStationAndLockList(BaojigroupPowerList);
|
}
|
}
|
//包机组添加用户|移除用户
|
@Transactional
|
public void updateUserList(List<BaojigroupUsr> baoJiGroupUserList, int operationFlag) {
|
if(operationFlag==1){
|
bjUsrmapper.insertBatchSomeColumn(baoJiGroupUserList);
|
}else{
|
bjUsrmapper.deleteUserList(baoJiGroupUserList);
|
}
|
}
|
// 包机组列表
|
public Response<List<Baojigroup>> getBaoJiGroup() {
|
List<Baojigroup> list=mapper.selectList((Wrapper<Baojigroup>) ActionUtil.objeNull);
|
return new Response().setII(1,list!=null,list,"包机组列表");
|
}
|
//包机组已添加用户和未添加用户列表
|
public Map<String, List<User>> getUserList(int id) {
|
Map<String, List<User>> resultMap = new LinkedHashMap<>();
|
//查询包机组已添加的用户信息
|
List<User> relatedUserList = bjUsrmapper.getUserList(id);
|
//查询包机组未添加的用户信息
|
QueryWrapper wrapper=new QueryWrapper();
|
wrapper.select("uid", "uname");
|
wrapper.gt("uid",100);
|
List<User> allUserList = UserMapper.selectList(wrapper);
|
List<User> notRelatedUserList = allUserList.stream().filter(User -> !relatedUserList.contains(User)).collect(Collectors.toList());
|
|
resultMap.put("usersAdded",relatedUserList);
|
resultMap.put("usersNotAdded",notRelatedUserList);
|
|
return resultMap;
|
}
|
//已添加机房锁和未添加机房锁
|
public Map<String, List<PowerInf>> getStationList(int id) {
|
Map<String,List<PowerInf>> resultMap = new LinkedHashMap<>();
|
//查询已添加的机房站点列表
|
List<PowerInf> powerList = bjPowermapper.getPowerList(id);
|
List<PowerInf> allPowerList = powerInfMapper.getPowerList();
|
List<PowerInf> notRelatedLockList = allPowerList.stream().filter(PowerInf -> !powerList.contains(PowerInf)).collect(Collectors.toList());
|
|
resultMap.put("stationsAdded",powerList);
|
resultMap.put("stationsNotAdded",notRelatedLockList);
|
|
return resultMap;
|
}
|
//当前用户所在包机组下所有的用户(下拉)
|
public Response getBaojiUserByUid(Integer uid) {
|
//查询当前人所在包机组
|
QueryWrapper wrapper=new QueryWrapper();
|
wrapper.select("baoji_id");
|
if(uid>100){
|
wrapper.eq("uid",uid);
|
}
|
List<BaojigroupUsr> usrList=bjUsrmapper.selectList(wrapper);
|
if(usrList.size()==0){
|
return new Response().setII(1,null,null,"当前用户未加入包机组");
|
}
|
List<Integer> baojiIdList=usrList.stream().map(BaojigroupUsr::getBaojiGroupId).collect(Collectors.toList());
|
List<User> list=UserMapper.getBaojiUserByUid(baojiIdList);
|
return new Response().setII(1,list!=null,list,"当前用户所在包机组下所有的用户(下拉)");
|
}
|
|
public boolean getGroupFlag(int userId){
|
Integer flag = bjUsrmapper.getGroupFlag(userId);
|
return flag != null && flag == 1;
|
}
|
//先校验当前用户是否存在包机组
|
public int checkUserBaojiGroup(Integer id) {
|
QueryWrapper wrapper=new QueryWrapper();
|
wrapper.eq("uid",id);
|
List<BaojigroupUsr> list=bjUsrmapper.selectList(wrapper);
|
return list == null?0:1 ;
|
}
|
}
|
