KdbndpInsertBuilder.cs 4.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108
  1. using System;
  2. using System.Linq;
  3. using System.Text;
  4. namespace SqlSugar
  5. {
  6. public class KdbndpInsertBuilder : InsertBuilder
  7. {
  8. public override string SqlTemplate
  9. {
  10. get
  11. {
  12. if (IsReturnIdentity)
  13. {
  14. return @"INSERT INTO {0}
  15. ({1})
  16. VALUES
  17. ({2}) returning $PrimaryKey";
  18. }
  19. else
  20. {
  21. return @"INSERT INTO {0}
  22. ({1})
  23. VALUES
  24. ({2}) ;";
  25. }
  26. }
  27. }
  28. public override string SqlTemplateBatch => "INSERT INTO {0} ({1})";
  29. public override string SqlTemplateBatchUnion => " VALUES ";
  30. public override string SqlTemplateBatchSelect => " {0} ";
  31. public override string ToSqlString()
  32. {
  33. if (IsNoInsertNull)
  34. {
  35. DbColumnInfoList = DbColumnInfoList.Where(it => it.Value != null).ToList();
  36. }
  37. var groupList = DbColumnInfoList.GroupBy(it => it.TableId).ToList();
  38. var isSingle = groupList.Count() == 1;
  39. string columnsString = string.Join(",", groupList.First().Select(it => Builder.GetTranslationColumnName(it.DbColumnName)));
  40. if (isSingle)
  41. {
  42. string columnParametersString = string.Join(",", this.DbColumnInfoList.Select(it =>base.GetDbColumn(it, Builder.SqlParameterKeyWord + it.DbColumnName)));
  43. return string.Format(SqlTemplate, GetTableNameString, columnsString, columnParametersString);
  44. }
  45. else
  46. {
  47. StringBuilder batchInsetrSql = new StringBuilder();
  48. int pageSize = 200;
  49. int pageIndex = 1;
  50. int totalRecord = groupList.Count;
  51. int pageCount = (totalRecord + pageSize - 1) / pageSize;
  52. while (pageCount >= pageIndex)
  53. {
  54. batchInsetrSql.AppendFormat(SqlTemplateBatch, GetTableNameString, columnsString);
  55. int i = 0;
  56. foreach (var columns in groupList.Skip((pageIndex - 1) * pageSize).Take(pageSize).ToList())
  57. {
  58. var isFirst = i == 0;
  59. if (isFirst)
  60. {
  61. batchInsetrSql.Append(SqlTemplateBatchUnion);
  62. }
  63. batchInsetrSql.Append("\r\n ( " + string.Join(",", columns.Select(it =>
  64. {
  65. if (it.InsertServerTime || it.InsertSql.HasValue() || it.SqlParameterDbType is Type || it?.PropertyType?.Name == "DateOnly" || it?.PropertyType?.Name == "TimeOnly")
  66. {
  67. return GetDbColumn(it, null);
  68. }
  69. object value = null;
  70. if (it.Value is DateTime)
  71. {
  72. value = ((DateTime)it.Value).ToString("O");
  73. }
  74. else if (it.Value is DateTimeOffset)
  75. {
  76. return FormatDateTimeOffset(it.Value);
  77. }
  78. else
  79. {
  80. value = it.Value;
  81. }
  82. if (value == null||value==DBNull.Value)
  83. {
  84. return string.Format(SqlTemplateBatchSelect, "NULL");
  85. }
  86. return string.Format(SqlTemplateBatchSelect, "'" + value.ObjToString().ToSqlFilter() + "'");
  87. })) + "),");
  88. ++i;
  89. }
  90. pageIndex++;
  91. batchInsetrSql.Remove(batchInsetrSql.Length - 1,1).Append("\r\n;\r\n");
  92. }
  93. return batchInsetrSql.ToString();
  94. }
  95. }
  96. public override string FormatDateTimeOffset(object value)
  97. {
  98. var date = UtilMethods.ConvertFromDateTimeOffset((DateTimeOffset)value);
  99. return "'" + date.ToString("yyyy-MM-dd HH:mm:ss.fff zzz") + "'";
  100. }
  101. }
  102. }