Added IDisposable to bot-repository

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

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/src/Services/Bot/src/Data/Repositories/BotRepository.cs b/src/Services/Bot/src/Data/Repositories/BotRepository.cs
@@ -4,11 +4,10 @@
 using CodeBreaker.Services.Bot.Data.Models.Extensions;
 using CodeBreaker.Services.Bot.Data.Repositories.Args;
 using Microsoft.EntityFrameworkCore;
-using Microsoft.EntityFrameworkCore.Query;
 
 namespace CodeBreaker.Services.Bot.Data.Repositories;
 
-public class BotRepository(BotDbContext dbContext) : IBotRepository
+public class BotRepository(BotDbContext dbContext) : IBotRepository, IDisposable
 {
     public IAsyncEnumerable<Models.Bot> GetBotsAsync(GetBotsArgs args) =>
         dbContext.Bots
@@ -42,4 +41,9 @@ await queryable.ExecuteUpdateAsync(x => x
         else
             await queryable.ExecuteUpdateAsync(x => x.SetProperty(bot => bot.State, newState), cancellationToken);
     }
+
+    public void Dispose()
+    {
+        dbContext.Dispose();
+    }
 }
diff --git a/src/Services/Bot/src/Data/Repositories/IBotRepository.cs b/src/Services/Bot/src/Data/Repositories/IBotRepository.cs
@@ -2,7 +2,7 @@
 using CodeBreaker.Services.Bot.Data.Repositories.Args;
 
 namespace CodeBreaker.Services.Bot.Data.Repositories;
-public interface IBotRepository
+public interface IBotRepository : IDisposable
 {
     Task<Models.Bot> CreateAsync(Models.Bot bot, CancellationToken cancellationToken = default);
     Task<Models.Bot> GetBotAsync(Guid id, CancellationToken cancellationToken);