UpdateablePage.cs 3.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102
  1. using System;
  2. using System.Collections.Generic;
  3. using System.Linq;
  4. using System.Text;
  5. using System.Threading.Tasks;
  6. namespace SqlSugar
  7. {
  8. public class UpdateablePage<T> where T:class,new()
  9. {
  10. public T[] DataList { get; set; }
  11. public SqlSugarProvider Context { get; set; }
  12. public int PageSize { get; internal set; }
  13. public string TableName { get; internal set; }
  14. public bool IsEnableDiffLogEvent { get; internal set; }
  15. public DiffLogModel DiffModel { get; internal set; }
  16. public List<string> UpdateColumns { get; internal set; }
  17. public string[] WhereColumnList { get; internal set; }
  18. public UpdateableFilter<T> EnableQueryFilter()
  19. {
  20. return new UpdateableFilter<T>()
  21. {
  22. Context= Context,
  23. DataList= DataList,
  24. DiffModel= DiffModel,
  25. IsEnableDiffLogEvent= IsEnableDiffLogEvent,
  26. PageSize=PageSize,
  27. TableName=TableName,
  28. UpdateColumns=UpdateColumns
  29. };
  30. }
  31. public int ExecuteCommand()
  32. {
  33. if (DataList.Count() == 1 && DataList.First() == null)
  34. {
  35. return 0;
  36. }
  37. if (PageSize == 0) { PageSize = 1000; }
  38. var result = 0;
  39. var isNoTran = this.Context.Ado.IsNoTran();
  40. try
  41. {
  42. if (isNoTran)
  43. {
  44. this.Context.Ado.BeginTran();
  45. }
  46. this.Context.Utilities.PageEach(DataList, PageSize, pageItem =>
  47. {
  48. result += this.Context.Updateable(pageItem).AS(TableName).WhereColumns(WhereColumnList).EnableDiffLogEventIF(IsEnableDiffLogEvent, DiffModel).UpdateColumns(UpdateColumns.ToArray()).ExecuteCommand();
  49. });
  50. if (isNoTran)
  51. {
  52. this.Context.Ado.CommitTran();
  53. }
  54. }
  55. catch (Exception)
  56. {
  57. if (isNoTran)
  58. {
  59. this.Context.Ado.RollbackTran();
  60. }
  61. throw;
  62. }
  63. return result;
  64. }
  65. public async Task<int> ExecuteCommandAsync()
  66. {
  67. if (DataList.Count() == 1 && DataList.First() == null)
  68. {
  69. return 0;
  70. }
  71. if (PageSize == 0) { PageSize = 1000; }
  72. var result = 0;
  73. var isNoTran = this.Context.Ado.IsNoTran();
  74. try
  75. {
  76. if (isNoTran)
  77. {
  78. await this.Context.Ado.BeginTranAsync();
  79. }
  80. await this.Context.Utilities.PageEachAsync(DataList, PageSize, async pageItem =>
  81. {
  82. result += await this.Context.Updateable(pageItem).AS(TableName).WhereColumns(WhereColumnList).EnableDiffLogEventIF(IsEnableDiffLogEvent, DiffModel).UpdateColumns(UpdateColumns.ToArray()).ExecuteCommandAsync();
  83. });
  84. if (isNoTran)
  85. {
  86. await this.Context.Ado.CommitTranAsync();
  87. }
  88. }
  89. catch (Exception)
  90. {
  91. if (isNoTran)
  92. {
  93. await this.Context.Ado.RollbackTranAsync();
  94. }
  95. throw;
  96. }
  97. return result;
  98. }
  99. }
  100. }