Made report-repositories disposable

CodebreakerApp · Jun 28, 2023 · c943df2 · c943df2
1 parent b1290e4
commit c943df2
Show file tree

Hide file tree

Showing 5 changed files with 17 additions and 7 deletions.
diff --git a/src/Services/Report/src/Data/Repositories/GameRepository.cs b/src/Services/Report/src/Data/Repositories/GameRepository.cs
@@ -8,7 +8,7 @@
 
 namespace CodeBreaker.Services.Report.Data.Repositories;
 
-public class GameRepository(ReportDbContext dbContext) : IGameRepository
+public class GameRepository(ReportDbContext dbContext) : IGameRepository, IDisposable
 {
     public IAsyncEnumerable<Game> GetAsync(GetGamesArgs args, CancellationToken cancellationToken = default) =>
         dbContext.Games
@@ -92,4 +92,9 @@ private static void SyncPositions(IEnumerable<KeyPeg> keyPegs)
         foreach (var keyPeg in keyPegs)
             keyPeg.Position = i++;
     }
+
+    public void Dispose()
+    {
+        dbContext.Dispose();
+    }
 }
diff --git a/src/Services/Report/src/Data/Repositories/IGameRepository.cs b/src/Services/Report/src/Data/Repositories/IGameRepository.cs
@@ -2,7 +2,7 @@
 using CodeBreaker.Services.Report.Data.Repositories.Args;
 
 namespace CodeBreaker.Services.Report.Data.Repositories;
-public interface IGameRepository
+public interface IGameRepository : IDisposable
 {
     Task<Game> CreateAsync(Game game, CancellationToken cancellationToken = default);
     Task DeleteAsync(Guid gameId, CancellationToken cancellationToken = default);

diff --git a/src/Services/Report/src/Data/Repositories/IStatisticsRepository.cs b/src/Services/Report/src/Data/Repositories/IStatisticsRepository.cs
@@ -1,7 +1,7 @@
 using CodeBreaker.Services.Report.Data.Models;
 
 namespace CodeBreaker.Services.Report.Data.Repositories;
-public interface IStatisticsRepository
+public interface IStatisticsRepository : IDisposable
 {
     Task<Statistics> GetStatisticsAsync(GetStatisticsArgs args, CancellationToken cancellationToken = default);
 }
diff --git a/src/Services/Report/src/Data/Repositories/StatisticsRepository.cs b/src/Services/Report/src/Data/Repositories/StatisticsRepository.cs
@@ -4,7 +4,7 @@
 
 namespace CodeBreaker.Services.Report.Data.Repositories;
 
-public class StatisticsRepository(ReportDbContext dbContext) : IStatisticsRepository
+public class StatisticsRepository(ReportDbContext dbContext) : IStatisticsRepository, IDisposable
 {
     private class StatisticsResult
     {
@@ -37,6 +37,11 @@ public async Task<Statistics> GetStatisticsAsync(GetStatisticsArgs args, Cancell
             })
             .SingleAsync(cancellationToken);
     }
+
+    public void Dispose()
+    {
+        dbContext.Dispose();
+    }
 }
 
 

diff --git a/src/Services/Report/src/MessageWorker/Services/MessageService.cs b/src/Services/Report/src/MessageWorker/Services/MessageService.cs
@@ -46,21 +46,21 @@ public async Task RunAsync(CancellationToken cancellationToken = default)
     private async Task OnGameCreatedCallbackAsync(GameCreatedPayload payload, CancellationToken cancellationToken)
     {
         _logger.LogInformation("OnGameCreated");
-        var repository = _services.GetRequiredService<IGameRepository>();
+        using var repository = _services.GetRequiredService<IGameRepository>();
         await repository.CreateAsync(payload.ToModel(), cancellationToken);
     }
 
     private async Task OnMoveCreatedCallbackAsync(MoveCreatedPayload payload, CancellationToken cancellationToken)
     {
         _logger.LogInformation("OnMoveCreated");
-        var repository = _services.GetRequiredService<IGameRepository>();
+        using var repository = _services.GetRequiredService<IGameRepository>();
         await repository.AddMoveAsync(payload.GameId, payload.ToMoveModel(), cancellationToken);
     }
 
     private async Task OnGameEndedCallbackAsync(GameEndedPayload payload, CancellationToken cancellationToken)
     {
         _logger.LogInformation("OnGameEnded");
-        var repository = _services.GetRequiredService<IGameRepository>();
+        using var repository = _services.GetRequiredService<IGameRepository>();
         var game = await repository.GetAsync(payload.Id, cancellationToken);
         payload.ToModel(game);
         await repository.UpdateAsync(payload.Id, game, cancellationToken);