Можно ли создать коллекцию анонимных типов из DataReader?

У меня есть этот код:

var query = "SELECT * FROM Cats";

var conn = new SqlConnection(sqlConnectionString);

conn.Open();

var cmd = new SqlCommand(query);
var reader = cmd.ExecuteReader();

while (reader.Read())
{
    var CatName = reader.GetString(0);
    var CatDOB = reader.GetDateTime(1);
    var CatStatus = reader.GetInt32(2);
}

Я хотел бы вытащить строки в анонимную коллекцию типов, что я обычно делал бы с помощью LINQ для итерации, но я не уверен, возможно ли это из-за способа вызова .Read() каждый раз, чтобы получить следующий ряд.

Есть ли способ сделать это?

4 ответа

Решение

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

private IEnumerable<T> Select<T>(DbDataReader reader, Func<DbDataReader, T> selector)
{
    while(reader.Read())
    {
        yield return selector(reader);
    }
}

использование:

var items = SelectFromReader(reader, r => new { CatName = r.GetString(0), CarDOB = r.GetDateTime(1), CatStatus = r.GetInt32(2) });

Вы даже можете сделать метод методом расширения на DbDataReader:

public static IEnumerable<T> Select<T>(this DbDataReader reader, Func<DbDataReader, T> selector)
{
    while (reader.Read())
    {
        yield return selector(reader);
    }
}

и использовать это так:

var items = reader.Select(r => new { CatName = r.GetString(0), CarDOB = r.GetDateTime(1), CatStatus = r.GetInt32(2) });

Вот пример того, как сделать это с помощью динамического (с которым, мне кажется, легче работать), но некоторые могут не согласиться с буквой вашего вопроса.

Назовите это так:

var result = SelectIntoList("SELECT * FROM Cats",sqlconnectionString);

Вы можете (как и я) поместить его в статический класс в отдельный файл для упрощения обслуживания.

public static IEnumerable<dynamic> SelectIntoList(string SQLselect, string connectionString, CommandType cType = CommandType.Text)
{
  using (SqlConnection conn = new SqlConnection(connectionString))
  {
    using (SqlCommand cmd = conn.CreateCommand())
    {
      cmd.CommandType = cType;
      cmd.CommandText = SQLselect;

      conn.Open();

      using (SqlDataReader reader = cmd.ExecuteReader())
      {

        if (reader.Read())  // read the first one to get the columns collection
        {
          var cols = reader.GetSchemaTable()
                       .Rows
                       .OfType<DataRow>()
                       .Select(r => r["ColumnName"]);

          do
          {
            dynamic t = new System.Dynamic.ExpandoObject();

            foreach (string col in cols)
            {
              ((IDictionary<System.String, System.Object>)t)[col] = reader[col];
            }

            yield return t;
          } while (reader.Read());
        }
      }

      conn.Close();
    }
  }
}

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

public static IEnumerable<T> Empty<T>(T dummyValue)
{
    return Enumerable.Empty<T>();
}

Это позволяет нам создать список анонимного типа:

var list = Empty(new
{
    CatName = "",
    CatDOB = DateTime.Today,
    CatStatus = 0
}).ToList();

(Элемент здесь не используется.)

Теперь мы можем добавить наши анонимные типы в этот список:

var cmd = new SqlCommand(query);
var reader = cmd.ExecuteReader();

while (reader.Read())
{
    list.Add(new
    {
        CatName = reader.GetString(0),
        CatDOB = reader.GetDateTime(1),
        CatStatus = reader.GetInt32(2),
    });
}

Конечно, использование именованного типа, вероятно, будет проще, поэтому я бы предложил использовать его, если только нет веской причины не делать этого. Это особенно верно, если вы планируете использовать список за пределами области, в которой он создан.

Технически, он может не отвечать на ваш вопрос, но просто не использовать читателя. Вместо этого используйте SqlDataAdapter Заполнить DataSet, если можете. Возьмите 0-ую таблицу этого DataSet и выберите новый анонимный объект из коллекции Rows.

using System.Data; // and project must reference System.Data.DataSetExtensions

var ds = new DataSet();
using (var conn = DbContext.Database.GetDbConnection())
using (var cmd = conn.CreateCommand())
{
  cmd.CommandType = CommandType.Text;
  cmd.CommandText = sqlText;
  conn.Open();
  (new SqlDataAdapter(cmd)).Fill(ds);
}
var rows = ds.Tables[0].AsEnumerable(); // AsEnumerable() is the extension
var anons = rows
  .Select(r => new { Val = r["Val"] })
  .ToList();
Другие вопросы по тегам