如何将大型SQL脚本分批解析

本文关键字:脚本 SQL 大型 | 更新日期: 2023-09-27 18:11:29

我有一个非常大的sql文件,我想把它分成几批执行。我想确保我解析它的方式与SSMS和SQLCMD相同。

Microsoft有一个很棒的混合模式程序集,名为Microsoft. sqlserver . batchparser,其中有一个名为Parser的类,看起来它可以做到这一点。

在调用Parse()之前,它需要一个IBatchSource的实现作为SetBatchSource的参数。

我在哪里可以找到IBatchSource的实现,以及如何使用此功能的更多信息?

如何将大型SQL脚本分批解析

我在GAC中找到了Microsoft.SqlServer.BatchParser以及它的朋友Microsoft.SqlServer.BatchParserClient,其中包含实现接口IBatchSource。

namespace Microsoft.SqlServer.Management.Common
{
  internal class BatchSourceFile : IBatchSource
  internal class BatchSourceString : IBatchSource
}

随后发生了以下对话。

组装:

你好!我叫 Microsoft.SqlServer.Management.Common.ExecuteBatch 。你想StringCollection GetStatements(字符串sqlCommand)吗?

Me:是的,我会,BatchParserClient程序集。谢谢你的关心!

可重复指令(在家里试试!)

  • 安装Microsoft SQL Server 2008 R2共享管理对象
  • 将Microsoft.SqlServer.BatchParser.dll和Microsoft.SqlServer.BatchParserClient.dll从GAC拷贝到您的解决方案的文件夹中。
  • 参考Microsoft.SqlServer.BatchParser &;Microsoft.SqlServer.BatchParserClient

Program.cs

using System;
using System.Collections.Specialized;
using System.IO;
using System.Text;
using Microsoft.SqlServer.Management.Common;
namespace ScriptParser
{
   class Program
   {
      static void Main(string[] args)
      {
         ExecuteBatch batcher = new ExecuteBatch();
         string text = File.ReadAllText(@"Path_To_My_Long_Sql_File.sql");
         StringCollection statements = batcher.GetStatements(text);
         foreach (string statement in statements)
         {
            Console.WriteLine(statement);
         }
      }
   }
}

应用程序。配置

<?xml version="1.0"?>
<configuration>
   <startup useLegacyV2RuntimeActivationPolicy="true">
      <supportedRuntime version="v4.0" sku=".NETFramework,Version=v4.5"/>
   </startup>
</configuration>

另一个选项是使用如下答案所述的ScriptDom: https://stackoverflow.com/a/32529415/26877.

using System;
using System.Collections.Generic;
using System.IO;
using Microsoft.SqlServer.TransactSql.ScriptDom;
namespace ScriptDomDemo
{
    class Program
    {
        static void Main(string[] args)
        {
            TSql120Parser parser = new TSql120Parser(false);
            IList<ParseError> errors;
            using (StringReader sr = new StringReader(@"create table t1 (c1 int primary key)
GO
create table t2 (c1 int primary key)"))
            {
                TSqlFragment fragment = parser.Parse(sr, out errors);
                IEnumerable<string> batches = GetBatches(fragment);
                foreach (var batch in batches)
                {
                    Console.WriteLine(batch);
                }
            }
        }
        private static IEnumerable<string> GetBatches(TSqlFragment fragment)
        {
            Sql120ScriptGenerator sg = new Sql120ScriptGenerator();
            TSqlScript script = fragment as TSqlScript;
            if (script != null)
            {
                foreach (var batch in script.Batches)
                {
                    yield return ScriptFragment(sg, batch);
                }
            }
            else
            {
                // TSqlFragment is a TSqlBatch or a TSqlStatement
                yield return ScriptFragment(sg, fragment);
            }
        }
        private static string ScriptFragment(SqlScriptGenerator sg, TSqlFragment fragment)
        {
            string resultString;
            sg.GenerateScript(fragment, out resultString);
            return resultString;
        }
    }
}