fix for fresh startup (#112)

mlapaglia · Jan 29, 2024 · 114e557 · 114e557
1 parent c169956
commit 114e557
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/OpenAlprWebhookProcessor/Hydration/HydrationService.cs b/OpenAlprWebhookProcessor/Hydration/HydrationService.cs
@@ -68,6 +68,13 @@ public async Task ScheduleHydrationAsync(CancellationToken cancellationToken)
 
                 var agent = await processorContext.Agents.FirstOrDefaultAsync(cancellationToken);
 
+                if (string.IsNullOrWhiteSpace(agent.Uid))
+                {
+                    var logger = scope.ServiceProvider.GetRequiredService<ILogger<HydrationService>>();
+                    logger.LogWarning("Agent UID is not set. Cannot schedule hydration.");
+                    return;
+                }
+
                 if (agent.ScheduledScrapingIntervalMinutes == null)
                 {
                     RecurringJob.RemoveIfExists(agent.Uid);

diff --git a/OpenAlprWebhookProcessor/Startup.cs b/OpenAlprWebhookProcessor/Startup.cs
@@ -84,6 +84,15 @@ public void ConfigureServices(IServiceCollection services)
             using (var context = new ProcessorContext(processorOptionsBuilder.Options))
             {
                 context.Database.Migrate();
+                var agent = context.Agents.FirstOrDefault();
+
+                if (agent == null)
+                {
+                    agent = new Data.Agent();
+
+                    context.Agents.Add(agent);
+                    context.SaveChanges();
+                }
             }
 
             var optionsBuilder = new DbContextOptionsBuilder<UsersContext>();
@@ -202,7 +211,7 @@ public void ConfigureServices(IServiceCollection services)
 
             services.AddSingleton<WebPushNotificationProducer>();
             services.AddSingleton<IHostedService>(p => p.GetService<WebPushNotificationProducer>());
-            
+
             services.AddSingleton<WebsocketClientOrganizer>();
             services.AddSingleton<IHostedService>(p => p.GetService<WebsocketClientOrganizer>());