mirror of
https://github.com/bitwarden/server.git
synced 2024-12-23 17:07:42 +01:00
42 lines
1.3 KiB
C#
42 lines
1.3 KiB
C#
using Bit.Core;
|
|
using Bit.Core.Utilities;
|
|
using Bit.Infrastructure.EntityFramework.Repositories;
|
|
using Microsoft.EntityFrameworkCore;
|
|
using Microsoft.Extensions.DependencyInjection;
|
|
using Microsoft.Extensions.Logging;
|
|
|
|
namespace Bit.PostgresMigrations;
|
|
|
|
public class PostgresDbMigrator : IDbMigrator
|
|
{
|
|
private readonly IServiceScopeFactory _serviceScopeFactory;
|
|
private readonly ILogger<PostgresDbMigrator> _logger;
|
|
|
|
public PostgresDbMigrator(IServiceScopeFactory serviceScopeFactory, ILogger<PostgresDbMigrator> logger)
|
|
{
|
|
_serviceScopeFactory = serviceScopeFactory;
|
|
_logger = logger;
|
|
}
|
|
|
|
public bool MigrateDatabase(bool enableLogging = true,
|
|
CancellationToken cancellationToken = default(CancellationToken))
|
|
{
|
|
if (enableLogging && _logger != null)
|
|
{
|
|
_logger.LogInformation(Constants.BypassFiltersEventId, "Migrating database.");
|
|
}
|
|
|
|
using var scope = _serviceScopeFactory.CreateScope();
|
|
var databaseContext = scope.ServiceProvider.GetRequiredService<DatabaseContext>();
|
|
databaseContext.Database.Migrate();
|
|
|
|
if (enableLogging && _logger != null)
|
|
{
|
|
_logger.LogInformation(Constants.BypassFiltersEventId, "Migration successful.");
|
|
}
|
|
|
|
cancellationToken.ThrowIfCancellationRequested();
|
|
return true;
|
|
}
|
|
}
|