| | | 1 | | using Microsoft.EntityFrameworkCore; |
| | | 2 | | using OzonDomains.Models; |
| | | 3 | | using OzonRepositories.Context; |
| | | 4 | | |
| | | 5 | | namespace OzonRepositories.Data |
| | | 6 | | { |
| | | 7 | | public class ShipmentRepository : MainRepository |
| | | 8 | | { |
| | 13 | 9 | | public ShipmentRepository(ApplicationDbContext orderContext) : base(orderContext) |
| | | 10 | | { |
| | 13 | 11 | | } |
| | | 12 | | |
| | | 13 | | /// <summary> |
| | | 14 | | /// Получает существующее отправление по номеру или создает новое. |
| | | 15 | | /// Возвращает Id отправления. |
| | | 16 | | /// </summary> |
| | | 17 | | /// <param name="shipmentNumber">Номер отправления</param> |
| | | 18 | | /// <returns>Id отправления</returns> |
| | | 19 | | public async Task<int> GetOrCreateAsync(string shipmentNumber) |
| | | 20 | | { |
| | 17 | 21 | | if (string.IsNullOrWhiteSpace(shipmentNumber)) |
| | 3 | 22 | | throw new ArgumentNullException(nameof(shipmentNumber)); |
| | | 23 | | |
| | 14 | 24 | | shipmentNumber = shipmentNumber.Trim(); |
| | | 25 | | |
| | 14 | 26 | | var local = _context.Shipments.Local |
| | 16 | 27 | | .FirstOrDefault(s => s.ShipmentNumber == shipmentNumber); |
| | 14 | 28 | | if (local != null && local.Id > 0) |
| | 2 | 29 | | return local.Id; |
| | | 30 | | |
| | 12 | 31 | | var existingId = await _context.Shipments |
| | 12 | 32 | | .AsNoTracking() |
| | 12 | 33 | | .Where(s => s.ShipmentNumber == shipmentNumber) |
| | 12 | 34 | | .Select(s => (int?)s.Id) |
| | 12 | 35 | | .FirstOrDefaultAsync(); |
| | | 36 | | |
| | 12 | 37 | | if (existingId.HasValue && existingId.Value > 0) |
| | 7 | 38 | | return existingId.Value; |
| | | 39 | | |
| | 5 | 40 | | if (_context.Database.IsRelational()) |
| | | 41 | | { |
| | 3 | 42 | | var provider = _context.Database.ProviderName ?? string.Empty; |
| | 3 | 43 | | if (provider.Contains("Sqlite", StringComparison.OrdinalIgnoreCase)) |
| | | 44 | | { |
| | 3 | 45 | | await _context.Database.ExecuteSqlRawAsync( |
| | 3 | 46 | | "INSERT OR IGNORE INTO Shipments (ShipmentNumber) VALUES ({0})", shipmentNumber); |
| | | 47 | | |
| | 3 | 48 | | var sqliteId = await _context.Database.SqlQueryRaw<int>( |
| | 3 | 49 | | "SELECT Id AS Value FROM Shipments WHERE ShipmentNumber = {0} LIMIT 1", shipmentNumber) |
| | 3 | 50 | | .SingleOrDefaultAsync(); |
| | | 51 | | |
| | 3 | 52 | | if (sqliteId > 0) |
| | 3 | 53 | | return sqliteId; |
| | | 54 | | |
| | 0 | 55 | | sqliteId = await _context.Shipments |
| | 0 | 56 | | .AsNoTracking() |
| | 0 | 57 | | .Where(s => s.ShipmentNumber == shipmentNumber) |
| | 0 | 58 | | .Select(s => s.Id) |
| | 0 | 59 | | .FirstOrDefaultAsync(); |
| | | 60 | | |
| | 0 | 61 | | if (sqliteId > 0) |
| | 0 | 62 | | return sqliteId; |
| | | 63 | | } |
| | | 64 | | else |
| | | 65 | | { |
| | | 66 | | // MySQL / MariaDB: атомарная вставка с игнорированием дубликатов по UX_Shipments_ShipmentNumber. |
| | | 67 | | // Исключает выброс DbUpdateException и ошибки RelationalEventId.SaveChangesFailed (EventId 10000) |
| | | 68 | | // при параллельном создании отправления в нескольких потоках / джобах (#1824, #1825). |
| | | 69 | | // LOCK IN SHARE MODE гарантирует чтение актуальной строки даже в транзакциях REPEATABLE READ. |
| | 0 | 70 | | await _context.Database.ExecuteSqlRawAsync( |
| | 0 | 71 | | "INSERT IGNORE INTO `Shipments` (`ShipmentNumber`) VALUES ({0})", shipmentNumber); |
| | | 72 | | |
| | 0 | 73 | | var mysqlId = await _context.Database.SqlQueryRaw<int>( |
| | 0 | 74 | | "SELECT `Id` AS `Value` FROM `Shipments` WHERE `ShipmentNumber` = {0} LOCK IN SHARE MODE", shipm |
| | 0 | 75 | | .SingleOrDefaultAsync(); |
| | | 76 | | |
| | 0 | 77 | | if (mysqlId > 0) |
| | 0 | 78 | | return mysqlId; |
| | | 79 | | |
| | | 80 | | // Fallback при задержке видимости или пробелах/collation |
| | 0 | 81 | | mysqlId = await _context.Database.SqlQueryRaw<int>( |
| | 0 | 82 | | "SELECT `Id` AS `Value` FROM `Shipments` WHERE TRIM(`ShipmentNumber`) = {0} LIMIT 1", shipmentNu |
| | 0 | 83 | | .SingleOrDefaultAsync(); |
| | | 84 | | |
| | 0 | 85 | | if (mysqlId > 0) |
| | 0 | 86 | | return mysqlId; |
| | | 87 | | |
| | 0 | 88 | | mysqlId = await _context.Shipments |
| | 0 | 89 | | .AsNoTracking() |
| | 0 | 90 | | .Where(s => s.ShipmentNumber == shipmentNumber) |
| | 0 | 91 | | .Select(s => s.Id) |
| | 0 | 92 | | .FirstOrDefaultAsync(); |
| | | 93 | | |
| | 0 | 94 | | if (mysqlId > 0) |
| | 0 | 95 | | return mysqlId; |
| | | 96 | | } |
| | | 97 | | |
| | 0 | 98 | | throw new InvalidOperationException($"Не удалось получить или создать отправление '{shipmentNumber}'."); |
| | | 99 | | } |
| | | 100 | | |
| | | 101 | | // Fallback исключительно для нереляционных провайдеров (UseInMemoryDatabase в тестах) |
| | | 102 | | try |
| | | 103 | | { |
| | 2 | 104 | | var newShipment = new Shipment { ShipmentNumber = shipmentNumber }; |
| | 2 | 105 | | await _context.Shipments.AddAsync(newShipment); |
| | 2 | 106 | | await _context.SaveChangesAsync(); |
| | 2 | 107 | | return newShipment.Id; |
| | | 108 | | } |
| | 0 | 109 | | catch (DbUpdateException) |
| | | 110 | | { |
| | 0 | 111 | | var failedEntry = _context.ChangeTracker.Entries<Shipment>() |
| | 0 | 112 | | .FirstOrDefault(e => e.Entity.ShipmentNumber == shipmentNumber && e.State == EntityState.Added); |
| | 0 | 113 | | if (failedEntry != null) |
| | 0 | 114 | | failedEntry.State = EntityState.Detached; |
| | | 115 | | |
| | 0 | 116 | | var existing = await _context.Shipments |
| | 0 | 117 | | .FirstOrDefaultAsync(s => s.ShipmentNumber == shipmentNumber); |
| | | 118 | | |
| | 0 | 119 | | if (existing != null) |
| | 0 | 120 | | return existing.Id; |
| | | 121 | | |
| | 0 | 122 | | throw; |
| | | 123 | | } |
| | 14 | 124 | | } |
| | | 125 | | |
| | | 126 | | /// <summary> |
| | | 127 | | /// Получает отправление по Id. |
| | | 128 | | /// </summary> |
| | | 129 | | public async Task<Shipment?> GetAsync(int id) |
| | | 130 | | { |
| | 1 | 131 | | return await _context.Shipments.FindAsync(id); |
| | 1 | 132 | | } |
| | | 133 | | |
| | | 134 | | /// <summary> |
| | | 135 | | /// Получает отправление по номеру. |
| | | 136 | | /// </summary> |
| | | 137 | | public async Task<Shipment?> GetByNumberAsync(string shipmentNumber) |
| | | 138 | | { |
| | 2 | 139 | | if (string.IsNullOrWhiteSpace(shipmentNumber)) |
| | 0 | 140 | | return null; |
| | | 141 | | |
| | 2 | 142 | | shipmentNumber = shipmentNumber.Trim(); |
| | 2 | 143 | | return await _context.Shipments |
| | 2 | 144 | | .FirstOrDefaultAsync(s => s.ShipmentNumber == shipmentNumber); |
| | 2 | 145 | | } |
| | | 146 | | |
| | | 147 | | /// <summary> |
| | | 148 | | /// Получает все отправления. |
| | | 149 | | /// </summary> |
| | | 150 | | public async Task<List<Shipment>> GetAllAsync() |
| | | 151 | | { |
| | 0 | 152 | | return await _context.Shipments.ToListAsync(); |
| | 0 | 153 | | } |
| | | 154 | | |
| | | 155 | | /// <summary> |
| | | 156 | | /// Добавляет новое отправление. |
| | | 157 | | /// </summary> |
| | | 158 | | public async Task<int> AddAsync(Shipment shipment) |
| | | 159 | | { |
| | 0 | 160 | | await _context.Shipments.AddAsync(shipment); |
| | 0 | 161 | | return await _context.SaveChangesAsync(); |
| | 0 | 162 | | } |
| | | 163 | | |
| | | 164 | | /// <summary> |
| | | 165 | | /// Удаляет отправление. |
| | | 166 | | /// </summary> |
| | | 167 | | public async Task<int> DeleteAsync(Shipment shipment) |
| | | 168 | | { |
| | 0 | 169 | | _context.Shipments.Remove(shipment); |
| | 0 | 170 | | return await _context.SaveChangesAsync(); |
| | 0 | 171 | | } |
| | | 172 | | } |
| | | 173 | | } |