- 增加 FreeSql.Provider.Dameng 基于 DmProvider Ado.net 访问达梦数据库；#155

Providers/FreeSql.Provider.Dameng/Curd/DamengInsert.cs

@@ -0,0 +1,222 @@
+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.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)
+        {
+        }
+
+        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)
+                        sb.Append(col.DbInsertValue);
+                    else
+                    {
+                        object val = col.GetMapValue(d);
+                        if (_noneParameter)
+                            sb.Append(_commonUtils.GetNoneParamaterSqlValue(specialParams, col.Attribute.MapType, val));
+                        else
+                        {
+                            sb.Append(_commonUtils.QuoteWriteParamter(col.Attribute.MapType, _commonUtils.QuoteParamterName($"{col.CsName}_{didx}")));
+                            _params[didx * colidx + colidx2] = _commonUtils.AppendParamter(null, $"{col.CsName}_{didx}", col, col.Attribute.MapType, val);
+                        }
+                    }
+                    ++colidx2;
+                }
+                sb.Append(")");
+                ++didx;
+            }
+            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, _params);
+                }
+                catch (Exception ex)
+                {
+                    exception = ex;
+                    throw ex;
+                }
+                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, dbParms);
+                long.TryParse(string.Concat(identParam.Value), out ret);
+            }
+            catch (Exception ex)
+            {
+                exception = ex;
+                throw ex;
+            }
+            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() => base.SplitExecuteAffrowsAsync(500, 999);
+        public override Task<long> ExecuteIdentityAsync() => base.SplitExecuteIdentityAsync(500, 999);
+        public override Task<List<T1>> ExecuteInsertedAsync() => base.SplitExecuteInsertedAsync(500, 999);
+
+        async protected override Task<long> RawExecuteIdentityAsync()
+        {
+            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, _params);
+                }
+                catch (Exception ex)
+                {
+                    exception = ex;
+                    throw ex;
+                }
+                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, dbParms);
+                long.TryParse(string.Concat(identParam.Value), out ret);
+            }
+            catch (Exception ex)
+            {
+                exception = ex;
+                throw ex;
+            }
+            finally
+            {
+                var after = new Aop.CurdAfterEventArgs(before, exception, ret);
+                _orm.Aop.CurdAfterHandler?.Invoke(this, after);
+            }
+            return ret;
+        }
+        async protected override Task<List<T1>> RawExecuteInsertedAsync()
+        {
+            var sql = this.ToSql();
+            if (string.IsNullOrEmpty(sql)) return new List<T1>();
+
+            var ret = _source.ToList();
+            await this.RawExecuteAffrowsAsync();
+            return ret;
+        }
+#endif
+    }
+}