Как получить идентификаторы вставленных записей данных, используя массовую копию SQL - PullRequest
Новая
       19

Как получить идентификаторы вставленных записей данных, используя массовую копию SQL

9 голосов
/ 24 марта 2010

У меня ADO.NET DataTable с примерно 100 000 записей. В этой таблице есть столбец xyID, в котором нет значений, поскольку этот столбец автоматически генерируется IDENTITY в моей базе данных SQL Server.

Мне нужно получить сгенерированные идентификаторы для других процессов. Я ищу способ массового копирования этого DataTable в базу данных SQL Server и в рамках того же «шага», чтобы «заполнить» мой DataTable созданными идентификаторами.

Как я могу получить значения идентичности записей, вставленных в таблицу, используя класс SqlBulkCopy?

1 Ответ

2 голосов
/ 11 июня 2010

В настоящее время я делаю что-то вроде этого: 

DataTable objects = new DataTable();
DataColumn keyColumn = new DataColumn("name", typeof(string));
DataColumn versionColumn = new DataColumn("version", typeof(int));
versionColumn.DefaultValue = iVersionID;

objects.Columns.Add(keyColumn);
objects.Columns.Add(versionColumn);

foreach (KeyValuePair<string, NamedObject> kvp in Directory)
{
    NamedObject o = kvp.Value;
    DataRow row = objects.NewRow();
    row[0] = o.Name;
    objects.Rows.Add(row);
}

using (SqlBulkCopy updater = new SqlBulkCopy(conn,
        SqlBulkCopyOptions.TableLock | SqlBulkCopyOptions.UseInternalTransaction, null))
{
    updater.DestinationTableName = "object_table";
    updater.WriteToServer(objects);
}

string sQuery = @"SELECT id, name FROM object_table WHERE version = @ver";
using (SqlCommand command = new SqlCommand(sQuery, conn))
{
    SqlParameter version = new SqlParameter("@ver", SqlDbType.Int, 4);
    version.Value = versionID;
    command.Parameters.Add(version);

    command.CommandTimeout = 600;

    using (SqlDataReader reader = command.ExecuteReader())
    {
        while (reader.Read())
        {
            string key = (string)reader[1];

            NamedObject item = Directory[key];
            item.ID = (int)reader[0];
        }
    }
}

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

Однако мой запрос на выборку в настоящее время истекает в ExecuteReader, даже с этим 10-минутным окном. Так что это не наше окончательное решение.

...