package com.whyc.service;
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import com.whyc.dto.Response;
|
import com.whyc.mapper.BaoJiGroupBattGroupMapper;
|
import com.whyc.mapper.BattInfMapper;
|
import com.whyc.pojo.BaoJiGroup;
|
import com.whyc.pojo.BaojiGroupBattGroup;
|
import com.whyc.pojo.Battinf;
|
import com.whyc.pojo.UserInf;
|
import org.assertj.core.util.Lists;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
|
import javax.annotation.Resource;
|
import java.sql.Wrapper;
|
import java.util.LinkedHashMap;
|
import java.util.LinkedList;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.stream.Collectors;
|
|
@Service
|
@Transactional
|
public class BaoJiGroupBattGroupService {
|
|
@Resource
|
private BaoJiGroupBattGroupMapper mapper;
|
|
@Resource
|
private BattInfMapper battInfMapper;
|
|
public Response<List<String>> searchStationName1InGroup(int userId) {
|
|
List<String> list = mapper.searchStationName1InGroup(userId);
|
|
return new Response<List<String>>().set(1, list);
|
}
|
|
public Response<List<Battinf>> searchStationByStationName1(String stationName1,int userId) {
|
|
List<Battinf> list = mapper.searchStationByStationName1(stationName1,userId);
|
|
return new Response<List<Battinf>>().set(1, list);
|
}
|
|
public Response<List<Battinf>> searchBattGroupIdInGroup(UserInf userInf) {
|
|
List<Battinf> list = mapper.searchBattGroupIdInGroup(userInf);
|
|
return new Response<List<Battinf>>().set(1, list);
|
}
|
|
public Map<String, List<Battinf>> getStationList(int baoJiGroupId) {
|
Map<String,List<Battinf>> resultMap = new LinkedHashMap<>();
|
//查询已添加的机房站点列表
|
List<Battinf> stationList = mapper.getStationList(baoJiGroupId);
|
//查询未添加的机房站点列表
|
/* QueryWrapper<Battinf> wrapper = Wrappers.query();
|
wrapper.select("distinct StationId,StationName");
|
List<Battinf> allStationList = battInfMapper.selectList(wrapper);*/
|
List<Battinf> allStationList = battInfMapper.getStationListOfBattPower();
|
List<Battinf> notRelatedStationList = allStationList.stream().filter(station -> !stationList.contains(station)).collect(Collectors.toList());
|
|
resultMap.put("stationsAdded",stationList);
|
resultMap.put("stationsNotAdded",notRelatedStationList);
|
|
return resultMap;
|
}
|
|
public void updateStationList(List<BaojiGroupBattGroup> baoJiGroupBattGroupList, int operationFlag) {
|
if(operationFlag==1){
|
//通过stationId查询出所有的battGroupId,再插入
|
List<BaojiGroupBattGroup> newBattGroupList = new LinkedList<>();
|
baoJiGroupBattGroupList.stream().forEach(baojiGroupBattGroup -> {
|
QueryWrapper<Battinf> wrapper = Wrappers.query();
|
wrapper.select("BattGroupId").eq("stationId",baojiGroupBattGroup.getStationId());
|
List<Battinf> battGroupList = battInfMapper.selectList(wrapper);
|
if(!battGroupList.isEmpty()) {
|
battGroupList.stream().forEach(battInf -> {
|
newBattGroupList.add(new BaojiGroupBattGroup(baojiGroupBattGroup.getBaojiGroupId(), baojiGroupBattGroup.getStationId(), battInf.getBattGroupId()));
|
|
});
|
}else{
|
newBattGroupList.add(new BaojiGroupBattGroup(baojiGroupBattGroup.getBaojiGroupId(), baojiGroupBattGroup.getStationId(), 0));
|
}
|
});
|
mapper.insertBatchSomeColumn(newBattGroupList);
|
}else{
|
mapper.deleteStationList(baoJiGroupBattGroupList);
|
}
|
}
|
|
public List<Battinf> getStationInfoList(Long uId, String stationName1) {
|
return mapper.getStationInfoList(uId,stationName1);
|
}
|
|
public List<BaojiGroupBattGroup> getBaoJiGroupListWithinBattGroupList() {
|
return mapper.getBaoJiGroupListWithinBattGroupList();
|
}
|
|
public List<BaojiGroupBattGroup> getBaoJiGroupListWithinBattGroupListOfUser(int userId) {
|
return mapper.getBaoJiGroupListWithinBattGroupListOfUser(userId);
|
}
|
}
|
