|
|
@@ -6,6 +6,7 @@ import com.alibaba.excel.read.metadata.ReadSheet;
|
|
|
import com.alibaba.fastjson.JSON;
|
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
|
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
|
|
+import com.google.common.collect.Maps;
|
|
|
import com.sckw.core.common.enums.enums.DictEnum;
|
|
|
import com.sckw.core.common.enums.enums.DictTypeEnum;
|
|
|
import com.sckw.core.common.enums.enums.ErrorCodeEnum;
|
|
|
@@ -55,6 +56,7 @@ import org.springframework.web.multipart.MultipartFile;
|
|
|
|
|
|
import java.io.InputStream;
|
|
|
import java.util.*;
|
|
|
+import java.util.function.Function;
|
|
|
import java.util.stream.Collectors;
|
|
|
|
|
|
/**
|
|
|
@@ -1202,13 +1204,30 @@ public class KwfDriverService {
|
|
|
if (CollectionUtils.isEmpty(associatedTrackList)) {
|
|
|
return Collections.emptyList();
|
|
|
}
|
|
|
+ //查询轴数表
|
|
|
+ Set<Integer> truckAxleIds = associatedTrackList.stream()
|
|
|
+ .filter(x -> Objects.nonNull(x.getTruckAxleId()))
|
|
|
+ .map(x -> Long.valueOf(x.getTruckAxleId().toString()).intValue())
|
|
|
+ .collect(Collectors.toSet());
|
|
|
+ Map<Integer, TmsTruckAxleNum> truckAxleNumMap = Maps.newHashMap();
|
|
|
+ if (CollectionUtils.isNotEmpty(truckAxleIds)) {
|
|
|
+ List<TmsTruckAxleNum> truckAxleNumList = truckAxleNumRepository.queryByIds(truckAxleIds);
|
|
|
+ if (CollectionUtils.isNotEmpty(truckAxleNumList)) {
|
|
|
+ truckAxleNumMap = truckAxleNumList.stream()
|
|
|
+ .collect(Collectors.toMap(TmsTruckAxleNum::getId, Function.identity(), (key1, key2) -> key1));
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ Map<Integer, TmsTruckAxleNum> finalTruckAxleNumMap = truckAxleNumMap;
|
|
|
List<KwfDriverAssociatedTrackVO> associatedTrackVOList = associatedTrackList.stream().map(truck ->{
|
|
|
KwfDriverAssociatedTrackVO vo = new KwfDriverAssociatedTrackVO();
|
|
|
TmsTruckAxleNum carAxisInfo = getCarAxisInfo(truck.getTruckAxleId());
|
|
|
vo.setId(truck.getId());
|
|
|
vo.setTruckNo(truck.getTruckNo());
|
|
|
vo.setTruckAxleName(carAxisInfo.getName());
|
|
|
- vo.setLegalLoad(truck.getLegalLoad());
|
|
|
+ TmsTruckAxleNum tmsTruckAxleNum = finalTruckAxleNumMap.getOrDefault(truck.getTruckAxleId(), new TmsTruckAxleNum());
|
|
|
+ vo.setLegalLoad(tmsTruckAxleNum.getLegalLoad());
|
|
|
vo.setStatus(truck.getStatus());
|
|
|
return vo;
|
|
|
}).collect(Collectors.toList());
|