package com.ruoyi.business.task;
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import com.ruoyi.business.entity.*;
|
import com.ruoyi.business.mapper.*;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.ibatis.session.ResultContext;
|
import org.apache.ibatis.session.ResultHandler;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.scheduling.annotation.Scheduled;
|
import org.springframework.stereotype.Component;
|
import org.springframework.transaction.annotation.Transactional;
|
|
import java.math.BigDecimal;
|
import java.util.ArrayList;
|
import java.util.List;
|
import java.util.concurrent.ExecutorService;
|
import java.util.concurrent.Executors;
|
|
import static com.ruoyi.business.constant.InventoryRecordConstant.OFFICIAL_INVENTORY;
|
import static com.ruoyi.business.constant.InventoryRecordConstant.PENDING_INVENTORY;
|
|
@Slf4j
|
@Component
|
public class InventorySummaryTask {
|
@Autowired
|
private PendingInventoryMapper pendingInventoryMapper;
|
@Autowired
|
private OfficialInventoryMapper officialInventoryMapper;
|
@Autowired
|
private InventorySummaryMapper inventorySummaryMapper;
|
@Autowired
|
private InputInventoryRecordMapper inputInventoryRecordMapper;
|
@Autowired
|
private OutputInventoryRecordMapper outputInventoryRecordMapper;
|
|
private final ExecutorService executor = Executors.newFixedThreadPool(5);
|
|
// 每隔10天的凌晨3点执行
|
@Scheduled(cron = "0 0 3 1/10 * ?")
|
@Transactional
|
public void updateInventorySummary() {
|
log.info("开始执行库存节点统计定时任务: " + System.currentTimeMillis());
|
log.info("更新待入库库存节点");
|
updatePendingInventory();
|
log.info("更新正式库存节点");
|
updateOfficialInventory();
|
log.info("结束执行库存节点统计定时任务: " + System.currentTimeMillis());
|
}
|
|
// 更新待入库库存节点
|
private void updatePendingInventory() {
|
pendingInventoryMapper.selectList(Wrappers.emptyWrapper(), new ResultHandler<PendingInventory>() {
|
|
@Override
|
public void handleResult(ResultContext<? extends PendingInventory> resultContext) {
|
PendingInventory pendingInventory = resultContext.getResultObject();
|
executor.submit(() -> {
|
handleInventorySummary(pendingInventory, null);
|
});
|
|
}
|
});
|
}
|
|
// 更新正式库存节点
|
private void updateOfficialInventory() {
|
officialInventoryMapper.selectList(Wrappers.emptyWrapper(), new ResultHandler<OfficialInventory>() {
|
|
@Override
|
public void handleResult(ResultContext<? extends OfficialInventory> resultContext) {
|
OfficialInventory officialInventory = resultContext.getResultObject();
|
executor.submit(() -> {
|
handleInventorySummary(null, officialInventory);
|
});
|
|
}
|
});
|
}
|
|
// 更新节点数据
|
private void handleInventorySummary(PendingInventory pendingInventory, OfficialInventory officialInventory) {
|
InventorySummary inventorySummary = null;
|
List<InputInventoryRecord> inputInventoryRecordList = new ArrayList<>();
|
List<OutputInventoryRecord> outputInventoryRecordList = new ArrayList<>();
|
|
if (pendingInventory != null) {
|
inventorySummary = inventorySummaryMapper.getInventorySummaryForUpdateByInventoryIdAndType(pendingInventory.getId(), PENDING_INVENTORY);
|
inputInventoryRecordList = inputInventoryRecordMapper.selectList(new LambdaQueryWrapper<InputInventoryRecord>()
|
.eq(InputInventoryRecord::getInventoryId, pendingInventory.getId())
|
.eq(InputInventoryRecord::getInventoryType, PENDING_INVENTORY)
|
.gt(InputInventoryRecord::getId, inventorySummary == null ? 0 : inventorySummary.getInputEndRecordId()));
|
outputInventoryRecordList = outputInventoryRecordMapper.selectList(new LambdaQueryWrapper<OutputInventoryRecord>()
|
.eq(OutputInventoryRecord::getInventoryId, pendingInventory.getId())
|
.eq(OutputInventoryRecord::getInventoryType, PENDING_INVENTORY)
|
.gt(OutputInventoryRecord::getId, inventorySummary == null ? 0 : inventorySummary.getOutputEndRecordId()));
|
|
if (inventorySummary == null) {
|
inventorySummary = new InventorySummary();
|
inventorySummary.setInventoryQuantity(BigDecimal.ZERO);
|
inventorySummary.setInventoryId(pendingInventory.getId());
|
inventorySummary.setInventoryType(PENDING_INVENTORY);
|
}
|
}
|
if (officialInventory != null) {
|
inventorySummary = inventorySummaryMapper.getInventorySummaryForUpdateByInventoryIdAndType(officialInventory.getId(), OFFICIAL_INVENTORY);
|
inputInventoryRecordList = inputInventoryRecordMapper.selectList(new LambdaQueryWrapper<InputInventoryRecord>()
|
.eq(InputInventoryRecord::getInventoryId, officialInventory.getId())
|
.eq(InputInventoryRecord::getInventoryType, OFFICIAL_INVENTORY)
|
.gt(InputInventoryRecord::getId, inventorySummary == null ? 0 : inventorySummary.getInputEndRecordId()));
|
outputInventoryRecordList = outputInventoryRecordMapper.selectList(new LambdaQueryWrapper<OutputInventoryRecord>()
|
.eq(OutputInventoryRecord::getInventoryId, officialInventory.getId())
|
.eq(OutputInventoryRecord::getInventoryType, OFFICIAL_INVENTORY)
|
.gt(OutputInventoryRecord::getId, inventorySummary == null ? 0 : inventorySummary.getOutputEndRecordId()));
|
if (inventorySummary == null) {
|
inventorySummary = new InventorySummary();
|
inventorySummary.setInventoryQuantity(BigDecimal.ZERO);
|
inventorySummary.setInventoryId(officialInventory.getId());
|
inventorySummary.setInventoryType(OFFICIAL_INVENTORY);
|
}
|
}
|
BigDecimal total = inventorySummary.getInventoryQuantity();
|
|
BigDecimal inputTotal = BigDecimal.ZERO;
|
if (CollectionUtils.isNotEmpty(inputInventoryRecordList)) {
|
inputTotal = inputInventoryRecordList.stream().map(InputInventoryRecord::getQuantity).reduce(BigDecimal::add).orElse(BigDecimal.ZERO);
|
inventorySummary.setInputEndRecordId(inputInventoryRecordList.get(inputInventoryRecordList.size() - 1).getId());
|
}
|
BigDecimal outputTotal = BigDecimal.ZERO;
|
if (CollectionUtils.isNotEmpty(outputInventoryRecordList)) {
|
outputTotal = outputInventoryRecordList.stream().map(OutputInventoryRecord::getQuantity).reduce(BigDecimal::add).orElse(BigDecimal.ZERO);
|
inventorySummary.setOutputEndRecordId(outputInventoryRecordList.get(outputInventoryRecordList.size() - 1).getId());
|
}
|
|
if (inputTotal.compareTo(outputTotal) > 0) {
|
total = total.add(inputTotal.subtract(outputTotal));
|
}
|
inventorySummary.setInventoryQuantity(total);
|
|
inventorySummaryMapper.insertOrUpdate(inventorySummary);
|
}
|
}
|
