2026-07-01 6b5f7c66fc40d7f6099d561e31a34fbd50dd20d3
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
package cn.iocoder.yudao.module.wms.service.home;
 
import cn.hutool.core.date.DatePattern;
import cn.hutool.core.map.MapUtil;
import cn.hutool.core.util.ObjectUtil;
import cn.iocoder.yudao.module.wms.controller.admin.home.vo.WmsHomeInventorySummaryRespVO;
import cn.iocoder.yudao.module.wms.controller.admin.home.vo.WmsHomeOrderStatusRespVO;
import cn.iocoder.yudao.module.wms.controller.admin.home.vo.WmsHomeOrderSummaryRespVO;
import cn.iocoder.yudao.module.wms.controller.admin.home.vo.WmsHomeOrderTrendRespVO;
import cn.iocoder.yudao.module.wms.dal.mysql.home.WmsHomeStatisticsMapper;
import cn.iocoder.yudao.module.wms.enums.order.WmsOrderStatusEnum;
import cn.iocoder.yudao.module.wms.enums.order.WmsOrderTypeEnum;
import cn.iocoder.yudao.module.wms.service.md.warehouse.WmsWarehouseService;
import jakarta.annotation.Resource;
import org.springframework.stereotype.Service;
 
import java.math.BigDecimal;
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.util.Collections;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
 
import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.convertList;
import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.getSumValue;
import static cn.iocoder.yudao.framework.common.util.collection.MapUtils.getBigDecimal;
import static cn.iocoder.yudao.framework.common.util.date.LocalDateTimeUtils.getDateList;
 
/**
 * WMS 首页统计 Service 实现类
 *
 * @author 芋道源码
 */
@Service
public class WmsHomeStatisticsServiceImpl implements WmsHomeStatisticsService {
 
    @Resource
    private WmsHomeStatisticsMapper homeStatisticsMapper;
    @Resource
    private WmsWarehouseService warehouseService;
 
    @Override
    public List<WmsHomeOrderSummaryRespVO> getOrderSummary(Long warehouseId) {
        validateWarehouseIfPresent(warehouseId);
        // 查询单据数量
        List<Map<String, Object>> stats = homeStatisticsMapper.selectOrderCountGroupByTypeAndStatus(warehouseId);
        // 按照单据类型 + 单据状态分组
        return convertList(WmsOrderTypeEnum.values(), orderTypeEnum -> {
            List<WmsHomeOrderStatusRespVO> statuses = convertList(WmsOrderStatusEnum.values(), statusEnum ->
                    new WmsHomeOrderStatusRespVO().setStatus(statusEnum.getStatus())
                            .setCount(getOrderCount(stats, orderTypeEnum.getType(), statusEnum.getStatus())));
            Long total = getSumValue(statuses, WmsHomeOrderStatusRespVO::getCount, Long::sum, 0L);
            return new WmsHomeOrderSummaryRespVO().setType(orderTypeEnum.getType()).setTotal(total).setStatuses(statuses);
        });
    }
 
    @Override
    public List<WmsHomeOrderTrendRespVO> getOrderTrend(Integer days, Long warehouseId) {
        validateWarehouseIfPresent(warehouseId);
        // 统计区间为 [今天 - days + 1, 明天),保证最近 days 天包含今天。
        LocalDate endDate = LocalDate.now().plusDays(1);
        LocalDate startDate = endDate.minusDays(days);
        LocalDateTime beginTime = startDate.atStartOfDay();
        LocalDateTime endTime = endDate.atStartOfDay();
        // 查询每天的单据数量
        List<Map<String, Object>> dbData = homeStatisticsMapper.selectDailyOrderTrend(beginTime, endTime, warehouseId);
        Map<String, Map<Integer, Long>> dateMap = new LinkedHashMap<>();
        for (Map<String, Object> row : dbData) {
            String date = MapUtil.getStr(row, "date");
            Integer orderType = MapUtil.getInt(row, "orderType");
            Long count = MapUtil.getLong(row, "count", 0L);
            dateMap.computeIfAbsent(date, key -> new HashMap<>()).put(orderType, count);
        }
        // 构造结果,保证每天都有数据
        return convertList(getDateList(startDate, days), d -> {
            String dateStr = DatePattern.NORM_DATE_FORMATTER.format(d);
            Map<Integer, Long> row = dateMap.getOrDefault(dateStr, Collections.emptyMap());
            return new WmsHomeOrderTrendRespVO().setTime(d.atStartOfDay())
                    .setReceiptCount(row.getOrDefault(WmsOrderTypeEnum.RECEIPT.getType(), 0L))
                    .setShipmentCount(row.getOrDefault(WmsOrderTypeEnum.SHIPMENT.getType(), 0L))
                    .setMovementCount(row.getOrDefault(WmsOrderTypeEnum.MOVEMENT.getType(), 0L))
                    .setCheckCount(row.getOrDefault(WmsOrderTypeEnum.CHECK.getType(), 0L));
        });
    }
 
    @Override
    public WmsHomeInventorySummaryRespVO getInventorySummary(Long warehouseId, Integer goodsLimit, Integer warehouseLimit) {
        validateWarehouseIfPresent(warehouseId);
        BigDecimal totalQuantity = ObjectUtil.defaultIfNull(homeStatisticsMapper.selectInventoryTotalQuantity(warehouseId), BigDecimal.ZERO);
        List<Map<String, Object>> itemRows = homeStatisticsMapper.selectInventoryItemRank(warehouseId, goodsLimit);
        List<Map<String, Object>> warehouseRows = homeStatisticsMapper.selectInventoryWarehouseRank(warehouseId, warehouseLimit);
        return new WmsHomeInventorySummaryRespVO().setTotalQuantity(totalQuantity)
                .setGoodsShareList(buildInventoryItemRankList(itemRows))
                .setWarehouseDistributionList(buildInventoryWarehouseRankList(warehouseRows));
    }
 
    // ==================== 工具方法 ====================
 
    /**
     * 仓库编号非空时,校验仓库是否存在,避免前端误传任意 id 直接落到 SQL。
     */
    private void validateWarehouseIfPresent(Long warehouseId) {
        if (warehouseId != null) {
            warehouseService.validateWarehouseExists(warehouseId);
        }
    }
 
    private long getOrderCount(List<Map<String, Object>> stats, Integer orderType, Integer status) {
        for (Map<String, Object> stat : stats) {
            if (ObjectUtil.equal(MapUtil.getInt(stat, "orderType"), orderType)
                    && ObjectUtil.equal(MapUtil.getInt(stat, "status"), status)) {
                return MapUtil.getLong(stat, "count", 0L);
            }
        }
        return 0L;
    }
 
    private List<WmsHomeInventorySummaryRespVO.ItemRank> buildInventoryItemRankList(List<Map<String, Object>> rows) {
        return convertList(rows, row -> new WmsHomeInventorySummaryRespVO.ItemRank()
                .setId(ObjectUtil.defaultIfNull(MapUtil.getLong(row, "id"), MapUtil.getLong(row, "itemId")))
                .setName(ObjectUtil.defaultIfNull(MapUtil.getStr(row, "name"), MapUtil.getStr(row, "itemName")))
                .setQuantity(getBigDecimal(row, "quantity", BigDecimal.ZERO)));
    }
 
    private List<WmsHomeInventorySummaryRespVO.WarehouseRank> buildInventoryWarehouseRankList(List<Map<String, Object>> rows) {
        return convertList(rows, row -> new WmsHomeInventorySummaryRespVO.WarehouseRank()
                .setId(ObjectUtil.defaultIfNull(MapUtil.getLong(row, "id"), MapUtil.getLong(row, "warehouseId")))
                .setName(ObjectUtil.defaultIfNull(MapUtil.getStr(row, "name"), MapUtil.getStr(row, "warehouseName")))
                .setQuantity(getBigDecimal(row, "quantity", BigDecimal.ZERO)));
    }
 
}