mirror of
https://github.com/nsnail/FreeSql.git
synced 2025-06-18 20:08:15 +08:00
- 增加 Oracle/达梦 BulkCopy 支持;
This commit is contained in:
@ -1,4 +1,8 @@
|
||||
public static partial class FreeSqlOracleGlobalExtensions
|
||||
using FreeSql;
|
||||
using Oracle.ManagedDataAccess.Client;
|
||||
using System;
|
||||
|
||||
public static partial class FreeSqlOracleGlobalExtensions
|
||||
{
|
||||
|
||||
/// <summary>
|
||||
@ -9,4 +13,93 @@
|
||||
/// <returns></returns>
|
||||
public static string FormatOracle(this string that, params object[] args) => _oracleAdo.Addslashes(that, args);
|
||||
static FreeSql.Oracle.OracleAdo _oracleAdo = new FreeSql.Oracle.OracleAdo();
|
||||
|
||||
#region ExecuteOracleBulkCopy
|
||||
/// <summary>
|
||||
/// Oracle CopyBulk 批量插入功能<para></para>
|
||||
/// 使用 IgnoreColumns/InsertColumns 设置忽略/指定导入的列<para></para>
|
||||
/// 使用 WithConnection/WithTransaction 传入连接/事务对象<para></para>
|
||||
/// 提示:若本方法不能满足,请使用 IInsert<T>.ToDataTable 方法得到 DataTable 对象后,自行处理。
|
||||
/// </summary>
|
||||
/// <typeparam name="T"></typeparam>
|
||||
/// <param name="that"></param>
|
||||
/// <param name="copyOptions"></param>
|
||||
/// <param name="batchSize"></param>
|
||||
/// <param name="bulkCopyTimeout"></param>
|
||||
public static void ExecuteOracleBulkCopy<T>(this IInsert<T> that, OracleBulkCopyOptions copyOptions = OracleBulkCopyOptions.Default, int? batchSize = null, int? bulkCopyTimeout = null) where T : class
|
||||
{
|
||||
var insert = that as FreeSql.Oracle.Curd.OracleInsert<T>;
|
||||
if (insert == null) throw new Exception("ExecuteOracleBulkCopy 是 FreeSql.Provider.Oracle 特有的功能");
|
||||
|
||||
var dt = that.ToDataTable();
|
||||
if (dt.Rows.Count == 0) return;
|
||||
|
||||
Action<OracleBulkCopy> writeToServer = bulkCopy =>
|
||||
{
|
||||
if (batchSize.HasValue) bulkCopy.BatchSize = batchSize.Value;
|
||||
if (bulkCopyTimeout.HasValue) bulkCopy.BulkCopyTimeout = bulkCopyTimeout.Value;
|
||||
bulkCopy.DestinationTableName = dt.TableName;
|
||||
for (int i = 0; i < dt.Columns.Count; i++)
|
||||
bulkCopy.ColumnMappings.Add(dt.Columns[i].ColumnName, dt.Columns[i].ColumnName);
|
||||
bulkCopy.WriteToServer(dt);
|
||||
};
|
||||
|
||||
try
|
||||
{
|
||||
if (insert.InternalConnection == null && insert.InternalTransaction == null)
|
||||
{
|
||||
using (var conn = insert.InternalOrm.Ado.MasterPool.Get())
|
||||
{
|
||||
using (var bulkCopy = copyOptions == OracleBulkCopyOptions.Default ?
|
||||
new OracleBulkCopy(conn.Value as OracleConnection) :
|
||||
new OracleBulkCopy(conn.Value as OracleConnection, copyOptions))
|
||||
{
|
||||
writeToServer(bulkCopy);
|
||||
}
|
||||
}
|
||||
}
|
||||
else if (insert.InternalTransaction != null)
|
||||
{
|
||||
using (var bulkCopy = copyOptions == OracleBulkCopyOptions.Default ?
|
||||
new OracleBulkCopy(insert.InternalTransaction.Connection as OracleConnection) :
|
||||
new OracleBulkCopy(insert.InternalTransaction.Connection as OracleConnection, copyOptions))
|
||||
{
|
||||
writeToServer(bulkCopy);
|
||||
}
|
||||
}
|
||||
else if (insert.InternalConnection != null)
|
||||
{
|
||||
var conn = insert.InternalConnection as OracleConnection;
|
||||
var isNotOpen = false;
|
||||
if (conn.State != System.Data.ConnectionState.Open)
|
||||
{
|
||||
isNotOpen = true;
|
||||
conn.Open();
|
||||
}
|
||||
try
|
||||
{
|
||||
using (var bulkCopy = copyOptions == OracleBulkCopyOptions.Default ?
|
||||
new OracleBulkCopy(conn) :
|
||||
new OracleBulkCopy(conn, copyOptions))
|
||||
{
|
||||
writeToServer(bulkCopy);
|
||||
}
|
||||
}
|
||||
finally
|
||||
{
|
||||
if (isNotOpen)
|
||||
conn.Close();
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
throw new NotImplementedException("ExecuteOracleBulkCopy 未实现错误,请反馈给作者");
|
||||
}
|
||||
}
|
||||
finally
|
||||
{
|
||||
dt.Clear();
|
||||
}
|
||||
}
|
||||
#endregion
|
||||
}
|
||||
|
Reference in New Issue
Block a user