2026-06-29 940c8481d2fdcf51341db4ccd6fd6fcc2d43debb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
package cn.iocoder.yudao.module.wms.controller.admin.inventory;
 
import cn.iocoder.yudao.framework.common.pojo.CommonResult;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import cn.iocoder.yudao.framework.dict.core.DictFrameworkUtils;
import cn.iocoder.yudao.module.wms.controller.admin.inventory.vo.WmsInventoryReservePageReqVO;
import cn.iocoder.yudao.module.wms.controller.admin.inventory.vo.WmsInventoryReserveRespVO;
import cn.iocoder.yudao.module.wms.dal.dataobject.inventory.WmsInventoryReserveDO;
import cn.iocoder.yudao.module.wms.dal.dataobject.md.item.WmsItemSkuDO;
import cn.iocoder.yudao.module.wms.dal.dataobject.md.warehouse.WmsWarehouseDO;
import cn.iocoder.yudao.module.wms.enums.DictTypeConstants;
import cn.iocoder.yudao.module.wms.service.inventory.WmsInventoryReserveService;
import cn.iocoder.yudao.module.wms.service.md.item.WmsItemSkuService;
import cn.iocoder.yudao.module.wms.service.md.warehouse.WmsWarehouseService;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.Parameter;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.annotation.Resource;
import jakarta.validation.Valid;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;
 
import java.util.List;
import java.util.Map;
 
import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success;
import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.convertSet;
 
@Tag(name = "管理后台 - 库存预留")
@RestController
@RequestMapping("/wms/inventory-reserve")
@Validated
public class WmsInventoryReserveController {
 
    @Resource
    private WmsInventoryReserveService reserveService;
 
    @Resource
    private WmsItemSkuService itemSkuService;
 
    @Resource
    private WmsWarehouseService warehouseService;
 
    @GetMapping("/page")
    @Operation(summary = "获得库存预留分页")
    @PreAuthorize("@ss.hasPermission('wms:inventory-reserve:query')")
    public CommonResult<PageResult<WmsInventoryReserveRespVO>> getInventoryReservePage(@Valid WmsInventoryReservePageReqVO reqVO) {
        PageResult<WmsInventoryReserveDO> pageResult = reserveService.getReservePage(reqVO);
        // 转换
        PageResult<WmsInventoryReserveRespVO> voPageResult = BeanUtils.toBean(pageResult, WmsInventoryReserveRespVO.class);
        // 补充信息
        enrichReserveList(voPageResult.getList());
        return success(voPageResult);
    }
 
    @GetMapping("/get")
    @Operation(summary = "获得库存预留详情")
    @Parameter(name = "id", description = "编号", required = true, example = "1")
    @PreAuthorize("@ss.hasPermission('wms:inventory-reserve:query')")
    public CommonResult<WmsInventoryReserveRespVO> getInventoryReserve(@RequestParam("id") Long id) {
        WmsInventoryReserveDO reserve = reserveService.getReserve(id);
        if (reserve == null) {
            return success(null);
        }
        WmsInventoryReserveRespVO respVO = BeanUtils.toBean(reserve, WmsInventoryReserveRespVO.class);
        // 补充信息
        enrichReserve(respVO);
        return success(respVO);
    }
 
    @PostMapping("/release")
    @Operation(summary = "释放库存预留")
    @Parameter(name = "id", description = "编号", required = true, example = "1")
    @PreAuthorize("@ss.hasPermission('wms:inventory-reserve:release')")
    public CommonResult<Boolean> releaseInventoryReserve(@RequestParam("id") Long id) {
        WmsInventoryReserveDO reserve = reserveService.getReserve(id);
        if (reserve != null) {
            reserveService.releaseReserve(reserve.getReserveNo());
        }
        return success(true);
    }
 
    /**
     * 批量补充预留记录信息
     */
    private void enrichReserveList(List<WmsInventoryReserveRespVO> list) {
        if (list == null || list.isEmpty()) {
            return;
        }
        // 获取 SKU 信息
        Map<Long, WmsItemSkuDO> skuMap = itemSkuService.getItemSkuMap(
                convertSet(list, WmsInventoryReserveRespVO::getSkuId));
        // 获取仓库信息
        Map<Long, WmsWarehouseDO> warehouseMap = warehouseService.getWarehouseMap(
                convertSet(list, WmsInventoryReserveRespVO::getWarehouseId));
 
        for (WmsInventoryReserveRespVO vo : list) {
            // 业务类型名称
            vo.setBizTypeName(DictFrameworkUtils.parseDictDataLabel(DictTypeConstants.WMS_RESERVE_BIZ_TYPE, vo.getBizType()));
            // 状态名称
            vo.setStatusName(DictFrameworkUtils.parseDictDataLabel(DictTypeConstants.WMS_RESERVE_STATUS, vo.getStatus()));
            // SKU 名称
            WmsItemSkuDO sku = skuMap.get(vo.getSkuId());
            if (sku != null) {
                vo.setSkuName(sku.getName());
            }
            // 仓库名称
            WmsWarehouseDO warehouse = warehouseMap.get(vo.getWarehouseId());
            if (warehouse != null) {
                vo.setWarehouseName(warehouse.getName());
            }
        }
    }
 
    /**
     * 补充单个预留记录信息
     */
    private void enrichReserve(WmsInventoryReserveRespVO vo) {
        // 业务类型名称
        vo.setBizTypeName(DictFrameworkUtils.parseDictDataLabel(DictTypeConstants.WMS_RESERVE_BIZ_TYPE, vo.getBizType()));
        // 状态名称
        vo.setStatusName(DictFrameworkUtils.parseDictDataLabel(DictTypeConstants.WMS_RESERVE_STATUS, vo.getStatus()));
        // SKU 名称
        List<WmsItemSkuDO> skuList = itemSkuService.getItemSkuListByIds(List.of(vo.getSkuId()));
        if (!skuList.isEmpty()) {
            vo.setSkuName(skuList.get(0).getName());
        }
        // 仓库名称
        WmsWarehouseDO warehouse = warehouseService.getWarehouse(vo.getWarehouseId());
        if (warehouse != null) {
            vo.setWarehouseName(warehouse.getName());
        }
    }
 
}