refactor(StockInfo): 重构 Get3DLayoutAsync 使用依赖注入服务

wanshenmean

2026-03-30 1f15dd43b334e31189c4580b98f57d2cc71935d8

 Code/WMS/WIDESEA_WMSServer/WIDESEA_StockService/StockInfoService.cs

@@ -1,22 +1,47 @@
using WIDESEA_Common.StockEnum;
using WIDESEA_Common.StockEnum;
using WIDESEA_Core.BaseRepository;
using WIDESEA_Core.BaseServices;
using WIDESEA_DTO.Stock;
using WIDESEA_IBasicService;
using WIDESEA_IStockService;
using WIDESEA_Model.Models;

namespace WIDESEA_StockService
{
    /// <summary>
    /// 库存信息服务实现类
    /// </summary>
    public partial class StockInfoService : ServiceBase<Dt_StockInfo, IRepository<Dt_StockInfo>>, IStockInfoService
    {
        /// <summary>
        /// 获取库存信息仓储接口
        /// </summary>
        public IRepository<Dt_StockInfo> Repository => BaseDal;

        public StockInfoService(IRepository<Dt_StockInfo> BaseDal) : base(BaseDal)
        /// <summary>
        /// 货位信息服务接口（用于获取仓库货位信息）
        /// </summary>
        private readonly ILocationInfoService _locationInfoService;

        /// <summary>
        /// 仓库信息服务接口（用于获取仓库基本信息）
        /// </summary>
        private readonly IWarehouseService _warehouseService;

        /// <summary>
        /// 构造函数
        /// </summary>
        /// <param name="baseDal">基础数据访问对象</param>
        public StockInfoService(IRepository<Dt_StockInfo> baseDal, ILocationInfoService locationInfoService, IWarehouseService warehouseService) : base(baseDal)
        {
            _locationInfoService = locationInfoService;
            _warehouseService = warehouseService;
        }

        /// <summary>
        /// 获取库存信息列表（出库日期小于当前时间且库存状态为入库完成的记录）
        /// </summary>
        /// <returns>库存信息列表</returns>
        public async Task<List<Dt_StockInfo>> GetStockInfoAsync()
        {
            return await BaseDal.QueryDataAsync(x =>
@@ -27,6 +52,8 @@
        /// <summary>
        /// 获取库存信息列表（出库日期小于当前时间且库存状态为入库完成的记录，且仓库ID匹配）
        /// </summary>
        /// <param name="warehouseId">仓库ID</param>
        /// <returns>库存信息列表</returns>
        public async Task<List<Dt_StockInfo>> GetStockInfoAsync(int warehouseId)
        {
            return await BaseDal.QueryDataAsync(x =>
@@ -38,29 +65,137 @@
        /// <summary>
        /// 获取库存信息（根据托盘码查询）
        /// </summary>
        /// <param name="palletCode">托盘编码</param>
        /// <returns>库存信息</returns>
        public async Task<Dt_StockInfo> GetStockInfoAsync(string palletCode)
        {
            return await BaseDal.QueryFirstAsync(x => x.PalletCode == palletCode);
            return await BaseDal.QueryDataNavFirstAsync(x => x.PalletCode == palletCode);
        }

        /// <summary>
        /// 更新库存数据
        /// </summary>
        /// <param name="stockInfo">库存信息对象</param>
        /// <returns>更新是否成功</returns>
        public async Task<bool> UpdateStockAsync(Dt_StockInfo stockInfo)
        {
            return await BaseDal.UpdateDataAsync(stockInfo);
        }


        /// <summary>
        /// 检索指定托盘在给定位置的库存详细信息。
        /// 检索指定托盘在给定位置的库存详细信息
        /// </summary>
        /// <param name="palletCode">请求库存信息的托盘唯一标识符。不能为 null 或空。</param>
        /// <param name="locationCode">表示托盘存储位置的代码。不能为 null 或空。</param>
        /// <returns>表示异步操作的任务。任务结果包含一个 <see cref="Dt_StockInfo"/> 对象，该对象包含指定托盘和位置的库存详细信息。如果未找到匹配的库存信息，则返回 null。</returns>
        /// <param name="palletCode">托盘编码</param>
        /// <param name="locationCode">货位编码</param>
        /// <returns>库存信息</returns>
        public async Task<Dt_StockInfo> GetStockInfoAsync(string palletCode, string locationCode)
        {
            return await BaseDal.QueryFirstAsync(x => x.PalletCode == palletCode && x.LocationCode == locationCode);
        }

        /// <summary>
        /// 获取仓库3D布局数据
        /// </summary>
        /// <param name="warehouseId">仓库ID</param>
        /// <returns>3D布局DTO</returns>
        public async Task<Stock3DLayoutDTO> Get3DLayoutAsync(int warehouseId)
        {
            // 1. 查询仓库信息
            var warehouse = await _warehouseService.Repository.QueryFirstAsync(x => x.WarehouseId == warehouseId);

            // 2. 查询该仓库所有货位
            var locations = await _locationInfoService.Repository.QueryDataAsync(x => x.WarehouseId == warehouseId);

            // 3. 查询该仓库所有库存信息（包含Details导航属性）
            var stockInfos = await Repository.QueryDataNavAsync(x => x.WarehouseId == warehouseId);

            // 4. 提取物料编号和批次号列表（去重）
            var materielCodeList = stockInfos
                .Where(s => s.Details != null)
                .SelectMany(s => s.Details)
                .Select(d => d.MaterielCode)
                .Where(c => !string.IsNullOrEmpty(c))
                .Distinct()
                .ToList();

            var batchNoList = stockInfos
                .Where(s => s.Details != null)
                .SelectMany(s => s.Details)
                .Select(d => d.BatchNo)
                .Where(b => !string.IsNullOrEmpty(b))
                .Distinct()
                .ToList();

            // 5. 创建库存字典用于快速查找（以LocationId为键）
            var stockDict = stockInfos.ToDictionary(s => s.LocationId, s => s);

            // 6. 映射每个货位到Location3DItemDTO
            const float defaultMaxCapacity = 100f;
            var locationItems = locations.Select(loc =>
            {
                var item = new Location3DItemDTO
                {
                    LocationId = loc.Id,
                    LocationCode = loc.LocationCode,
                    Row = loc.Row,
                    Column = loc.Column,
                    Layer = loc.Layer,
                    LocationStatus = loc.LocationStatus,
                    MaxCapacity = defaultMaxCapacity
                };

                // 尝试从库存字典中获取库存信息
                if (stockDict.TryGetValue(loc.Id, out var stockInfo) && stockInfo.Details != null)
                {
                    item.PalletCode = stockInfo.PalletCode;
                    item.StockQuantity = stockInfo.Details.Sum(d => d.StockQuantity);

                    // 获取第一个明细的物料信息（如果存在）
                    var firstDetail = stockInfo.Details.FirstOrDefault();
                    if (firstDetail != null)
                    {
                        item.MaterielCode = firstDetail.MaterielCode;
                        item.MaterielName = firstDetail.MaterielName;
                        item.BatchNo = firstDetail.BatchNo;
                    }

                    // 计算库存状态
                    var ratio = item.MaxCapacity > 0 ? item.StockQuantity / item.MaxCapacity : 0;
                    if (ratio >= 0.9f)
                        item.StockStatus = 3; // 已满 (FULL)
                    else if (ratio >= 0.1f)
                        item.StockStatus = 1; // 有货 (HAS_STOCK)
                    else if (ratio > 0)
                        item.StockStatus = 2; // 库存紧张 (LOW_STOCK)
                    else
                        item.StockStatus = 0; // 无货 (EMPTY)
                }
                else
                {
                    item.StockStatus = 0; // 无货 (EMPTY)
                    item.StockQuantity = 0;
                }

                return item;
            }).ToList();

            // 7. 计算仓库尺寸
            var maxRow = locations.Any() ? locations.Max(l => l.Row) : 0;
            var maxColumn = locations.Any() ? locations.Max(l => l.Column) : 0;
            var maxLayer = locations.Any() ? locations.Max(l => l.Layer) : 0;

            // 8. 构建返回结果
            return new Stock3DLayoutDTO
            {
                WarehouseId = warehouseId,
                WarehouseName = warehouse?.WarehouseName ?? string.Empty,
                MaxRow = maxRow,
                MaxColumn = maxColumn,
                MaxLayer = maxLayer,
                MaterielCodeList = materielCodeList,
                BatchNoList = batchNoList,
                Locations = locationItems
            };
        }
    }
}