解决ClickHouse 批量更新DateTime问题

Author: d4ilys

FreeSql.Tests/FreeSql.Tests/ClickHouse/ClickhouseIssueTest.cs

@@ -22,7 +22,7 @@ public ClickhouseIssueTest(ITestOutputHelper output)
             _fsql = new FreeSqlBuilder().UseConnectionString(DataType.ClickHouse,
                     "Host=192.168.1.123;Port=8123;Database=test_issue;Compress=True;Min Pool Size=1")
                 .UseMonitorCommand(cmd => _output.WriteLine($"线程：{cmd.CommandText}\r\n"))
-                .UseNoneCommandParameter(true)
+                .UseNoneCommandParameter(false)
                 .UseAdoConnectionPool(true)
                 .Build();
         }
@@ -32,25 +32,35 @@ public ClickhouseIssueTest(ITestOutputHelper output)
         [Fact]
         public void TestIssue1813()
         {
-            //var personsUpdate = new List<Person>
-            //{
-            //    new Person
-            //    {
-            //        Id = 1,
-            //        Name = $"test2{DateTime.Now.Millisecond}",
-            //        Age = 20,
-            //        CreateTime = DateTime.Now
-            //    },
-            //    new Person
-            //    {
-            //        Id = 2,
-            //        Name = "test3"+ 286,
-            //        Age = 22,
-            //        CreateTime = DateTime.Now
-            //    }
-            //};
-
-            //_fsql.Update<Person>().SetSource()
+            //普通修改
+            _fsql.Update<Person>()
+                .Set(p => p.Name == "update_name")
+                .Set(p => p.UpdateTime == DateTime.Now)
+                .Where(p => p.Id == "25e8d92e-29f2-43ff-b861-9ade0eec4041")
+                .ExecuteAffrows();
+
+            //批量修改
+            var updatePerson = new List<Person>();
+            updatePerson.Add(new Person
+            {
+                Id = "9cd7af52-85cc-4d26-898a-4020cadb0491",
+                Name = "update_name1",
+                UpdateTime = DateTime.Now
+            });
+
+            updatePerson.Add(new Person
+            {
+                Id = "bd9f9ed6-bd03-4675-abb4-12b7fdac7678",
+                Name = "update_name2",
+                UpdateTime = DateTime.Now
+            });
+
+            _fsql.Update<Person>().SetSource(updatePerson).UpdateColumns(person => new
+            {
+                person.Name,
+                person.UpdateTime,
+            }).ExecuteAffrows();
+
         }
 
         [Fact]

Providers/FreeSql.Provider.ClickHouse/ClickHouseUtils.cs

@@ -158,7 +158,7 @@ public override string GetNoneParamaterSqlValue(List<DbParameter> specialParams,
                 var ts = (TimeSpan)value;
                 value = $"{Math.Floor(ts.TotalHours)}:{ts.Minutes}:{ts.Seconds}";
             }
-			else if (value is Array)
+            else if (value is Array)
 			{
 				var valueArr = value as Array;
 				var eleType = type.GetElementType();

Providers/FreeSql.Provider.ClickHouse/Curd/ClickHouseUpdate.cs

@@ -7,6 +7,7 @@
 using System.Text;
 using System.Threading;
 using System.Threading.Tasks;
+using DateTime = System.DateTime;
 
 namespace FreeSql.ClickHouse.Curd
 {
@@ -167,6 +168,16 @@ public override void ToSqlExtension110(StringBuilder sb, bool isAsTableSplited)
 
                                 var colsql = _noneParameter ? _commonUtils.GetNoneParamaterSqlValue(_paramsSource, "u", col, col.Attribute.MapType, val) :
                                     _commonUtils.QuoteWriteParamterAdapter(col.Attribute.MapType, _commonUtils.QuoteParamterName($"p_{_paramsSource.Count}"));
+
+                                //判断是否是DateTime类型，如果是DateTime类型，需要转换成ClickHouse支持的时间格式
+                                if (col.Attribute.MapType == typeof(DateTime) || col.Attribute.MapType == typeof(DateTime?) )
+                                {
+                                    //获取当前实时区
+                                    var timeZone = TimeZoneInfo.Local;
+
+                                    colsql = $"toDateTime({colsql},'Asia/Shanghai')";
+                                }
+
                                 cwsb.Append(_commonUtils.RewriteColumn(col, colsql));
                                 if (_noneParameter == false)
                                     _commonUtils.AppendParamter(_paramsSource, null, col, col.Attribute.MapType, val);