Question

我正在尝试将数据库对象的脚本备份到磁盘。我已经能够将下面的代码组合在一起，但我必须指定每个对象类型（例如表，函数，存储过程）。

有没有办法在不指定每个集合的情况下遍历所有这些对象？

using System;
using System.Data;
using Microsoft.SqlServer.Management.Smo;
using Microsoft.SqlServer.Management.Common;
using System.IO;
using System.Text;

public static void BackupDB(Server srv, Database db, string savePath)
{
    string objPath;
    string objFile;
    string objTxt;

    foreach (StoredProcedure obj in db.StoredProcedures) 
    {
        if (obj.IsSystemObject == false)
        {
            objPath = savePath + "Stored Procedures\\";
            Directory.CreateDirectory(objPath);

            objFile = objPath + obj.Schema + "." + obj.Name + ".sql";
            objTxt = GetScriptString(srv, obj);
            File.WriteAllText(objFile, objTxt);
        }
    }

    foreach (Table obj in db.Tables)
    {
        if (obj.IsSystemObject == false)
        {
            objPath = savePath + "Tables\\";
            Directory.CreateDirectory(objPath);

            objFile = objPath + obj.Schema + "." + obj.Name + ".txt";
            objTxt = GetScriptString(srv, obj);
            File.WriteAllText(objFile, objTxt);
        }
    }

static private string GetScriptString(Server server, SqlSmoObject obj)
{
    StringBuilder output = new StringBuilder();
    Scripter scr = new Scripter(server);
    var script = scr.EnumScript(new SqlSmoObject[] { obj });
    foreach (var line in script)
    {
        output.AppendLine(line);
    }
    return output.ToString();
}

编辑：使用Grant Winney的答案，我能够将下面的方法放在一起来迭代遍历所有对象而不重复行：

public static void BackupDB(Server srv, Database db, string savePath)
{
    string objPath;
    string objFile;
    string objTxt;

    Dictionary<string, IEnumerable<ScriptSchemaObjectBase>> dboDict =
        new Dictionary<string, IEnumerable<ScriptSchemaObjectBase>>();

    dboDict.Add("Stored Procedures", db.StoredProcedures.Cast<StoredProcedure>().Where(x => !x.IsSystemObject));
    dboDict.Add("Functions", db.UserDefinedFunctions.Cast<UserDefinedFunction>().Where(x => !x.IsSystemObject));
    dboDict.Add("Tables", db.Tables.Cast<Table>().Where(x => !x.IsSystemObject));
    dboDict.Add("Views", db.Views.Cast<View>().Where(x => !x.IsSystemObject));

    foreach (KeyValuePair<string, IEnumerable<ScriptSchemaObjectBase>> dict in dboDict)
    {
        IEnumerable<ScriptSchemaObjectBase> dboObjects = dict.Value;
        objPath = savePath + dict.Key + "\\";
        Directory.CreateDirectory(objPath);

        foreach (ScriptSchemaObjectBase obj in dboObjects)
        {
            objFile = objPath + obj.Schema + "." + obj.Name + ".sql";
            objTxt = GetScriptString(srv, obj);
            File.WriteAllText(objFile, objTxt);
        }
    }

}

Answer 1

将类似的代码移动到一个单独的方法中，并利用每个类实现相同基类的事实，即Name和public static void BackupDB(Server srv, Database db, string savePath) { string objPath; string objFile; string objTxt; BackupObjects(db.StoredProcedures.Cast<StoredProcedure>().Where(x => !x.IsSystemObject)); BackupObjects(db.Tables.Cast<Table>().Where(x => !x.IsSystemObject)); } private static void BackupObjects(IEnumerable<ScriptSchemaObjectBase> objects) { foreach (ScriptSchemaObjectBase obj in objects) { objPath = savePath + "Stored Procedures\\"; Directory.CreateDirectory(objPath); objFile = objPath + obj.Schema + "." + obj.Name + ".sql"; objTxt = GetScriptString(srv, obj); File.WriteAllText(objFile, objTxt); } }所在的基类。

您可以先使用LINQ过滤项目，然后将过滤后的列表传递给其他方法。

str

如何使用SMO在C＃中迭代数据库对象

1 个答案: