- 补充 fsql.InsertOrUpdate UpdateColumns 数据存在时只更新指定的字段；

Author: 28810

FreeSql.DbContext/FreeSql.DbContext.xml

@@ -296,6 +296,7 @@ public void AsType()
         public void EnableAddOrUpdateNavigateList_OneToMany()
         {
             var repo = g.sqlite.GetRepository<Cagetory>();
+            repo.DbContextOptions.EnableAddOrUpdateNavigateList = true;
             var cts = new[] {
                 new Cagetory
                 {

FreeSql.Tests/FreeSql.Tests.DbContext/RepositoryTests.cs

@@ -159,6 +159,8 @@ public Author123(long id, long songId)
         [Fact]
         public void Test03()
         {
+            var sqlxx = g.pgsql.InsertOrUpdate<userinfo>().SetSource(new userinfo { userid = 10 }).UpdateColumns(a => new { a.birthday, a.CardNo }).ToSql();
+
             var aff1 = g.sqlite.GetRepository<Edi, long>().Delete(10086);
             var aff2 = g.sqlite.Delete<Edi>(10086).ExecuteAffrows();
             Assert.Equal(aff1, aff2);

FreeSql.Tests/FreeSql.Tests/UnitTest3.cs

@@ -42,6 +42,19 @@ public interface IInsertOrUpdate<T1> where T1 : class
         /// <returns></returns>
         IInsertOrUpdate<T1> IfExistsDoNothing();
 
+        /// <summary>
+        /// 当记录存在时，指定只更新的字段，UpdateColumns(a => a.Name) | UpdateColumns(a => new{a.Name,a.Time}) | UpdateColumns(a => new[]{"name","time"})
+        /// </summary>
+        /// <param name="columns">lambda选择列</param>
+        /// <returns></returns>
+        IInsertOrUpdate<T1> UpdateColumns(Expression<Func<T1, object>> columns);
+        /// <summary>
+        /// 当记录存在时，指定只更新的字段
+        /// </summary>
+        /// <param name="columns">属性名，或者字段名</param>
+        /// <returns></returns>
+        IInsertOrUpdate<T1> UpdateColumns(string[] columns);
+
         /// <summary>
         /// 设置表名规则，可用于分库/分表，参数1：默认表名；返回值：新表名；
         /// </summary>

FreeSql/FreeSql.xml

@@ -15,17 +15,18 @@ namespace FreeSql.Internal.CommonProvider
 
     public abstract partial class InsertOrUpdateProvider<T1> : IInsertOrUpdate<T1> where T1 : class
     {
-        protected IFreeSql _orm;
-        protected CommonUtils _commonUtils;
-        protected CommonExpression _commonExpression;
-        protected List<T1> _source = new List<T1>();
-        protected bool _doNothing = false;
-        protected Dictionary<string, bool> _auditValueChangedDict = new Dictionary<string, bool>(StringComparer.CurrentCultureIgnoreCase);
-        protected TableInfo _table;
-        protected Func<string, string> _tableRule;
-        protected DbParameter[] _params;
-        protected DbTransaction _transaction;
-        protected DbConnection _connection;
+        public IFreeSql _orm;
+        public CommonUtils _commonUtils;
+        public CommonExpression _commonExpression;
+        public List<T1> _source = new List<T1>();
+        public bool _doNothing = false;
+        public Dictionary<string, bool> _updateIgnore = new Dictionary<string, bool>(StringComparer.CurrentCultureIgnoreCase);
+        public Dictionary<string, bool> _auditValueChangedDict = new Dictionary<string, bool>(StringComparer.CurrentCultureIgnoreCase);
+        public TableInfo _table;
+        public Func<string, string> _tableRule;
+        public DbParameter[] _params;
+        public DbTransaction _transaction;
+        public DbConnection _connection;
         public ColumnInfo IdentityColumn { get; }
 
         public InsertOrUpdateProvider(IFreeSql orm, CommonUtils commonUtils, CommonExpression commonExpression)
@@ -57,6 +58,17 @@ public IInsertOrUpdate<T1> WithConnection(DbConnection connection)
             return this;
         }
 
+        public IInsertOrUpdate<T1> UpdateColumns(Expression<Func<T1, object>> columns) => UpdateColumns(_commonExpression.ExpressionSelectColumns_MemberAccess_New_NewArrayInit(null, columns?.Body, false, null));
+        public IInsertOrUpdate<T1> UpdateColumns(string[] columns)
+        {
+            var cols = columns.Distinct().ToDictionary(a => a);
+            _updateIgnore.Clear();
+            foreach (var col in _table.Columns.Values)
+                if (cols.ContainsKey(col.Attribute.Name) == false && cols.ContainsKey(col.CsName) == false)
+                    _updateIgnore.Add(col.Attribute.Name, true);
+            return this;
+        }
+
         public static void AuditDataValue(object sender, IEnumerable<T1> data, IFreeSql orm, TableInfo table, Dictionary<string, bool> changedDict)
         {
             if (data?.Any() != true) return;

FreeSql/Interface/Curd/IInsertOrUpdate.cs

@@ -37,7 +37,7 @@ string getMergeSql(List<T1> data)
                 WriteSourceSelectUnionAll(data, sb, dbParams);
                 sb.Append(" ) t2 ON (").Append(string.Join(" AND ", _table.Primarys.Select(a => $"t1.{_commonUtils.QuoteSqlName(a.Attribute.Name)} = t2.{a.Attribute.Name}"))).Append(") \r\n");
 
-                var cols = _table.Columns.Values.Where(a => a.Attribute.IsPrimary == false && a.Attribute.CanUpdate == true);
+                var cols = _table.Columns.Values.Where(a => a.Attribute.IsPrimary == false && a.Attribute.CanUpdate == true && _updateIgnore.ContainsKey(a.Attribute.Name) == false);
                 if (_doNothing == false && cols.Any())
                     sb.Append("WHEN MATCHED THEN \r\n")
                         .Append("  update set ").Append(string.Join(", ", cols.Select(a =>

FreeSql/Internal/CommonProvider/InsertOrUpdateProvider.cs

@@ -44,7 +44,12 @@ string getInsertSql(List<T1> data, bool flagInsert)
                 {
                     insert.InsertIdentity();
                     if (_doNothing == false)
-                        sql = new OnDuplicateKeyUpdate<T1>(insert).ToSql();
+                    {
+                        var cols = _table.Columns.Values.Where(a => a.Attribute.IsPrimary == false && a.Attribute.CanUpdate == true && _updateIgnore.ContainsKey(a.Attribute.Name) == false);
+                        sql = new OnDuplicateKeyUpdate<T1>(insert)
+                            .UpdateColumns(cols.Select(a => a.Attribute.Name).ToArray())
+                            .ToSql();
+                    }
                     else
                     {
                         if (_table.Primarys.Any() == false) throw new Exception($"fsql.InsertOrUpdate + IfExistsDoNothing + MySql 要求实体类 {_table.CsName} 必须有主键");

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

@@ -37,7 +37,7 @@ string getMergeSql(List<T1> data)
                 WriteSourceSelectUnionAll(data, sb, dbParams);
                 sb.Append(" ) t2 ON (").Append(string.Join(" AND ", _table.Primarys.Select(a => $"t1.{_commonUtils.QuoteSqlName(a.Attribute.Name)} = t2.{a.Attribute.Name}"))).Append(") \r\n");
 
-                var cols = _table.Columns.Values.Where(a => a.Attribute.IsPrimary == false && a.Attribute.CanUpdate == true);
+                var cols = _table.Columns.Values.Where(a => a.Attribute.IsPrimary == false && a.Attribute.CanUpdate == true && _updateIgnore.ContainsKey(a.Attribute.Name) == false);
                 if (_doNothing == false && cols.Any())
                     sb.Append("WHEN MATCHED THEN \r\n")
                         .Append("  update set ").Append(string.Join(", ", cols.Select(a =>

Providers/FreeSql.Provider.MySql/Curd/MySqlInsertOrUpdate.cs

@@ -43,8 +43,9 @@ string getInsertSql(List<T1> data, bool flagInsert)
                 else
                 {
                     var ocdu = new OdbcKingbaseESOnConflictDoUpdate<T1>(insert.InsertIdentity());
-                    ocdu.IgnoreColumns(_table.Columns.Values.Where(a => a.Attribute.CanUpdate == false).Select(a => a.Attribute.Name).ToArray());
-                    if (_doNothing == true || _table.Columns.Values.Where(a => a.Attribute.IsPrimary == false && a.Attribute.CanUpdate == true).Any() == false)
+                    var cols = _table.Columns.Values.Where(a => a.Attribute.IsPrimary == false && a.Attribute.CanUpdate == true && _updateIgnore.ContainsKey(a.Attribute.Name) == false);
+                    ocdu.UpdateColumns(cols.Select(a => a.Attribute.Name).ToArray());
+                    if (_doNothing == true || cols.Any() == false)
                         ocdu.DoNothing();
                     sql = ocdu.ToSql();
                 }