zengzy / hhrcs_service · Files

using Microsoft.EntityFrameworkCore;
using Rcs.Domain.Entities;
using Rcs.Domain.Repositories;
using Rcs.Infrastructure.DB.MsSql;

namespace Rcs.Infrastructure.DB.Repositories
{
    /// <summary>
    /// 库区仓储实现
    /// @author zzy
    /// </summary>
    public class StorageAreaRepository : Repository<StorageArea>, IStorageAreaRepository
    {
        public StorageAreaRepository(AppDbContext context) : base(context)
        {
        }

        public async Task<StorageArea?> GetByAreaCodeAsync(string areaCode, CancellationToken cancellationToken = default)
        {
            return await _dbSet
                .FirstOrDefaultAsync(a => a.AreaCode == areaCode, cancellationToken);
        }

        public async Task<StorageArea?> GetWithLocationsAsync(Guid areaId, CancellationToken cancellationToken = default)
        {
            return await _dbSet
                .Include(a => a.StorageLocations)
                .FirstOrDefaultAsync(a => a.AreaId == areaId, cancellationToken);
        }
    }
}