C# - 使用内联 SQL 恢复 SQL Server 数据库

Question

我正在构建一个 C# 控制台应用程序，它使用可以在开关菜单中修改的预定义参数来恢复 SQL Server 数据库。此应用程序的重点是简单性。它专为任何人设计，只需按一下按钮即可恢复 SQL Server 数据库。是的，应该有人可以胜任这项任务，但是有很多方法可以完成……不管我已经建立了循环遍历备份目录的逻辑，并根据一个点挑选出正确的备份文件及时。除实际恢复部分外，一切正常。

从我读过的其他问题来看，我担心无法从以这种方式修改服务器的 C# 运行 SQL。我不愿意使用 SMO 对象，因为这涉及我想避免的额外复杂性，但如果这是我可以做到这一点的唯一方法，那么我就会这样做。

每当我尝试运行此代码时，它都会提示 @dbName 值无效。

public static void RestoreDatabase(string Server_Name, string Instance_Name, string DB_Name, FileInfo BakFile, FileInfo DiffFile, FileInfo TrnFile, DateTime Point_In_Time)
{
    SqlConnection conn = new SqlConnection();
    conn.ConnectionString = "Data Source=" + Server_Name + "\\" + Instance_Name + ";Initial Catalog=master;Integrated Security=True";

    string SqlQuery = @"ALTER DATABASE @dbName
                        SET SINGLE_USER
                        WITH ROLLBACK IMMEDIATE;

                        RESTORE DATABASE @dbName
                        FROM DISK = @BakFilePath

                        WITH NORECOVERY, REPLACE;

                        RESTORE DATABASE @dbName
                        FROM DISK = @DiffFilePath
                        WITH NORECOVERY;

                        RESTORE DATABASE @dbName
                        FROM DISK = @TrnFilePath
                        WITH RECOVERY, STOPAT = @RecoveryTime;

                        ALTER DATABASE @dbName
                        SET MULTI_USER;";

    SqlCommand cmd = new SqlCommand();
    cmd.CommandType = System.Data.CommandType.Text;
    cmd.Connection = conn;
    cmd.CommandText = SqlQuery;

    try
    {
        cmd.Parameters.Add(new SqlParameter("@dbName", SqlDbType.NVarChar, 30));
        cmd.Parameters.Add(new SqlParameter("@BakFilePath", SqlDbType.NVarChar, 255));
        cmd.Parameters.Add(new SqlParameter("@DiffFilePath", SqlDbType.NVarChar, 255));
        cmd.Parameters.Add(new SqlParameter("@TrnFilePath", SqlDbType.NVarChar, 255));
        cmd.Parameters.Add(new SqlParameter("@RecoveryTime", SqlDbType.DateTime));
        cmd.Parameters["@dbName"].Value = DB_Name;
        cmd.Parameters["@BakFilePath"].Value = BakFile.FullName;
        cmd.Parameters["@DiffFilePath"].Value = DiffFile.FullName;
        cmd.Parameters["@TrnFilePath"].Value = TrnFile.FullName;
        cmd.Parameters["@RecoveryTime"].Value = Point_In_Time;
    }
    catch (Exception ex)
    {
        Console.WriteLine(ex);
        Console.ReadLine();
    }

    try
    {
        Console.WriteLine("Restoring {0}...", DB_Name);

        cmd.Connection.Open();
        cmd.ExecuteNonQuery();
        cmd.Connection.Close();

        Console.WriteLine("Restore Complete!");
        Console.ReadLine();
    }
    catch (SqlException ex)
    {
        Console.WriteLine("Connection could not open. Error: {0}", ex);
        Console.ReadLine();
    }     
}

2016 年 8 月 15 日 10:38

根据一些答案，你们推荐动态 SQL。我担心 SQL 注入(inject)，但只有管理员才能访问该程序。无论如何，我仍然会添加更严格的验证，因为谁不喜欢将垃圾输入用于踢球。

我尝试过动态 SQL。这是我的代码，但是我没有花太多时间检查错误，所以我会再次走这条路并尝试清除所有错误。

string SqlQuery = string.Format(@"DECLARE @dbName NVARCHAR(MAX), @strSQL NVarchar(MAX)=N'';
                            SET @dbName = {0};

                            SELECT
                            @strSQL += 'DECLARE @BakFilePath nvarchar(255) = N''{1}'','
                             + N' @DiffFilePath nvarchar(255) = N''{2}'','
                             + N' @TrnFilePath nvarchar(255) = N''{3}'','
                             + N' @RecoveryTime DateTime = N''{4}'''
                             + N' ALTER DATABASE '+ @dbName
                             + N' SET SINGLE_USER'
                             + N' WITH ROLLBACK IMMEDIATE;'
                             + N' RESTORE DATABASE ' + @dbName 
                             + N' FROM DISK = @BakFilePath'
                             + N' WITH NORECOVERY, REPLACE;'
                             + N' RESTORE DATABASE ' + @dbName 
                             + N' FROM DISK = @DiffFilePath'
                             + N' WITH NORECOVERY;'  
                             + N' RESTORE DATABASE ' + @dbName 
                             + N' FROM DISK = @TrnFilePath'
                             + N' WITH RECOVERY, STOPAT = @RecoveryTime;'
                             + N' ALTER DATABASE ' + @dbName 
                             + N' SET MULTI_USER;
                             '
                             EXEC sp_executesql @strSQL",DB_Name, BakFile.FullName, DiffFile.FullName, TrnFile.FullName, Point_In_Time);

Answer 1

不幸的是，您的 SQL 编写方式存在问题。我提供了一个具有类似潜在问题的问题的答案，您可以在这里找到它:

DROP PROCEDURE throws syntax error

sql-server 会提示使用:

Restore Database @dbName

因为它是作为文字字符串传递的。您需要编写类似这样的代码才能执行您的代码。

declare @sql varchar(64);

set @sql = 'RESTORE DATABASE ' + @dbName;

exec(@sql);

更新
根据斯科特的评论，使用:

set @sql = 'RESTORE DATABASE ' + QUOTENAME(@dbName);

将转义字符串以创建有效的 SQL Server 分隔标识符:

https://msdn.microsoft.com/en-us/library/ms176114.aspx