feat(warehouse): 添加仓库使用次数统计功能
- 在WmsMaterialCoilMapper中新增selectWarehouseIdCount方法用于统计仓库钢卷数量 - 在WmsWarehouseBo中新增orderByUseCount字段支持按使用次数排序 - 修改WmsWarehouseServiceImpl实现按使用次数或序号排序逻辑 - 创建WarehouseUseCountTask定时任务每晚1点更新仓库使用次数 - 实现定时计算各仓库中钢卷数量并更新到useCount字段功能
This commit is contained in:
@@ -0,0 +1,61 @@
|
||||
package com.klp.task;
|
||||
|
||||
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
||||
import com.klp.domain.WmsWarehouse;
|
||||
import com.klp.mapper.WmsMaterialCoilMapper;
|
||||
import com.klp.mapper.WmsWarehouseMapper;
|
||||
import lombok.RequiredArgsConstructor;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.scheduling.annotation.Scheduled;
|
||||
import org.springframework.stereotype.Component;
|
||||
import org.springframework.transaction.annotation.Transactional;
|
||||
|
||||
import java.util.HashMap;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.stream.Collectors;
|
||||
|
||||
@Slf4j
|
||||
@RequiredArgsConstructor
|
||||
@Component
|
||||
public class WarehouseUseCountTask {
|
||||
|
||||
private final WmsMaterialCoilMapper materialCoilMapper;
|
||||
private final WmsWarehouseMapper warehouseMapper;
|
||||
|
||||
@Scheduled(cron = "0 0 1 * * ?")
|
||||
@Transactional(rollbackFor = Exception.class)
|
||||
public void calculateWarehouseUseCount() {
|
||||
log.info("[仓库使用次数定时任务] 开始计算仓库使用次数");
|
||||
|
||||
LambdaQueryWrapper<WmsWarehouse> wrapper = new LambdaQueryWrapper<>();
|
||||
wrapper.select(WmsWarehouse::getWarehouseId);
|
||||
List<WmsWarehouse> warehouses = warehouseMapper.selectList(wrapper);
|
||||
|
||||
if (warehouses.isEmpty()) {
|
||||
log.info("[仓库使用次数定时任务] 未找到仓库记录");
|
||||
return;
|
||||
}
|
||||
|
||||
List<Long> warehouseIds = warehouses.stream()
|
||||
.map(WmsWarehouse::getWarehouseId)
|
||||
.collect(Collectors.toList());
|
||||
|
||||
List<Map<String, Object>> countList = materialCoilMapper.selectWarehouseIdCount(warehouseIds);
|
||||
|
||||
Map<Long, Long> countMap = new HashMap<>();
|
||||
for (Map<String, Object> item : countList) {
|
||||
Long wid = ((Number) item.get("warehouse_id")).longValue();
|
||||
Long count = ((Number) item.get("coil_count")).longValue();
|
||||
countMap.put(wid, count);
|
||||
}
|
||||
|
||||
for (WmsWarehouse warehouse : warehouses) {
|
||||
Long count = countMap.getOrDefault(warehouse.getWarehouseId(), 0L);
|
||||
warehouse.setUseCount(count.intValue());
|
||||
warehouseMapper.updateById(warehouse);
|
||||
}
|
||||
|
||||
log.info("[仓库使用次数定时任务] 完成,共更新 {} 个仓库", warehouses.size());
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user