added start hooks for augmentors

bradygaster · Feb 28, 2024 · 71b914c · 71b914c
1 parent 1bb02c7
commit 71b914c
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 2 deletions.
diff --git a/Backend/Augmentors/BaseAugmentor.cs b/Backend/Augmentors/BaseAugmentor.cs
@@ -6,4 +6,5 @@ public abstract class BaseAugmentor(SemanticKernelWrapper semanticKernelWrapper,
     protected ILogger<BaseAugmentor> logger = logger;
 
     public abstract Task Load();
+    public virtual Task OnStarted() => Task.CompletedTask;
 }
diff --git a/Backend/Augmentors/UrlAugmentor.cs b/Backend/Augmentors/UrlAugmentor.cs
@@ -10,10 +10,12 @@ public class UrlAugmentor(SemanticKernelWrapper semanticKernelWrapper,
     private readonly HistoryApiClient historyApiClient = historyApiClient;
     private readonly LiveUpdateService liveUpdateService = liveUpdateService;
 
+    public override async Task OnStarted() =>
+        await queueServiceClient.GetQueueClient("incoming-urls").CreateIfNotExistsAsync();
+
     public override async Task Load()
     {
         var incomingQueueClient = queueServiceClient.GetQueueClient("incoming-urls");
-        await incomingQueueClient.CreateIfNotExistsAsync();
 
         QueueMessage[] messages = await incomingQueueClient.ReceiveMessagesAsync(maxMessages: 8);
 

diff --git a/Backend/Augmentors/UrlListAugmentor.cs b/Backend/Augmentors/UrlListAugmentor.cs
@@ -9,10 +9,12 @@ public class UrlListAugmentor(
     LiveUpdateService liveUpdateService)
         : AzureBlobBaseAugmentor(semanticKernelWrapper, logger, blobServiceClient)
 {
+    public override async Task OnStarted() =>
+        await blobServiceClient.GetBlobContainerClient("incoming-urllist").CreateIfNotExistsAsync();
+
     public override async Task Load()
     {
         var incomingContainerClient = blobServiceClient.GetBlobContainerClient("incoming-urllist");
-        await incomingContainerClient.CreateIfNotExistsAsync();
 
         var archivedContainerClient = blobServiceClient.GetBlobContainerClient("archive-urllist");
         await archivedContainerClient.CreateIfNotExistsAsync();

diff --git a/Backend/Workers/AugmentationWorker.cs b/Backend/Workers/AugmentationWorker.cs
@@ -36,4 +36,10 @@ protected override async Task ExecuteAsync(CancellationToken stoppingToken)
             await liveUpdateService.ShowSystemUpdate("System is up");
         }
     }
+
+    public override async Task StartAsync(CancellationToken cancellationToken)
+    {
+        await urlAugmentor.OnStarted();
+        await urlListAugmentor.OnStarted();
+    }
 }