SqlCommand () ExecuteNonQuery () усекает текст команды

Я создаю пользовательскую утилиту развертывания БД, мне нужно прочитать текстовые файлы, содержащие сценарии sql, и выполнить их против базы данных.

довольно простой материал, до сих пор так хорошо.

однако я столкнулся с проблемой, содержимое файла читается успешно и полностью, но после передачи в SqlCommand, а затем выполняется с SqlCommand.ExecuteNonQuery выполняется только часть скрипта.

Я запустил Профайлер и подтвердил, что мой код не передавая весь сценарий.

    private void ExecuteScript(string cmd, SqlConnection sqlConn, SqlTransaction trans)
    {

        SqlCommand sqlCmd = new SqlCommand(cmd, sqlConn, trans);
        sqlCmd.CommandType = CommandType.Text;
        sqlCmd.CommandTimeout = 9000000; // for testing
        sqlCmd.ExecuteNonQuery();

    }

    // I call it like this, readDMLScript contains 543 lines of T-SQL
    string readDMLScript = ReadFile(dmlFile);
    ExecuteScript(readDMLScript, sqlConn, trans);

6 ответов


да, каждый попадает в эту загвоздку в первый раз, когда они начинают отправлять содержимое файлов сценариев SQL в базу данных.

GO не является командой T-SQL. Это маркер конца пакета, распознаваемый всеми инструментами Microsoft interactive SQL (Management Studio, isql, osql). Чтобы справиться с этим, вам нужно будет написать свой собственный парсер, чтобы вырвать каждый блок текста в файле между GO операторы и передать их в базу данных в виде отдельных команд.

как вы реализуете свой парсер до вас. Это может быть просто (читайте в каждой строке за раз, обнаруживайте строки, которые состоят только из GO и пробела) или комплекс (tokenising все заявления и Ли GO - Это подлинный оператор или бит текста внутри строки или многострочного комментария).

лично я пошел с первым вариантом. Он обрабатывает 99% всех файлов SQL, с которыми вы когда-либо сталкивались без суеты. Если вы хотите пойти до конца и написать токенизатор, я уверен, что многие люди уже сделали это, просто Google для этого.

пример:

using(var reader = new SqlBatchReader(new StreamReader(dmlFile))) {
    string batch;
    while((batch = reader.ReadBatch()) != null) {
        var cmd = new SqlCommand(batch, conn, trans) { CommandType = CommandType.Text };
        cmd.ExecuteNonQuery();
    }
}

class SqlBatchReader : IDisposable {
    private TextReader _reader;
    public SqlBatchReader(TextReader reader) {
        _reader = reader;
    }
    /// <summary>
    /// Return the next command batch in the file, or null if end-of-file reached.
    /// </summary>
    public string ReadBatch() {
        // TODO: Implement your parsing logic here.
    }
}

Я нашел код ниже при поиске ответа на этот вопрос:

http://blogs.msdn.com/b/onoj/archive/2008/02/26/incorrect-syntax-near-go-sqlcommand-executenonquery.aspx

плюсы: это короткое и простое понимание и отлично работает для моих нужд.

минусы: он менее эффективен, чем потоковые решения, и чувствителен к регистру (i.e " GO " не "go").

string[] commands = sql.Split(new string[]{"GO\r\n", "GO ", "GO\t"}, StringSplitOptions.RemoveEmptyEntries );
foreach (string c in commands)
{
    var command = new SqlCommand(c, masterConnection);
    command.ExecuteNonQuery();
}


ответ на основе комментариев под оригинальным сообщением:

GO является маркером для Management Studio / osql / isql. Он сообщает, чтобы отправить пакет команд на SQL Server. В вашей утилите вы должны разделить входные данные с помощью GO в качестве разделителя и отправить каждый элемент отдельно (без команды GO)


Это то, что мы используем:)

public static class ExtensionMethodsSqlCommand
{
    #region Public

    private static bool IsGo(string psCommandLine)
    {
        if (psCommandLine == null)
            return false;
        psCommandLine = psCommandLine.Trim();
        if (string.Compare(psCommandLine, "GO", StringComparison.OrdinalIgnoreCase) == 0)
            return true;
        if (psCommandLine.StartsWith("GO", StringComparison.OrdinalIgnoreCase))
        {
            psCommandLine = (psCommandLine + "--").Substring(2).Trim();
            if (psCommandLine.StartsWith("--"))
                return true;
        }
        return false;
    }

    [System.Diagnostics.DebuggerHidden]
    public static void ExecuteNonQueryWithGos(this SqlCommand poSqlCommand)
    {
        string sCommandLong = poSqlCommand.CommandText;
        using (StringReader oStringReader = new StringReader(sCommandLong))
        {
            string sCommandLine;
            string sCommandShort = string.Empty;
            while ((sCommandLine = oStringReader.ReadLine()) != null)
                if (ExtensionMethodsSqlCommand.IsGo(sCommandLine))
                {
                    if (sCommandShort.IsNullOrWhiteSpace() == false)
                    {
                        if ((poSqlCommand.Connection.State & ConnectionState.Open) == 0)
                            poSqlCommand.Connection.Open();
                        using (SqlCommand oSqlCommand = new SqlCommand(sCommandShort, poSqlCommand.Connection))
                            oSqlCommand.ExecuteNonQuery();
                    }
                    sCommandShort = string.Empty;
                }
                else
                    sCommandShort += sCommandLine + "\r\n";
        }
    }

    #endregion Public
}

Я закончила реализацию StringReader для этого.

ручки:

  1. пропуская мимо GO содержится в тире тире комментарии
  2. пропуская мимо GO содержится в Слэш star комментарии
  3. пропуск мимо GO содержится в литералах (i.e одинарные кавычки)
  4. пропуская мимо пройти, содержащиеся в имена столбцов и т. д.

поэтому он будет обнаруживать ключевое слово GO только при использовании в качестве разделителя пакетов. Этот означает, что он разбивает текст SQL правильно.

Он также обрабатывает, если вы добавили SQL-Терминатор (точку с запятой) к слову GO

вы можете найти код для него здесь:

вы используете его так:

using (var reader = new SqlCommandReader(scriptContents))
       {
            var commands = new List<string>();
            reader.ReadAllCommands(c => commands.Add(c));
            // commands now contains each seperated sql batch.
        }