mirror of
https://github.com/nsnail/FreeSql.git
synced 2025-04-19 01:12:49 +08:00
229 lines
10 KiB
C#
229 lines
10 KiB
C#
using FreeSql.Internal;
|
|
using FreeSql.Internal.Model;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Data;
|
|
using System.Data.Common;
|
|
using System.Linq;
|
|
using System.Text;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
|
|
namespace FreeSql.Dameng.Curd
|
|
{
|
|
|
|
class DamengInsert<T1> : Internal.CommonProvider.InsertProvider<T1> where T1 : class
|
|
{
|
|
public DamengInsert(IFreeSql orm, CommonUtils commonUtils, CommonExpression commonExpression)
|
|
: base(orm, commonUtils, commonExpression)
|
|
{
|
|
}
|
|
|
|
internal IFreeSql InternalOrm => _orm as IFreeSql;
|
|
internal DbConnection InternalConnection => _connection;
|
|
internal DbTransaction InternalTransaction => _transaction;
|
|
|
|
public override int ExecuteAffrows() => base.SplitExecuteAffrows(_batchValuesLimit > 0 ? _batchValuesLimit : 500, _batchParameterLimit > 0 ? _batchParameterLimit : 999);
|
|
public override long ExecuteIdentity() => base.SplitExecuteIdentity(_batchValuesLimit > 0 ? _batchValuesLimit : 500, _batchParameterLimit > 0 ? _batchParameterLimit : 999);
|
|
public override List<T1> ExecuteInserted() => base.SplitExecuteInserted(_batchValuesLimit > 0 ? _batchValuesLimit : 500, _batchParameterLimit > 0 ? _batchParameterLimit : 999);
|
|
|
|
public override string ToSql()
|
|
{
|
|
if (_source == null || _source.Any() == false) return null;
|
|
var sb = new StringBuilder();
|
|
sb.Append("INSERT ");
|
|
if (_source.Count > 1) sb.Append("ALL");
|
|
|
|
_identCol = null;
|
|
var sbtb = new StringBuilder();
|
|
sbtb.Append("INTO ");
|
|
sbtb.Append(_commonUtils.QuoteSqlName(TableRuleInvoke())).Append("(");
|
|
var colidx = 0;
|
|
foreach (var col in _table.Columns.Values)
|
|
{
|
|
if (col.Attribute.IsIdentity) _identCol = col;
|
|
if (col.Attribute.IsIdentity && _insertIdentity == false && string.IsNullOrEmpty(col.DbInsertValue)) continue;
|
|
if (col.Attribute.IsIdentity == false && _ignore.ContainsKey(col.Attribute.Name)) continue;
|
|
|
|
if (colidx > 0) sbtb.Append(", ");
|
|
sbtb.Append(_commonUtils.QuoteSqlName(col.Attribute.Name));
|
|
++colidx;
|
|
}
|
|
sbtb.Append(") ");
|
|
|
|
_params = _noneParameter ? new DbParameter[0] : new DbParameter[colidx * _source.Count];
|
|
var specialParams = new List<DbParameter>();
|
|
var didx = 0;
|
|
foreach (var d in _source)
|
|
{
|
|
if (_source.Count > 1) sb.Append("\r\n");
|
|
sb.Append(sbtb);
|
|
sb.Append("VALUES");
|
|
sb.Append("(");
|
|
var colidx2 = 0;
|
|
foreach (var col in _table.Columns.Values)
|
|
{
|
|
if (col.Attribute.IsIdentity && _insertIdentity == false && string.IsNullOrEmpty(col.DbInsertValue)) continue;
|
|
if (col.Attribute.IsIdentity == false && _ignore.ContainsKey(col.Attribute.Name)) continue;
|
|
|
|
if (colidx2 > 0) sb.Append(", ");
|
|
if (string.IsNullOrEmpty(col.DbInsertValue) == false && _ignoreInsertValueSql.ContainsKey(col.Attribute.Name) == false)
|
|
sb.Append(col.DbInsertValue);
|
|
else
|
|
{
|
|
object val = col.GetDbValue(d);
|
|
if (val == null && col.Attribute.IsNullable == false) val = col.CsType == typeof(string) ? "" : Utils.GetDataReaderValue(col.CsType.NullableTypeOrThis(), null);//#384
|
|
|
|
var colsql = _noneParameter ? _commonUtils.GetNoneParamaterSqlValue(specialParams, _noneParameterFlag, col, col.Attribute.MapType, val) :
|
|
_commonUtils.QuoteWriteParamterAdapter(col.Attribute.MapType, _commonUtils.QuoteParamterName($"{col.CsName}_{didx}"));
|
|
sb.Append(_commonUtils.RewriteColumn(col, colsql));
|
|
if (_noneParameter == false)
|
|
_params[didx * colidx + colidx2] = _commonUtils.AppendParamter(null, $"{col.CsName}_{didx}", col, col.Attribute.MapType, val);
|
|
}
|
|
++colidx2;
|
|
}
|
|
sb.Append(")");
|
|
++didx;
|
|
}
|
|
if (_noneParameter && specialParams.Any()) _params = specialParams.ToArray();
|
|
if (_source.Count > 1) sb.Append("\r\n SELECT 1 FROM DUAL");
|
|
return sb.ToString();
|
|
}
|
|
|
|
ColumnInfo _identCol;
|
|
protected override long RawExecuteIdentity()
|
|
{
|
|
var sql = this.ToSql();
|
|
if (string.IsNullOrEmpty(sql)) return 0;
|
|
|
|
long ret = 0;
|
|
Exception exception = null;
|
|
Aop.CurdBeforeEventArgs before = null;
|
|
|
|
if (_identCol == null || _source.Count > 1)
|
|
{
|
|
before = new Aop.CurdBeforeEventArgs(_table.Type, _table, Aop.CurdType.Insert, sql, _params);
|
|
_orm.Aop.CurdBeforeHandler?.Invoke(this, before);
|
|
try
|
|
{
|
|
ret = _orm.Ado.ExecuteNonQuery(_connection, _transaction, CommandType.Text, sql, _commandTimeout, _params);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
exception = ex;
|
|
throw;
|
|
}
|
|
finally
|
|
{
|
|
var after = new Aop.CurdAfterEventArgs(before, exception, ret);
|
|
_orm.Aop.CurdAfterHandler?.Invoke(this, after);
|
|
}
|
|
return 0;
|
|
}
|
|
var identColName = _commonUtils.QuoteSqlName(_identCol.Attribute.Name);
|
|
var identParam = _commonUtils.AppendParamter(null, $"{_identCol.CsName}99", _identCol, _identCol.Attribute.MapType, 0);
|
|
identParam.Direction = ParameterDirection.Output;
|
|
sql = $"{sql} RETURNING {identColName} INTO {identParam.ParameterName}";
|
|
var dbParms = _params.Concat(new[] { identParam }).ToArray();
|
|
before = new Aop.CurdBeforeEventArgs(_table.Type, _table, Aop.CurdType.Insert, sql, dbParms);
|
|
_orm.Aop.CurdBeforeHandler?.Invoke(this, before);
|
|
try
|
|
{
|
|
_orm.Ado.ExecuteNonQuery(_connection, _transaction, CommandType.Text, sql, _commandTimeout, dbParms);
|
|
long.TryParse(string.Concat(identParam.Value), out ret);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
exception = ex;
|
|
throw;
|
|
}
|
|
finally
|
|
{
|
|
var after = new Aop.CurdAfterEventArgs(before, exception, ret);
|
|
_orm.Aop.CurdAfterHandler?.Invoke(this, after);
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
protected override List<T1> RawExecuteInserted()
|
|
{
|
|
var sql = this.ToSql();
|
|
if (string.IsNullOrEmpty(sql)) return new List<T1>();
|
|
|
|
var ret = _source.ToList();
|
|
this.RawExecuteAffrows();
|
|
return ret;
|
|
}
|
|
|
|
#if net40
|
|
#else
|
|
public override Task<int> ExecuteAffrowsAsync(CancellationToken cancellationToken = default) => base.SplitExecuteAffrowsAsync(_batchValuesLimit > 0 ? _batchValuesLimit : 500, _batchParameterLimit > 0 ? _batchParameterLimit : 999, cancellationToken);
|
|
public override Task<long> ExecuteIdentityAsync(CancellationToken cancellationToken = default) => base.SplitExecuteIdentityAsync(_batchValuesLimit > 0 ? _batchValuesLimit : 500, _batchParameterLimit > 0 ? _batchParameterLimit : 999, cancellationToken);
|
|
public override Task<List<T1>> ExecuteInsertedAsync(CancellationToken cancellationToken = default) => base.SplitExecuteInsertedAsync(_batchValuesLimit > 0 ? _batchValuesLimit : 500, _batchParameterLimit > 0 ? _batchParameterLimit : 999, cancellationToken);
|
|
|
|
async protected override Task<long> RawExecuteIdentityAsync(CancellationToken cancellationToken = default)
|
|
{
|
|
var sql = this.ToSql();
|
|
if (string.IsNullOrEmpty(sql)) return 0;
|
|
|
|
long ret = 0;
|
|
Exception exception = null;
|
|
Aop.CurdBeforeEventArgs before = null;
|
|
|
|
if (_identCol == null || _source.Count > 1)
|
|
{
|
|
before = new Aop.CurdBeforeEventArgs(_table.Type, _table, Aop.CurdType.Insert, sql, _params);
|
|
_orm.Aop.CurdBeforeHandler?.Invoke(this, before);
|
|
try
|
|
{
|
|
ret = await _orm.Ado.ExecuteNonQueryAsync(_connection, _transaction, CommandType.Text, sql, _commandTimeout, _params, cancellationToken);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
exception = ex;
|
|
throw;
|
|
}
|
|
finally
|
|
{
|
|
var after = new Aop.CurdAfterEventArgs(before, exception, ret);
|
|
_orm.Aop.CurdAfterHandler?.Invoke(this, after);
|
|
}
|
|
return 0;
|
|
}
|
|
var identColName = _commonUtils.QuoteSqlName(_identCol.Attribute.Name);
|
|
var identParam = _commonUtils.AppendParamter(null, $"{_identCol.CsName}99", _identCol, _identCol.Attribute.MapType, 0);
|
|
identParam.Direction = ParameterDirection.Output;
|
|
sql = $"{sql} RETURNING {identColName} INTO {identParam.ParameterName}";
|
|
var dbParms = _params.Concat(new[] { identParam }).ToArray();
|
|
before = new Aop.CurdBeforeEventArgs(_table.Type, _table, Aop.CurdType.Insert, sql, dbParms);
|
|
_orm.Aop.CurdBeforeHandler?.Invoke(this, before);
|
|
try
|
|
{
|
|
await _orm.Ado.ExecuteNonQueryAsync(_connection, _transaction, CommandType.Text, sql, _commandTimeout, dbParms, cancellationToken);
|
|
long.TryParse(string.Concat(identParam.Value), out ret);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
exception = ex;
|
|
throw;
|
|
}
|
|
finally
|
|
{
|
|
var after = new Aop.CurdAfterEventArgs(before, exception, ret);
|
|
_orm.Aop.CurdAfterHandler?.Invoke(this, after);
|
|
}
|
|
return ret;
|
|
}
|
|
async protected override Task<List<T1>> RawExecuteInsertedAsync(CancellationToken cancellationToken = default)
|
|
{
|
|
var sql = this.ToSql();
|
|
if (string.IsNullOrEmpty(sql)) return new List<T1>();
|
|
|
|
var ret = _source.ToList();
|
|
await this.RawExecuteAffrowsAsync(cancellationToken);
|
|
return ret;
|
|
}
|
|
#endif
|
|
}
|
|
}
|