| | |
| | | package com.trafficaudit.holiday.service; |
| | | |
| | | import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; |
| | | import com.trafficaudit.holiday.HolidayConstants; |
| | | import com.trafficaudit.holiday.HolidayNumberUtils; |
| | | import com.trafficaudit.holiday.entity.HolidayVehicleAdjustment; |
| | | import com.trafficaudit.holiday.entity.HolidayVehicleFlow; |
| | | import com.trafficaudit.holiday.mapper.HolidayVehicleFlowMapper; |
| | | import com.trafficaudit.holiday.mapper.HolidayVehicleAdjustmentMapper; |
| | | import org.springframework.stereotype.Service; |
| | | import org.springframework.transaction.annotation.Transactional; |
| | |
| | | import java.math.BigDecimal; |
| | | import java.time.LocalDate; |
| | | import java.time.LocalDateTime; |
| | | import java.util.HashMap; |
| | | import java.util.HashSet; |
| | | import java.util.List; |
| | | import java.util.Map; |
| | | import java.util.Set; |
| | | |
| | | @Service |
| | | public class HolidayVehicleAdjustmentService { |
| | |
| | | |
| | | @Resource |
| | | private HolidayVehicleAdjustmentMapper adjustmentMapper; |
| | | @Resource |
| | | private HolidayVehicleFlowMapper vehicleFlowMapper; |
| | | |
| | | public List<HolidayVehicleAdjustment> list(Integer year) { |
| | | return adjustmentMapper.selectList(new LambdaQueryWrapper<HolidayVehicleAdjustment>() |
| | | List<HolidayVehicleAdjustment> rows = adjustmentMapper.selectList( |
| | | new LambdaQueryWrapper<HolidayVehicleAdjustment>() |
| | | .eq(year != null, HolidayVehicleAdjustment::getYear, year) |
| | | .orderByDesc(HolidayVehicleAdjustment::getYear) |
| | | .orderByAsc(HolidayVehicleAdjustment::getFlowDate)); |
| | | if (rows.isEmpty()) return rows; |
| | | |
| | | Set<Integer> years = new HashSet<>(); |
| | | for (HolidayVehicleAdjustment row : rows) { |
| | | if (row.getYear() != null) years.add(row.getYear()); |
| | | } |
| | | if (years.isEmpty()) return rows; |
| | | |
| | | List<HolidayVehicleFlow> sourceRows = vehicleFlowMapper.selectList( |
| | | new LambdaQueryWrapper<HolidayVehicleFlow>() |
| | | .eq(HolidayVehicleFlow::getRoadCategory, HolidayConstants.ROAD_NATIONAL_PROVINCIAL) |
| | | .eq(HolidayVehicleFlow::getVehicleType, HolidayConstants.VEHICLE_MEDIUM_SMALL_PASSENGER) |
| | | .in(HolidayVehicleFlow::getYear, years)); |
| | | Map<String, BigDecimal> originalValues = new HashMap<>(); |
| | | for (HolidayVehicleFlow sourceRow : sourceRows) { |
| | | if (sourceRow.getYear() == null || sourceRow.getFlowDate() == null || sourceRow.getVehicleCount() == null) { |
| | | continue; |
| | | } |
| | | originalValues.put(sourceKey(sourceRow.getYear(), sourceRow.getFlowDate()), sourceRow.getVehicleCount()); |
| | | } |
| | | for (HolidayVehicleAdjustment row : rows) { |
| | | if (row.getYear() != null && row.getFlowDate() != null) { |
| | | row.setOriginalValue(originalValues.get(sourceKey(row.getYear(), row.getFlowDate()))); |
| | | } |
| | | } |
| | | return rows; |
| | | } |
| | | |
| | | public BigDecimal findOriginalValue(Integer year, LocalDate flowDate) { |
| | | if (year == null || flowDate == null || flowDate.getYear() != year) { |
| | | throw new IllegalArgumentException("年份和日期不正确"); |
| | | } |
| | | HolidayVehicleFlow sourceRow = vehicleFlowMapper.selectOne( |
| | | new LambdaQueryWrapper<HolidayVehicleFlow>() |
| | | .eq(HolidayVehicleFlow::getYear, year) |
| | | .eq(HolidayVehicleFlow::getFlowDate, flowDate) |
| | | .eq(HolidayVehicleFlow::getRoadCategory, HolidayConstants.ROAD_NATIONAL_PROVINCIAL) |
| | | .eq(HolidayVehicleFlow::getVehicleType, HolidayConstants.VEHICLE_MEDIUM_SMALL_PASSENGER) |
| | | .last("limit 1")); |
| | | return sourceRow == null ? null : sourceRow.getVehicleCount(); |
| | | } |
| | | |
| | | @Transactional |
| | |
| | | adjustmentMapper.deleteById(id); |
| | | } |
| | | |
| | | private String sourceKey(Integer year, LocalDate flowDate) { |
| | | return year + "|" + flowDate; |
| | | } |
| | | |
| | | private Long longValue(Object value) { |
| | | if (value == null || value.toString().trim().isEmpty()) return null; |
| | | return Long.valueOf(value.toString().trim()); |