have mqtt subscriber as hosted service

Core/Services/External/BackgroundRemoval/IImageBackgroundRemoverService.cs

@@ -1,4 +1,4 @@
-namespace Core.Services.External;
+namespace Core.Services.External.BackgroundRemoval;
 
 public interface IImageBackgroundRemoverService
 {

Core/Services/External/BackgroundRemoval/MockImageBackgroundRemoverService.cs

@@ -1,3 +1,5 @@
+using Core.Services.External.BackgroundRemoval;
+
 namespace Core.Services.External;
 
 public class MockImageBackgroundRemoverService : IImageBackgroundRemoverService

Core/Services/MqttSubscriberService.cs

@@ -1,17 +1,19 @@
 using System.Text;
 using System.Text.Json;
 using Core.Options;
+using Microsoft.Extensions.Hosting;
 using Microsoft.Extensions.Options;
 using MQTTnet;
 using MQTTnet.Client;
 using Shared.Dtos;
 
 namespace Core.Services;
 
-public class MqttSubscriberService
+public class MqttSubscriberService : IHostedService
 {
     private readonly IOptions<MqttOptions> _options;
     private readonly ConditionsLogsService _conditionsLogService;
+    private IMqttClient? _mqttClient;
 
     public MqttSubscriberService(IOptions<MqttOptions> options, ConditionsLogsService conditionsLogService)
     {
@@ -22,17 +24,17 @@ public MqttSubscriberService(IOptions<MqttOptions> options, ConditionsLogsServic
             throw new Exception("MQTT username not set in appsettings.json");
     }
 
-    public async Task SubscribeAsync()
+    public async Task StartAsync(CancellationToken cancellationToken)
     {
         var mqttFactory = new MqttFactory();
 
-        using var mqttClient = mqttFactory.CreateMqttClient();
+        _mqttClient = mqttFactory.CreateMqttClient();
         var mqttClientOptions = new MqttClientOptionsBuilder()
             .WithTcpServer(_options.Value.Server, _options.Value.Port)
             .WithCredentials(_options.Value.Username)
             .Build();
 
-        mqttClient.ApplicationMessageReceivedAsync += async e =>
+        _mqttClient.ApplicationMessageReceivedAsync += async e =>
         {
             var payload = Encoding.UTF8.GetString(e.ApplicationMessage.PayloadSegment);
             var conditions = JsonSerializer.Deserialize<CreateConditionsLogDto>(payload, options:
@@ -43,15 +45,24 @@ public async Task SubscribeAsync()
             await _conditionsLogService.CreateConditionsLogAsync(conditions);
         };
 
-        await mqttClient.ConnectAsync(mqttClientOptions, CancellationToken.None);
+        await _mqttClient.ConnectAsync(mqttClientOptions, CancellationToken.None);
 
         var mqttSubscribeOptions = mqttFactory.CreateSubscribeOptionsBuilder()
             .WithTopicFilter(
                 f => { f.WithTopic(_options.Value.SubscribeTopic); })
             .Build();
 
-        await mqttClient.SubscribeAsync(mqttSubscribeOptions, CancellationToken.None);
+        await _mqttClient.SubscribeAsync(mqttSubscribeOptions, CancellationToken.None);
 
-        Console.ReadLine();
+        Console.WriteLine($"Subscribed to {_options.Value.SubscribeTopic}");
+    }
+
+    public async Task StopAsync(CancellationToken cancellationToken)
+    {
+        if (_mqttClient != null)
+        {
+            await _mqttClient.DisconnectAsync(cancellationToken: cancellationToken);
+            _mqttClient.Dispose();
+        }
     }
 }

api/Events/ImageUpload/ClientWantsToRemoveBackgroundFromImage.cs

@@ -1,5 +1,6 @@
 using api.Extensions;
 using Core.Services.External;
+using Core.Services.External.BackgroundRemoval;
 using Fleck;
 using lib;
 using Shared.Exceptions;

api/Extensions/AddServicesAndRepositoriesExtension.cs

@@ -25,7 +25,7 @@ public static void AddServicesAndRepositories(this IServiceCollection services)
         services.AddSingleton<CollectionsService>();
         services.AddSingleton<ConditionsLogsService>();
         services.AddSingleton<RequirementService>();
-        services.AddSingleton<MqttSubscriberService>();
+        services.AddHostedService<MqttSubscriberService>();
         services.AddSingleton<MqttPublisherService>();
 
         // External services

api/Program.cs

@@ -151,12 +151,6 @@ public static async Task<WebApplication> StartApi(string[] args)
             };
         });
 
-        /*
-        // Connect and subscribe to MQTT
-        var mqttSubscriberService = app.Services.GetRequiredService<MqttSubscriberService>();
-        _ = mqttSubscriberService.SubscribeAsync();
-        */
-
         return app;
     }
 }