feat(restore): logical file rename (#102)

* feat(restore): logical file rename

* fix(logical rename): after chain restored

* refactor(logical rename): files arrays

* fix(rename): rename on primary only

* Revert "fix(rename): rename on primary only"

This reverts commit 2c151d7.

* feat(renamer): public func

* refactor(method): name

* feat(move): logical rename on finalize

Author: mja9

src/AgDatabase.cs

@@ -34,6 +34,7 @@ public interface IAgDatabase
     void Restore(IEnumerable<BackupMetadata> backupOrder, Func<int, TimeSpan> retryDurationProvider,
       Func<string, string> fileRelocation = null);
 
+    void RenameLogicalFileName(Func<string, string> fileRenamer);
     void AddLogin(LoginProperties login);
     IEnumerable<LoginProperties> AssociatedLogins();
     void DropLogin(LoginProperties login);
@@ -118,6 +119,11 @@ public void Restore(IEnumerable<BackupMetadata> backupOrder, Func<int, TimeSpan>
       _listener.ForEachAgInstance(s => s.Restore(backupOrder, Name, retryDurationProvider, fileRelocation));
     }
 
+    public void RenameLogicalFileName(Func<string, string> fileRenamer)
+    {
+      _listener.Primary.RenameLogicalFileName(Name, fileRenamer);
+    }
+
     /// <summary>
     ///   Builds a list of recent backups from msdb on each AG instance.
     /// </summary>

src/AgDatabaseMove.cs

@@ -80,8 +80,10 @@ public decimal Move(decimal? lastLsn = null)
         foreach(var loginProperty in _options.Source.AssociatedLogins().Select(UpdateDefaultDb))
           _options.Destination.AddLogin(loginProperty);
 
-      if(_options.Finalize)
+      if(_options.Finalize) {
         _options.Destination.JoinAg();
+        _options.Destination.RenameLogicalFileName(_options.FileRelocator);
+      }
 
       return backupList.Max(bl => bl.LastLsn);
     }

src/SmoFacade/Database.cs

@@ -12,7 +12,7 @@ namespace AgDatabaseMove.SmoFacade
   /// </summary>
   public class Database
   {
-    private readonly Microsoft.SqlServer.Management.Smo.Database _database;
+    internal readonly Microsoft.SqlServer.Management.Smo.Database _database;
     private readonly Server _server;
 
     internal Database(Microsoft.SqlServer.Management.Smo.Database database, Server server)

src/SmoFacade/Server.cs

@@ -227,6 +227,23 @@ public void Restore(IEnumerable<BackupMetadata> backupOrder, string databaseName
       }
     }
 
+    public void RenameLogicalFileName(string databaseName, Func<string, string> fileRenamer)
+    {
+      var db = Database(databaseName);
+
+      foreach (FileGroup fileGroup in db._database.FileGroups)
+      {
+        var dataFiles = new DataFile[fileGroup.Files.Count];
+        fileGroup.Files.CopyTo(dataFiles, 0);
+        foreach (var dataFile in dataFiles)
+          dataFile.Rename(fileRenamer(dataFile.Name));
+      }
+      var logFiles = new LogFile[db._database.LogFiles.Count];
+      db._database.LogFiles.CopyTo(logFiles, 0);
+      foreach (var logFile in logFiles)
+        logFile.Rename(fileRenamer(logFile.Name));
+    }
+
     /// <summary>
     ///   Generate a log backup, truncating the transaction log, and storing it in the default backup destination.
     ///   TODO: we should support a backup path somehow with configuration