package cn.iocoder.yudao.module.wms.api.inventory;
|
|
import cn.hutool.core.collection.CollUtil;
|
import cn.iocoder.yudao.framework.common.pojo.CommonResult;
|
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
|
import cn.iocoder.yudao.module.wms.api.inventory.dto.*;
|
import cn.iocoder.yudao.module.wms.dal.dataobject.inventory.WmsInventoryDO;
|
import cn.iocoder.yudao.module.wms.dal.dataobject.inventory.WmsInventoryTransactionDO;
|
import cn.iocoder.yudao.module.wms.dal.mysql.inventory.WmsInventoryMapper;
|
import cn.iocoder.yudao.module.wms.enums.inventory.WmsTransactionTypeEnum;
|
import cn.iocoder.yudao.module.wms.service.inventory.dto.WmsInventoryChangeReqDTO;
|
import cn.iocoder.yudao.module.wms.service.inventory.WmsInventoryReserveService;
|
import cn.iocoder.yudao.module.wms.service.inventory.WmsInventoryService;
|
import cn.iocoder.yudao.module.wms.service.inventory.WmsInventoryTransactionService;
|
import jakarta.annotation.Resource;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
import org.springframework.validation.annotation.Validated;
|
import org.springframework.web.bind.annotation.RequestMapping;
|
import org.springframework.web.bind.annotation.RestController;
|
|
import java.math.BigDecimal;
|
import java.time.LocalDateTime;
|
import java.time.format.DateTimeFormatter;
|
import java.util.ArrayList;
|
import java.util.List;
|
import java.util.UUID;
|
|
import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success;
|
|
/**
|
* WMS 库存操作 API 实现类
|
*
|
* @author 芋道源码
|
*/
|
@RestController
|
@RequestMapping(WmsInventoryApi.PREFIX)
|
@Service
|
@Validated
|
@Slf4j
|
public class WmsInventoryApiImpl implements WmsInventoryApi {
|
|
@Resource
|
private WmsInventoryService inventoryService;
|
|
@Resource
|
private WmsInventoryReserveService reserveService;
|
|
@Resource
|
private WmsInventoryTransactionService transactionService;
|
|
@Resource
|
private WmsInventoryMapper inventoryMapper;
|
|
// ==================== 入库操作 ====================
|
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public CommonResult<WmsReceiptResultDTO> receipt(WmsReceiptReqDTO reqDTO) {
|
// 1. 构建库存变更请求
|
WmsInventoryChangeReqDTO changeReq = buildInventoryChangeReq(reqDTO);
|
|
// 2. 执行入库变更
|
inventoryService.changeInventory(changeReq);
|
|
// 3. 记录事务流水
|
List<WmsInventoryTransactionDO> transactions = buildReceiptTransactions(reqDTO);
|
transactionService.createTransactionList(transactions);
|
|
// 4. 构建返回结果
|
WmsReceiptResultDTO result = new WmsReceiptResultDTO();
|
result.setReceiptOrderNo(generateReceiptNo());
|
result.setTotalQuantity(calculateTotalQuantity(reqDTO.getItems()));
|
return success(result);
|
}
|
|
// ==================== 出库操作 ====================
|
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public CommonResult<WmsShipmentResultDTO> shipment(WmsShipmentReqDTO reqDTO) {
|
// 1. 如果有预留单号，先使用预留
|
if (reqDTO.getReserveNo() != null) {
|
WmsUseReserveReqDTO useReq = new WmsUseReserveReqDTO();
|
useReq.setReserveNo(reqDTO.getReserveNo());
|
useReq.setQuantity(calculateTotalQuantity(reqDTO.getItems()));
|
reserveService.useReserve(useReq);
|
}
|
|
// 2. 构建库存变更请求（出库为负数）
|
WmsInventoryChangeReqDTO changeReq = buildShipmentChangeReq(reqDTO);
|
|
// 3. 执行出库变更
|
inventoryService.changeInventory(changeReq);
|
|
// 4. 记录事务流水
|
List<WmsInventoryTransactionDO> transactions = buildShipmentTransactions(reqDTO);
|
transactionService.createTransactionList(transactions);
|
|
// 5. 构建返回结果
|
WmsShipmentResultDTO result = new WmsShipmentResultDTO();
|
result.setShipmentOrderNo(generateShipmentNo());
|
result.setTotalQuantity(calculateTotalQuantity(reqDTO.getItems()));
|
return success(result);
|
}
|
|
// ==================== 库存预留 ====================
|
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public CommonResult<String> reserve(WmsReserveReqDTO reqDTO) {
|
return success(reserveService.createReserve(reqDTO));
|
}
|
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public CommonResult<Boolean> releaseReserve(String reserveNo) {
|
reserveService.releaseReserve(reserveNo);
|
return success(true);
|
}
|
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public CommonResult<Boolean> useReserve(WmsUseReserveReqDTO reqDTO) {
|
reserveService.useReserve(reqDTO);
|
return success(true);
|
}
|
|
// ==================== 库存查询 ====================
|
|
@Override
|
public CommonResult<List<WmsInventoryRespDTO>> getInventoryList(WmsInventoryQueryReqDTO reqDTO) {
|
// 1. 查询库存
|
List<WmsInventoryDO> inventories = inventoryMapper.selectListByQuery(reqDTO);
|
|
// 2. 转换结果
|
List<WmsInventoryRespDTO> results = new ArrayList<>();
|
for (WmsInventoryDO inventory : inventories) {
|
WmsInventoryRespDTO resp = BeanUtils.toBean(inventory, WmsInventoryRespDTO.class);
|
// 计算预留数量
|
BigDecimal reserveQuantity = reserveService.getReserveQuantity(
|
inventory.getSkuId(), inventory.getWarehouseId());
|
resp.setReserveQuantity(reserveQuantity);
|
resp.setAvailableQuantity(inventory.getQuantity().subtract(reserveQuantity));
|
results.add(resp);
|
}
|
return success(results);
|
}
|
|
@Override
|
public CommonResult<BigDecimal> getAvailableQuantity(Long skuId, Long warehouseId) {
|
return success(inventoryService.getAvailableQuantity(skuId, warehouseId));
|
}
|
|
// ==================== 私有方法 ====================
|
|
private WmsInventoryChangeReqDTO buildInventoryChangeReq(WmsReceiptReqDTO reqDTO) {
|
WmsInventoryChangeReqDTO changeReq = new WmsInventoryChangeReqDTO();
|
changeReq.setOrderType(reqDTO.getBizType());
|
changeReq.setOrderNo(reqDTO.getBizNo());
|
|
List<WmsInventoryChangeReqDTO.Item> items = new ArrayList<>();
|
for (WmsReceiptReqDTO.Item item : reqDTO.getItems()) {
|
WmsInventoryChangeReqDTO.Item changeItem = new WmsInventoryChangeReqDTO.Item();
|
changeItem.setSkuId(item.getSkuId());
|
changeItem.setWarehouseId(reqDTO.getWarehouseId());
|
changeItem.setQuantity(item.getQuantity()); // 入库为正数
|
changeItem.setPrice(item.getPrice());
|
changeItem.setRemark(item.getRemark());
|
items.add(changeItem);
|
}
|
changeReq.setItems(items);
|
return changeReq;
|
}
|
|
private WmsInventoryChangeReqDTO buildShipmentChangeReq(WmsShipmentReqDTO reqDTO) {
|
WmsInventoryChangeReqDTO changeReq = new WmsInventoryChangeReqDTO();
|
changeReq.setOrderType(reqDTO.getBizType());
|
changeReq.setOrderNo(reqDTO.getBizNo());
|
|
List<WmsInventoryChangeReqDTO.Item> items = new ArrayList<>();
|
for (WmsShipmentReqDTO.Item item : reqDTO.getItems()) {
|
WmsInventoryChangeReqDTO.Item changeItem = new WmsInventoryChangeReqDTO.Item();
|
changeItem.setSkuId(item.getSkuId());
|
changeItem.setWarehouseId(reqDTO.getWarehouseId());
|
changeItem.setQuantity(item.getQuantity().negate()); // 出库为负数
|
changeItem.setRemark(item.getRemark());
|
items.add(changeItem);
|
}
|
changeReq.setItems(items);
|
return changeReq;
|
}
|
|
private List<WmsInventoryTransactionDO> buildReceiptTransactions(WmsReceiptReqDTO reqDTO) {
|
List<WmsInventoryTransactionDO> transactions = new ArrayList<>();
|
for (WmsReceiptReqDTO.Item item : reqDTO.getItems()) {
|
WmsInventoryTransactionDO transaction = WmsInventoryTransactionDO.builder()
|
.transactionNo(generateTransactionNo())
|
.transactionType(WmsTransactionTypeEnum.RECEIPT.getType())
|
.bizType(reqDTO.getBizType())
|
.bizId(reqDTO.getBizId())
|
.bizNo(reqDTO.getBizNo())
|
.skuId(item.getSkuId())
|
.warehouseId(reqDTO.getWarehouseId())
|
.quantity(item.getQuantity())
|
.transactionTime(reqDTO.getReceiptTime() != null ? reqDTO.getReceiptTime() : LocalDateTime.now())
|
.remark(item.getRemark())
|
.build();
|
transactions.add(transaction);
|
}
|
return transactions;
|
}
|
|
private List<WmsInventoryTransactionDO> buildShipmentTransactions(WmsShipmentReqDTO reqDTO) {
|
List<WmsInventoryTransactionDO> transactions = new ArrayList<>();
|
for (WmsShipmentReqDTO.Item item : reqDTO.getItems()) {
|
WmsInventoryTransactionDO transaction = WmsInventoryTransactionDO.builder()
|
.transactionNo(generateTransactionNo())
|
.transactionType(WmsTransactionTypeEnum.SHIPMENT.getType())
|
.bizType(reqDTO.getBizType())
|
.bizId(reqDTO.getBizId())
|
.bizNo(reqDTO.getBizNo())
|
.skuId(item.getSkuId())
|
.warehouseId(reqDTO.getWarehouseId())
|
.quantity(item.getQuantity().negate())
|
.reserveNo(reqDTO.getReserveNo())
|
.transactionTime(reqDTO.getShipmentTime() != null ? reqDTO.getShipmentTime() : LocalDateTime.now())
|
.remark(item.getRemark())
|
.build();
|
transactions.add(transaction);
|
}
|
return transactions;
|
}
|
|
private BigDecimal calculateTotalQuantity(List<?> items) {
|
if (CollUtil.isEmpty(items)) {
|
return BigDecimal.ZERO;
|
}
|
BigDecimal total = BigDecimal.ZERO;
|
for (Object item : items) {
|
if (item instanceof WmsReceiptReqDTO.Item) {
|
total = total.add(((WmsReceiptReqDTO.Item) item).getQuantity());
|
} else if (item instanceof WmsShipmentReqDTO.Item) {
|
total = total.add(((WmsShipmentReqDTO.Item) item).getQuantity());
|
}
|
}
|
return total;
|
}
|
|
private String generateReceiptNo() {
|
return "RCPT" + LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyyMMddHHmmss"))
|
+ UUID.randomUUID().toString().substring(0, 6).toUpperCase();
|
}
|
|
private String generateShipmentNo() {
|
return "SHIP" + LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyyMMddHHmmss"))
|
+ UUID.randomUUID().toString().substring(0, 6).toUpperCase();
|
}
|
|
private String generateTransactionNo() {
|
return "TRX" + LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyyMMddHHmmss"))
|
+ UUID.randomUUID().toString().substring(0, 6).toUpperCase();
|
}
|
|
}
|
