| | | 1 | | using Hangfire; |
| | | 2 | | using Microsoft.EntityFrameworkCore; |
| | | 3 | | using OzonDomains.Models; |
| | | 4 | | using OzonRepositories.Context; |
| | | 5 | | |
| | | 6 | | namespace Services.BackgroundJobsServices; |
| | | 7 | | |
| | | 8 | | public interface IBackgroundJobScheduleService |
| | | 9 | | { |
| | | 10 | | Task<List<BackgroundJobSchedule>> GetAllSchedulesAsync(CancellationToken cancellationToken = default); |
| | | 11 | | Task<List<BackgroundJobSchedule>> GetActiveSchedulesAsync(CancellationToken cancellationToken = default); |
| | | 12 | | Task<BackgroundJobSchedule?> GetScheduleByIdAsync(int id, CancellationToken cancellationToken = default); |
| | | 13 | | Task<int> CreateScheduleAsync(BackgroundJobSchedule schedule, CancellationToken cancellationToken = default); |
| | | 14 | | Task UpdateScheduleAsync(BackgroundJobSchedule schedule, CancellationToken cancellationToken = default); |
| | | 15 | | Task DeleteScheduleAsync(int id, CancellationToken cancellationToken = default); |
| | | 16 | | Task ActivateScheduleAsync(int id, CancellationToken cancellationToken = default); |
| | | 17 | | Task DeactivateScheduleAsync(int id, CancellationToken cancellationToken = default); |
| | | 18 | | Task RegisterAllActiveSchedulesAsync(CancellationToken cancellationToken = default); |
| | | 19 | | Task RegisterScheduleInHangfireAsync(BackgroundJobSchedule schedule, CancellationToken cancellationToken = default); |
| | | 20 | | Task UnregisterScheduleFromHangfireAsync(int scheduleId, CancellationToken cancellationToken = default); |
| | | 21 | | } |
| | | 22 | | |
| | | 23 | | public class BackgroundJobScheduleService : IBackgroundJobScheduleService |
| | | 24 | | { |
| | | 25 | | private readonly ApplicationDbContext _context; |
| | | 26 | | private readonly IRecurringJobManager _recurringJobManager; |
| | | 27 | | |
| | 14 | 28 | | public BackgroundJobScheduleService( |
| | 14 | 29 | | ApplicationDbContext context, |
| | 14 | 30 | | IRecurringJobManager recurringJobManager) |
| | | 31 | | { |
| | 14 | 32 | | _context = context; |
| | 14 | 33 | | _recurringJobManager = recurringJobManager; |
| | 14 | 34 | | } |
| | | 35 | | |
| | | 36 | | public async Task<List<BackgroundJobSchedule>> GetAllSchedulesAsync(CancellationToken cancellationToken = default) |
| | | 37 | | { |
| | 0 | 38 | | return await _context.BackgroundJobSchedules |
| | 0 | 39 | | .Include(s => s.OzonClient) |
| | 0 | 40 | | .OrderByDescending(s => s.CreatedAt) |
| | 0 | 41 | | .ToListAsync(cancellationToken); |
| | 0 | 42 | | } |
| | | 43 | | |
| | | 44 | | public async Task<List<BackgroundJobSchedule>> GetActiveSchedulesAsync(CancellationToken cancellationToken = default |
| | | 45 | | { |
| | 1 | 46 | | return await _context.BackgroundJobSchedules |
| | 1 | 47 | | .Include(s => s.OzonClient) |
| | 1 | 48 | | .Where(s => s.IsActive) |
| | 1 | 49 | | .OrderByDescending(s => s.CreatedAt) |
| | 1 | 50 | | .ToListAsync(cancellationToken); |
| | 1 | 51 | | } |
| | | 52 | | |
| | | 53 | | public async Task<BackgroundJobSchedule?> GetScheduleByIdAsync(int id, CancellationToken cancellationToken = default |
| | | 54 | | { |
| | 0 | 55 | | return await _context.BackgroundJobSchedules |
| | 0 | 56 | | .Include(s => s.OzonClient) |
| | 0 | 57 | | .FirstOrDefaultAsync(s => s.Id == id, cancellationToken); |
| | 0 | 58 | | } |
| | | 59 | | |
| | | 60 | | public async Task<int> CreateScheduleAsync(BackgroundJobSchedule schedule, CancellationToken cancellationToken = def |
| | | 61 | | { |
| | 5 | 62 | | NormalizeTransactionJobDaysBack(schedule); |
| | | 63 | | |
| | | 64 | | // Auto-generate name if empty |
| | 5 | 65 | | if (string.IsNullOrWhiteSpace(schedule.Name)) |
| | | 66 | | { |
| | 1 | 67 | | schedule.Name = GenerateScheduleName(schedule); |
| | | 68 | | } |
| | | 69 | | |
| | 5 | 70 | | schedule.CreatedAt = DateTime.UtcNow; |
| | 5 | 71 | | schedule.UpdatedAt = DateTime.UtcNow; |
| | | 72 | | |
| | 5 | 73 | | _context.BackgroundJobSchedules.Add(schedule); |
| | 5 | 74 | | await _context.SaveChangesAsync(cancellationToken); |
| | | 75 | | |
| | 5 | 76 | | if (schedule.IsActive) |
| | | 77 | | { |
| | 1 | 78 | | await RegisterScheduleInHangfireAsync(schedule, cancellationToken); |
| | | 79 | | } |
| | | 80 | | |
| | 5 | 81 | | return schedule.Id; |
| | 5 | 82 | | } |
| | | 83 | | |
| | | 84 | | public async Task UpdateScheduleAsync(BackgroundJobSchedule schedule, CancellationToken cancellationToken = default) |
| | | 85 | | { |
| | 0 | 86 | | var existing = await _context.BackgroundJobSchedules.FindAsync(new object[] { schedule.Id }, cancellationToken); |
| | 0 | 87 | | if (existing == null) |
| | 0 | 88 | | throw new InvalidOperationException($"Schedule with ID {schedule.Id} not found"); |
| | | 89 | | |
| | 0 | 90 | | existing.Name = schedule.Name; |
| | 0 | 91 | | existing.JobType = schedule.JobType; |
| | 0 | 92 | | existing.CronExpression = schedule.CronExpression; |
| | 0 | 93 | | existing.IsActive = schedule.IsActive; |
| | 0 | 94 | | existing.OzonClientId = schedule.OzonClientId; |
| | 0 | 95 | | NormalizeTransactionJobDaysBack(schedule); |
| | 0 | 96 | | existing.DaysBack = schedule.DaysBack; |
| | 0 | 97 | | existing.UpdatedAt = DateTime.UtcNow; |
| | | 98 | | |
| | 0 | 99 | | await _context.SaveChangesAsync(cancellationToken); |
| | | 100 | | |
| | 0 | 101 | | if (existing.IsActive) |
| | | 102 | | { |
| | 0 | 103 | | await RegisterScheduleInHangfireAsync(existing, cancellationToken); |
| | | 104 | | } |
| | | 105 | | else |
| | | 106 | | { |
| | 0 | 107 | | await UnregisterScheduleFromHangfireAsync(existing.Id, cancellationToken); |
| | | 108 | | } |
| | 0 | 109 | | } |
| | | 110 | | |
| | | 111 | | public async Task DeleteScheduleAsync(int id, CancellationToken cancellationToken = default) |
| | | 112 | | { |
| | 0 | 113 | | var schedule = await _context.BackgroundJobSchedules.FindAsync(new object[] { id }, cancellationToken); |
| | 0 | 114 | | if (schedule == null) |
| | 0 | 115 | | throw new InvalidOperationException($"Schedule with ID {id} not found"); |
| | | 116 | | |
| | 0 | 117 | | await UnregisterScheduleFromHangfireAsync(id, cancellationToken); |
| | | 118 | | |
| | 0 | 119 | | _context.BackgroundJobSchedules.Remove(schedule); |
| | 0 | 120 | | await _context.SaveChangesAsync(cancellationToken); |
| | 0 | 121 | | } |
| | | 122 | | |
| | | 123 | | public async Task ActivateScheduleAsync(int id, CancellationToken cancellationToken = default) |
| | | 124 | | { |
| | 0 | 125 | | var schedule = await _context.BackgroundJobSchedules.FindAsync(new object[] { id }, cancellationToken); |
| | 0 | 126 | | if (schedule == null) |
| | 0 | 127 | | throw new InvalidOperationException($"Schedule with ID {id} not found"); |
| | | 128 | | |
| | 0 | 129 | | schedule.IsActive = true; |
| | 0 | 130 | | schedule.UpdatedAt = DateTime.UtcNow; |
| | 0 | 131 | | await _context.SaveChangesAsync(cancellationToken); |
| | | 132 | | |
| | 0 | 133 | | await RegisterScheduleInHangfireAsync(schedule, cancellationToken); |
| | 0 | 134 | | } |
| | | 135 | | |
| | | 136 | | public async Task DeactivateScheduleAsync(int id, CancellationToken cancellationToken = default) |
| | | 137 | | { |
| | 0 | 138 | | var schedule = await _context.BackgroundJobSchedules.FindAsync(new object[] { id }, cancellationToken); |
| | 0 | 139 | | if (schedule == null) |
| | 0 | 140 | | throw new InvalidOperationException($"Schedule with ID {id} not found"); |
| | | 141 | | |
| | 0 | 142 | | schedule.IsActive = false; |
| | 0 | 143 | | schedule.UpdatedAt = DateTime.UtcNow; |
| | 0 | 144 | | await _context.SaveChangesAsync(cancellationToken); |
| | | 145 | | |
| | 0 | 146 | | await UnregisterScheduleFromHangfireAsync(id, cancellationToken); |
| | 0 | 147 | | } |
| | | 148 | | |
| | | 149 | | public async Task RegisterAllActiveSchedulesAsync(CancellationToken cancellationToken = default) |
| | | 150 | | { |
| | 1 | 151 | | var activeSchedules = await GetActiveSchedulesAsync(cancellationToken); |
| | 4 | 152 | | foreach (var schedule in activeSchedules) |
| | | 153 | | { |
| | 1 | 154 | | await RegisterScheduleInHangfireAsync(schedule, cancellationToken); |
| | | 155 | | } |
| | 1 | 156 | | } |
| | | 157 | | |
| | | 158 | | public async Task RegisterScheduleInHangfireAsync(BackgroundJobSchedule schedule, CancellationToken cancellationToke |
| | | 159 | | { |
| | 9 | 160 | | var jobId = $"background-job-schedule-{schedule.Id}"; |
| | | 161 | | |
| | 9 | 162 | | _recurringJobManager.RemoveIfExists(jobId); |
| | 9 | 163 | | if (string.Equals(schedule.JobType, "CustomerReturn", StringComparison.OrdinalIgnoreCase)) |
| | | 164 | | { |
| | 1 | 165 | | _recurringJobManager.AddOrUpdate<IAutomatedTransactionProcessingService>( |
| | 1 | 166 | | jobId, |
| | 1 | 167 | | service => service.ProcessCustomerReturnAsync(schedule.Id, CancellationToken.None), |
| | 1 | 168 | | schedule.CronExpression, |
| | 1 | 169 | | new RecurringJobOptions |
| | 1 | 170 | | { |
| | 1 | 171 | | TimeZone = RussianTimeZoneHelper.Zone |
| | 1 | 172 | | }); |
| | | 173 | | } |
| | | 174 | | else |
| | | 175 | | { |
| | 8 | 176 | | _recurringJobManager.AddOrUpdate<IAutomatedTransactionProcessingService>( |
| | 8 | 177 | | jobId, |
| | 8 | 178 | | service => service.ProcessShippedToCommissionerAsync(schedule.Id, CancellationToken.None), |
| | 8 | 179 | | schedule.CronExpression, |
| | 8 | 180 | | new RecurringJobOptions |
| | 8 | 181 | | { |
| | 8 | 182 | | TimeZone = RussianTimeZoneHelper.Zone |
| | 8 | 183 | | }); |
| | | 184 | | } |
| | | 185 | | |
| | 9 | 186 | | var tracked = await _context.BackgroundJobSchedules.FindAsync(new object[] { schedule.Id }, cancellationToken); |
| | 9 | 187 | | if (tracked != null) |
| | | 188 | | { |
| | 9 | 189 | | tracked.NextRunTime = GetNextRunTime(schedule.CronExpression); |
| | 9 | 190 | | tracked.UpdatedAt = DateTime.UtcNow; |
| | 9 | 191 | | await _context.SaveChangesAsync(cancellationToken); |
| | | 192 | | } |
| | 9 | 193 | | } |
| | | 194 | | |
| | | 195 | | public async Task UnregisterScheduleFromHangfireAsync(int scheduleId, CancellationToken cancellationToken = default) |
| | | 196 | | { |
| | 2 | 197 | | var jobId = $"background-job-schedule-{scheduleId}"; |
| | 2 | 198 | | _recurringJobManager.RemoveIfExists(jobId); |
| | | 199 | | |
| | 2 | 200 | | var schedule = await _context.BackgroundJobSchedules.FindAsync(new object[] { scheduleId }, cancellationToken); |
| | 2 | 201 | | if (schedule != null) |
| | | 202 | | { |
| | 2 | 203 | | schedule.NextRunTime = null; |
| | 2 | 204 | | await _context.SaveChangesAsync(cancellationToken); |
| | | 205 | | } |
| | 2 | 206 | | } |
| | | 207 | | |
| | | 208 | | private DateTime? GetNextRunTime(string cronExpression) |
| | | 209 | | { |
| | | 210 | | try |
| | | 211 | | { |
| | 9 | 212 | | var schedule = Cronos.CronExpression.Parse(cronExpression); |
| | 9 | 213 | | var russianTz = RussianTimeZoneHelper.Zone; |
| | 9 | 214 | | var nowInRussia = TimeZoneInfo.ConvertTimeFromUtc(DateTime.UtcNow, russianTz); |
| | 9 | 215 | | var next = schedule.GetNextOccurrence(nowInRussia, russianTz); |
| | 0 | 216 | | return next.HasValue ? TimeZoneInfo.ConvertTimeToUtc(next.Value, russianTz) : null; |
| | | 217 | | } |
| | 9 | 218 | | catch |
| | | 219 | | { |
| | 9 | 220 | | return null; |
| | | 221 | | } |
| | 9 | 222 | | } |
| | | 223 | | |
| | | 224 | | private string GenerateScheduleName(BackgroundJobSchedule schedule) |
| | | 225 | | { |
| | | 226 | | try |
| | | 227 | | { |
| | 1 | 228 | | var client = _context.OzonClients.Find(schedule.OzonClientId); |
| | 1 | 229 | | var clientName = client?.Name ?? $"Client#{schedule.OzonClientId}"; |
| | | 230 | | |
| | 1 | 231 | | var jobTypeName = schedule.JobType switch |
| | 1 | 232 | | { |
| | 1 | 233 | | "ShippedToCommissioner" => "Отгружен комиссионеру", |
| | 0 | 234 | | "CustomerReturn" => "Возврат покупателя", |
| | 0 | 235 | | _ => schedule.JobType |
| | 1 | 236 | | }; |
| | | 237 | | |
| | 1 | 238 | | var periodText = FormatDaysBack(schedule.DaysBack); |
| | 1 | 239 | | var cronDesc = FormatCronExpression(schedule.CronExpression); |
| | | 240 | | |
| | 1 | 241 | | return $"{jobTypeName} {periodText} для {clientName} ({cronDesc})"; |
| | | 242 | | } |
| | 0 | 243 | | catch |
| | | 244 | | { |
| | 0 | 245 | | return $"Расписание #{Guid.NewGuid().ToString().Substring(0, 8)}"; |
| | | 246 | | } |
| | 1 | 247 | | } |
| | | 248 | | |
| | | 249 | | private static void NormalizeTransactionJobDaysBack(BackgroundJobSchedule schedule) |
| | | 250 | | { |
| | 5 | 251 | | if (IsTransactionJobType(schedule.JobType)) |
| | 5 | 252 | | schedule.DaysBack = ReportDeliveryDateFilter.TransactionJobDaysBack; |
| | 5 | 253 | | } |
| | | 254 | | |
| | | 255 | | private static bool IsTransactionJobType(string jobType) => |
| | 5 | 256 | | string.Equals(jobType, "ShippedToCommissioner", StringComparison.OrdinalIgnoreCase) |
| | 5 | 257 | | || string.Equals(jobType, "CustomerReturn", StringComparison.OrdinalIgnoreCase); |
| | | 258 | | |
| | 1 | 259 | | private string FormatDaysBack(int days) => "за вчера"; |
| | | 260 | | |
| | | 261 | | private string FormatCronExpression(string cron) |
| | | 262 | | { |
| | | 263 | | try |
| | | 264 | | { |
| | 1 | 265 | | var parts = cron.Split(' '); |
| | 1 | 266 | | if (parts.Length < 5) return cron; |
| | | 267 | | |
| | 1 | 268 | | var minute = parts[0]; |
| | 1 | 269 | | var hour = parts[1]; |
| | 1 | 270 | | var dayOfMonth = parts[2]; |
| | 1 | 271 | | var month = parts[3]; |
| | 1 | 272 | | var dayOfWeek = parts[4]; |
| | | 273 | | |
| | 1 | 274 | | if (dayOfMonth == "*" && month == "*" && dayOfWeek == "*") |
| | | 275 | | { |
| | 1 | 276 | | if (hour.StartsWith("*/")) |
| | | 277 | | { |
| | 0 | 278 | | var interval = hour.Substring(2); |
| | 0 | 279 | | return $"каждые {interval}ч"; |
| | | 280 | | } |
| | 1 | 281 | | if (hour != "*") |
| | | 282 | | { |
| | 1 | 283 | | return $"ежедневно в {hour.PadLeft(2, '0')}:{minute.PadLeft(2, '0')}"; |
| | | 284 | | } |
| | 0 | 285 | | return "каждый час"; |
| | | 286 | | } |
| | | 287 | | |
| | 0 | 288 | | if (dayOfWeek != "*" && dayOfMonth == "*") |
| | | 289 | | { |
| | 0 | 290 | | var dayNames = new[] { "Вс", "Пн", "Вт", "Ср", "Чт", "Пт", "Сб" }; |
| | 0 | 291 | | var dayName = int.TryParse(dayOfWeek, out var dayIndex) && dayIndex < 7 |
| | 0 | 292 | | ? dayNames[dayIndex] |
| | 0 | 293 | | : dayOfWeek; |
| | 0 | 294 | | return $"{dayName} в {hour.PadLeft(2, '0')}:{minute.PadLeft(2, '0')}"; |
| | | 295 | | } |
| | | 296 | | |
| | 0 | 297 | | if (dayOfMonth != "*" && month == "*") |
| | | 298 | | { |
| | 0 | 299 | | return $"{dayOfMonth}-го числа в {hour.PadLeft(2, '0')}:{minute.PadLeft(2, '0')}"; |
| | | 300 | | } |
| | | 301 | | |
| | 0 | 302 | | return cron; |
| | | 303 | | } |
| | 0 | 304 | | catch |
| | | 305 | | { |
| | 0 | 306 | | return cron; |
| | | 307 | | } |
| | 1 | 308 | | } |
| | | 309 | | } |