123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108 |
- using System;
- using System.Linq;
- using System.Text;
- namespace SqlSugar
- {
- public class KdbndpInsertBuilder : InsertBuilder
- {
- public override string SqlTemplate
- {
- get
- {
- if (IsReturnIdentity)
- {
- return @"INSERT INTO {0}
- ({1})
- VALUES
- ({2}) returning $PrimaryKey";
- }
- else
- {
- return @"INSERT INTO {0}
- ({1})
- VALUES
- ({2}) ;";
- }
- }
- }
- public override string SqlTemplateBatch => "INSERT INTO {0} ({1})";
- public override string SqlTemplateBatchUnion => " VALUES ";
- public override string SqlTemplateBatchSelect => " {0} ";
- public override string ToSqlString()
- {
- if (IsNoInsertNull)
- {
- DbColumnInfoList = DbColumnInfoList.Where(it => it.Value != null).ToList();
- }
- var groupList = DbColumnInfoList.GroupBy(it => it.TableId).ToList();
- var isSingle = groupList.Count() == 1;
- string columnsString = string.Join(",", groupList.First().Select(it => Builder.GetTranslationColumnName(it.DbColumnName)));
- if (isSingle)
- {
- string columnParametersString = string.Join(",", this.DbColumnInfoList.Select(it =>base.GetDbColumn(it, Builder.SqlParameterKeyWord + it.DbColumnName)));
- return string.Format(SqlTemplate, GetTableNameString, columnsString, columnParametersString);
- }
- else
- {
- StringBuilder batchInsetrSql = new StringBuilder();
- int pageSize = 200;
- int pageIndex = 1;
- int totalRecord = groupList.Count;
- int pageCount = (totalRecord + pageSize - 1) / pageSize;
- while (pageCount >= pageIndex)
- {
- batchInsetrSql.AppendFormat(SqlTemplateBatch, GetTableNameString, columnsString);
- int i = 0;
- foreach (var columns in groupList.Skip((pageIndex - 1) * pageSize).Take(pageSize).ToList())
- {
- var isFirst = i == 0;
- if (isFirst)
- {
- batchInsetrSql.Append(SqlTemplateBatchUnion);
- }
- batchInsetrSql.Append("\r\n ( " + string.Join(",", columns.Select(it =>
- {
- if (it.InsertServerTime || it.InsertSql.HasValue() || it.SqlParameterDbType is Type || it?.PropertyType?.Name == "DateOnly" || it?.PropertyType?.Name == "TimeOnly")
- {
- return GetDbColumn(it, null);
- }
- object value = null;
- if (it.Value is DateTime)
- {
- value = ((DateTime)it.Value).ToString("O");
- }
- else if (it.Value is DateTimeOffset)
- {
- return FormatDateTimeOffset(it.Value);
- }
- else
- {
- value = it.Value;
- }
- if (value == null||value==DBNull.Value)
- {
- return string.Format(SqlTemplateBatchSelect, "NULL");
- }
- return string.Format(SqlTemplateBatchSelect, "'" + value.ObjToString().ToSqlFilter() + "'");
- })) + "),");
- ++i;
- }
- pageIndex++;
- batchInsetrSql.Remove(batchInsetrSql.Length - 1,1).Append("\r\n;\r\n");
- }
- return batchInsetrSql.ToString();
- }
- }
- public override string FormatDateTimeOffset(object value)
- {
- var date = UtilMethods.ConvertFromDateTimeOffset((DateTimeOffset)value);
- return "'" + date.ToString("yyyy-MM-dd HH:mm:ss.fff zzz") + "'";
- }
- }
- }
|