diff options
| author | crupest <crupest@outlook.com> | 2021-02-12 22:10:56 +0800 |
|---|---|---|
| committer | crupest <crupest@outlook.com> | 2021-02-12 22:13:51 +0800 |
| commit | 83de31a999132b052f1ccec75eb664c9e4a18e87 (patch) | |
| tree | 079e6c09044d97c1ee07a83ca8c379facc8c3281 /BackEnd/Timeline/Services/Migration/CustomMigrationManager.cs | |
| parent | 1caf863b870d4b628533db8279e6e5758832eef0 (diff) | |
| download | timeline-83de31a999132b052f1ccec75eb664c9e4a18e87.tar.gz timeline-83de31a999132b052f1ccec75eb664c9e4a18e87.tar.bz2 timeline-83de31a999132b052f1ccec75eb664c9e4a18e87.zip | |
feat: Add databse custom migration service.
Diffstat (limited to 'BackEnd/Timeline/Services/Migration/CustomMigrationManager.cs')
| -rw-r--r-- | BackEnd/Timeline/Services/Migration/CustomMigrationManager.cs | 55 |
1 files changed, 55 insertions, 0 deletions
diff --git a/BackEnd/Timeline/Services/Migration/CustomMigrationManager.cs b/BackEnd/Timeline/Services/Migration/CustomMigrationManager.cs new file mode 100644 index 00000000..ba86e10b --- /dev/null +++ b/BackEnd/Timeline/Services/Migration/CustomMigrationManager.cs @@ -0,0 +1,55 @@ +using System.Collections.Generic;
+using System.Threading.Tasks;
+using Microsoft.EntityFrameworkCore;
+using Microsoft.Extensions.Logging;
+using Timeline.Entities;
+
+namespace Timeline.Services.Migration
+{
+ public interface ICustomMigrationManager
+ {
+ Task Migrate();
+ }
+
+ public class CustomMigrationManager : ICustomMigrationManager
+ {
+ private IEnumerable<ICustomMigration> _migrations;
+ private DatabaseContext _database;
+
+ private ILogger<CustomMigrationManager> _logger;
+
+ public CustomMigrationManager(IEnumerable<ICustomMigration> migrations, DatabaseContext database, ILogger<CustomMigrationManager> logger)
+ {
+ _migrations = migrations;
+ _database = database;
+ _logger = logger;
+ }
+
+ public async Task Migrate()
+ {
+ foreach (var migration in _migrations)
+ {
+ var name = migration.GetName();
+ var did = await _database.Migrations.AnyAsync(m => m.Name == name);
+
+ _logger.LogInformation("Found custom migration '{0}'. Did: {1}.", name, did);
+
+ if (!did)
+ {
+ _logger.LogInformation("Begin custom migration '{0}'.", name);
+
+ await using var transaction = await _database.Database.BeginTransactionAsync();
+
+ await migration.Execute(_database);
+
+ _database.Migrations.Add(new MigrationEntity { Name = name });
+ await _database.SaveChangesAsync();
+
+ await transaction.CommitAsync();
+
+ _logger.LogInformation("End custom migration '{0}'.", name);
+ }
+ }
+ }
+ }
+}
|