Added ready-healthcheck for message-publisher to game-service

CodebreakerApp · Jun 29, 2023 · d53d828 · d53d828
1 parent c15d494
commit d53d828
Show file tree

Hide file tree

Showing 4 changed files with 26 additions and 3 deletions.
diff --git a/src/Services/Game/src/GameService/HealthChecks/MessagePublisherReadyCheck.cs b/src/Services/Game/src/GameService/HealthChecks/MessagePublisherReadyCheck.cs
@@ -0,0 +1,12 @@
+using CodeBreaker.Services.Games.Messaging.Services;
+using Microsoft.Extensions.Diagnostics.HealthChecks;
+
+namespace CodeBreaker.Services.Games.HealthChecks;
+
+internal class MessagePublisherReadyCheck(IMessagePublisher messagePublisher) : IHealthCheck
+{
+    public Task<HealthCheckResult> CheckHealthAsync(HealthCheckContext context, CancellationToken cancellationToken = default) =>
+        messagePublisher.IsClosed
+            ? Task.FromResult(HealthCheckResult.Unhealthy())
+            : Task.FromResult(HealthCheckResult.Healthy());
+}
diff --git a/src/Services/Game/src/GameService/Program.cs b/src/Services/Game/src/GameService/Program.cs
@@ -3,6 +3,7 @@
 using CodeBreaker.Services.Games.Data.Repositories;
 using CodeBreaker.Services.Games.Endpoints;
 using CodeBreaker.Services.Games.Extensions;
+using CodeBreaker.Services.Games.HealthChecks;
 using CodeBreaker.Services.Games.Messaging.Services;
 using CodeBreaker.Services.Games.Services;
 using GameService.Serialization;
@@ -36,15 +37,19 @@
 #endif
 });
 
+var serviceBusNamespace = builder.Configuration.GetRequired("GameService:ServiceBus:Namespace");
 builder.Services.AddAzureClients(clientBuilder =>
 {
-    var serviceBusNamespace = builder.Configuration.GetRequired("GameService:ServiceBus:Namespace");
     clientBuilder.AddServiceBusClientWithNamespace(serviceBusNamespace);
     clientBuilder.UseCredential(azureCredential);
 });
 
-builder.Services.AddSingleton<IMessagePublisher, MessagePublisher>(); builder.Services.AddHealthChecks()
-    .AddDbContextCheck<GamesDbContext>("Ready", tags: new[] { "ready" });
+builder.Services.AddSingleton<IMessagePublisher, MessagePublisher>();
+
+string[] readyHealthCheckTags = { "ready" };
+builder.Services.AddHealthChecks()
+    .AddDbContextCheck<GamesDbContext>("DbReady", tags: readyHealthCheckTags)
+    .AddCheck<MessagePublisherReadyCheck>("MessagePublisherReady", tags: readyHealthCheckTags);
 
 builder.Services.ConfigureHttpJsonOptions(options =>
 {

diff --git a/src/Services/Game/src/Messaging/Services/IMessagePublisher.cs b/src/Services/Game/src/Messaging/Services/IMessagePublisher.cs
@@ -3,6 +3,7 @@
 namespace CodeBreaker.Services.Games.Messaging.Services;
 public interface IMessagePublisher
 {
+    bool IsClosed { get; }
     Task PublishGameCreatedAsync(GameCreatedPayload payload, CancellationToken cancellationToken);
     Task PublishGameEndedAsync(GameEndedPayload payload, CancellationToken cancellationToken);
     Task PublishMoveCreatedAsync(MoveCreatedPayload payload, CancellationToken cancellationToken);

diff --git a/src/Services/Game/src/Messaging/Services/MessagePublisher.cs b/src/Services/Game/src/Messaging/Services/MessagePublisher.cs
@@ -16,6 +16,11 @@ public class MessagePublisher(ServiceBusClient serviceBusClient) : IMessagePubli
 
     private readonly ServiceBusSender _gameEndedSender = serviceBusClient.CreateSender(MessagingTopics.Game_Ended);
 
+    public bool IsClosed =>
+        _gameCreatedSender.IsClosed ||
+        _moveCreatedSender.IsClosed ||
+        _gameEndedSender.IsClosed;
+
     public async Task PublishGameCreatedAsync(GameCreatedPayload payload, CancellationToken cancellationToken) =>
         await Publish(_gameCreatedSender, payload, cancellationToken);