diff --git a/Backend/Augmentors/BaseAugmentor.cs b/Backend/Augmentors/BaseAugmentor.cs index 774807c..617b5d4 100644 --- a/Backend/Augmentors/BaseAugmentor.cs +++ b/Backend/Augmentors/BaseAugmentor.cs @@ -6,4 +6,5 @@ public abstract class BaseAugmentor(SemanticKernelWrapper semanticKernelWrapper, protected ILogger logger = logger; public abstract Task Load(); + public virtual Task OnStarted() => Task.CompletedTask; } diff --git a/Backend/Augmentors/UrlAugmentor.cs b/Backend/Augmentors/UrlAugmentor.cs index a33876f..2b98214 100644 --- 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 index 5985c23..dbca533 100644 --- 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 index 6d1ea01..89f5059 100644 --- 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(); + } }