diff --git a/src/Services/Bot/src/Data/Repositories/BotRepository.cs b/src/Services/Bot/src/Data/Repositories/BotRepository.cs index 9da349a..64b65bf 100644 --- 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 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(); + } } \ No newline at end of file diff --git a/src/Services/Bot/src/Data/Repositories/IBotRepository.cs b/src/Services/Bot/src/Data/Repositories/IBotRepository.cs index 2c52ffc..abafde9 100644 --- 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 CreateAsync(Models.Bot bot, CancellationToken cancellationToken = default); Task GetBotAsync(Guid id, CancellationToken cancellationToken);