Получить количество результирующих наборов из SqlDataReader

Question

У меня есть хранимая процедура SQL Server, которая возвращает несколько результатов. Тело хранимой процедуры может выглядеть так:

SELECT * FROM tableA; 
SELECT * FROM tableB; 
SELECT * FROM tableC; 

В этом случае хранимая процедура возвращает 3 набора результатов. Другие хранимые процедуры могут возвращаться, например, 1, 0 или любое количество наборов результатов. Каждый набор результатов может содержать 0 или более строк. При загрузке этих данных мне нужно будет позвонить IDataReader.NextResult() для перемещения между наборами результатов.

Как я могу надежно получить количество наборов результатов (а не число строк) в C#?

Answer 1

Использование DataReader.NextResult для продвижения читателя к следующему результату набора .:

using (var con = new SqlConnection(Properties.Settings.Default.ConnectionString)) 
{ 
    using (var cmd = new SqlCommand("SELECT * FROM TableA; SELECT * FROM TableB; SELECT * FROM TableC;", con)) 
    { 
     con.Open(); 
     using (IDataReader rdr = cmd.ExecuteReader()) 
     { 
      while (rdr.Read()) 
      { 
       int firstIntCol = rdr.GetInt32(0); // assuming the first column is of type Int32 
       // other fields ... 
      } 
      if (rdr.NextResult()) 
      { 
       while (rdr.Read()) 
       { 
        int firstIntCol = rdr.GetInt32(0); // assuming the first column is of type Int32 
        // other fields ... 
       } 
       if (rdr.NextResult()) 
       { 
        while (rdr.Read()) 
        { 
         int firstIntCol = rdr.GetInt32(0); // assuming the first column is of type Int32 
         // other fields ... 
        } 
       } 
      } 
     } 
    } 
} 

Answer 2

Там, кажется, нет никакой собственности или метод, который непосредственно вычисляет количество результатов в IDataReader. Этот интерфейс скорее намерен потребляться инкрементным/потоковым способом. Таким образом, чтобы подсчитать количество возвращаемых наборов результатов, увеличивайте счетчик каждый раз, когда вы вызываете IDataReader.NextResult(), и он возвращает true при потреблении данных.

Однако есть улов. документация для IDataReader.NextResult() состояний:

По умолчанию считыватель данных устанавливается на первый результат.

Рассмотрим следующие сценарии:

• Команда возвращается 0 наборов результатов. Ваш первый звонок IDataReader.NextResult() возвращает false.
• Команда вернула 1 результирующий набор. Ваш первый звонок IDataReader.NextResult() возвращает false.
• Команда вернула 2 набора результатов. Ваш второй звонок IDataReader.NextResult() возвращает false.

Вы можете видеть, что у нас достаточно информации для подсчета количества наборов результатов, пока существует хотя бы один результирующий набор. Это будет количество раз, которое IDataReader.NextResult() возвращено true плюс один.

Для определения наличия или отсутствия 0 наборов результатов мы используем другое свойство от читателя: IDataRecord.FieldCount. В документации для этого свойства указано:

Если не установлено в действительном наборе записей, 0; в противном случае - количество столбцов в текущей записи. Значение по умолчанию - -1.

Таким образом, мы можем прочитать это поле при первом открытии считывателя, чтобы определить, находимся ли мы в допустимом наборе результатов или нет. Если команда не генерирует результирующих наборов, значение IDataRecord.FieldCount на считывающем устройстве первоначально будет меньше 1. Если команда генерирует хотя бы один результирующий набор, значение первоначально будет положительным. Это предполагает, что для набора результатов невозможно иметь 0 столбцов (что, я думаю, можно предположить с помощью SQL, не уверен).

Итак, для подсчета количества наборов результатов я бы использовал следующее. Если вам также необходимо сохранить данные, что логика должна быть вставлена ​​в это:

using (var reader = command.ExecuteReader()) 
{ 
    var resultCount = 0; 
    do 
    { 
     if (reader.FieldCount > 0) 
      resultCount++; 

     while (reader.Read()) 
     { 
      // Insert logic to actually consume data here… 
      // HandleRecordByResultIndex(resultCount - 1, (IDataRecord)reader); 
     } 
    } while (reader.NextResult()); 
} 

Я проверил это с System.Data.SqlClient и команды PRINT 'hi' (0 результирующие наборы), SELECT 1 x WHERE 1=0 (1 набор результатов), и SELECT 1 x WHERE 1=0; SELECT 1 x WHERE 1=0 (2 набора результатов).