Dapper BulkInsert 實作紀錄

Dapper, BulkInsert, bulk insert, SqlBulkCopy

參考

Dapper.NET Tutorial => Bulk Copy

Creating a Generic List DataReader for SqlBulkCopy

原始碼紀錄

DBHelper.cs

public static class DBHelperClassExtensions
{
  /// <summary>
  /// BulkInsert/BulkCopy
  /// ref → https://riptutorial.com/dapper/example/21711/bulk-copy
  /// </summary>
  public static void BulkInsert<TTable>(this SqlConnection conn, IEnumerable<TTable> entityList, int bulkCopyTimeout = 30, int batchSize = 500)
  {
    var tableType = typeof(TTable);

    using (IDataReader reader = entityList.GetDataReader())
    using (var bulkCopy = new SqlBulkCopy(conn))
    {
      bulkCopy.BulkCopyTimeout = bulkCopyTimeout;
      bulkCopy.BatchSize = batchSize;
      bulkCopy.DestinationTableName = tableType.Name;
      bulkCopy.EnableStreaming = true;
      using (var dataReader = reader)
      {
        bulkCopy.WriteToServer(dataReader);
      }
    }
  }
}

GenericListDataReader.cs

using System;
using System.Collections.Generic;
using System.Data;
using System.Reflection;

/***************************************************************************
參考引用自：https://www.csvreader.com/posts/generic_list_datareader.php
***************************************************************************/

namespace YourProject.Models;

public static class GenericListDataReaderExtensions
{
  public static GenericListDataReader<T> GetDataReader<T>(this IEnumerable<T> list)
  {
    return new GenericListDataReader<T>(list);
  }
}

public class GenericListDataReader<T> : IDataReader
{
  private IEnumerator<T> list = null;
  private List<PropertyInfo> properties = new List<PropertyInfo>();
  private Dictionary<string, int> nameLookup = new Dictionary<string, int>();

  public GenericListDataReader(IEnumerable<T> list)
  {
    this.list = list.GetEnumerator();

    properties.AddRange(
      typeof(T)
      .GetProperties(
        BindingFlags.GetProperty |
        BindingFlags.Instance |
        BindingFlags.Public |
        BindingFlags.DeclaredOnly
        ));

    for (int i = 0; i < properties.Count; i++)
    {
      nameLookup[properties[i].Name] = i;
    }
  }

  #region IDataReader Members

  public void Close()
  {
    list.Dispose();
  }

  public int Depth
  {
    get { throw new NotImplementedException(); }
  }

  public DataTable GetSchemaTable()
  {
    throw new NotImplementedException();
  }

  public bool IsClosed
  {
    get { throw new NotImplementedException(); }
  }

  public bool NextResult()
  {
    throw new NotImplementedException();
  }

  public bool Read()
  {
    return list.MoveNext();
  }

  public int RecordsAffected
  {
    get { throw new NotImplementedException(); }
  }

  #endregion

  #region IDisposable Members

  public void Dispose()
  {
    Close();
  }

  #endregion

  #region IDataRecord Members

  public int FieldCount
  {
    get { return properties.Count; }
  }

  public bool GetBoolean(int i)
  {
    return (bool)GetValue(i);
  }

  public byte GetByte(int i)
  {
    return (byte)GetValue(i);
  }

  public long GetBytes(int i, long fieldOffset, byte[] buffer, int bufferoffset, int length)
  {
    throw new NotImplementedException();
  }

  public char GetChar(int i)
  {
    return (char)GetValue(i);
  }

  public long GetChars(int i, long fieldoffset, char[] buffer, int bufferoffset, int length)
  {
    throw new NotImplementedException();
  }

  public IDataReader GetData(int i)
  {
    throw new NotImplementedException();
  }

  public string GetDataTypeName(int i)
  {
    throw new NotImplementedException();
  }

  public DateTime GetDateTime(int i)
  {
    return (DateTime)GetValue(i);
  }

  public decimal GetDecimal(int i)
  {
    return (decimal)GetValue(i);
  }

  public double GetDouble(int i)
  {
    return (double)GetValue(i);
  }

  public Type GetFieldType(int i)
  {
    return properties[i].PropertyType;
  }

  public float GetFloat(int i)
  {
    return (float)GetValue(i);
  }

  public Guid GetGuid(int i)
  {
    return (Guid)GetValue(i);
  }

  public short GetInt16(int i)
  {
    return (short)GetValue(i);
  }

  public int GetInt32(int i)
  {
    return (int)GetValue(i);
  }

  public long GetInt64(int i)
  {
    return (long)GetValue(i);
  }

  public string GetName(int i)
  {
    return properties[i].Name;
  }

  public int GetOrdinal(string name)
  {
    if (nameLookup.ContainsKey(name))
    {
      return nameLookup[name];
    }
    else
    {
      return -1;
    }
  }

  public string GetString(int i)
  {
    return (string)GetValue(i);
  }

  public object GetValue(int i)
  {
    return properties[i].GetValue(list.Current, null);
  }

  public int GetValues(object[] values)
  {
    int getValues = Math.Max(FieldCount, values.Length);

    for (int i = 0; i < getValues; i++)
    {
      values[i] = GetValue(i);
    }

    return getValues;
  }

  public bool IsDBNull(int i)
  {
    return GetValue(i) == null;
  }

  public object this[string name]
  {
    get
    {
      return GetValue(GetOrdinal(name));
    }
  }

  public object this[int i]
  {
    get
    {
      return GetValue(i);
    }
  }

  #endregion
}