tk/FreeSql
1
0
Fork 0
mirror of https://github.com/nsnail/FreeSql.git synced 2025-06-19 12:28:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

- 修复 pgsql OnConflictDoUpdate 临时主键问题；#1393

Browse Source
This commit is contained in:
2881099
2023-01-05 16:53:25 +08:00
parent 4cd57a0afa
commit 3897da4f4f
7 changed files with 127 additions and 185 deletions
Download Patch File Download Diff File Expand all files Collapse all files

122
FreeSql.Tests/FreeSql.Tests/PostgreSQL/Curd/OnConflictDoUpdateTest.cs
View File

@ -1,7 +1,8 @@
using FreeSql.DataAnnotations;
using FreeSql.DataAnnotations;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Reflection;
using Xunit;
namespace FreeSql.Tests.PostgreSQL
@ -16,6 +17,125 @@ namespace FreeSql.Tests.PostgreSQL
public DateTime? time { get; set; }
}
[Table(Name = "demo_class1"), Index("uk_demo1", "name", true)]
public class DemoClass1
{
[Column(Name = "id")]
public int Id { get; set; }
[Column(Name = "name", IsNullable = false)]
public string Name { get; set; }
[Column(Name = "desc")]
public string Description { get; set; }
#region 系统非业务基础字段
//更新操作忽略此字段 只在OnConflictDoUpdate的插入操作时生效
[Column(Name = "created_id", CanUpdate = false, InsertValueSql = "1")]
public virtual int CreatedId { get; set; }
//插入操作忽略此字段 只在OnConflictDoUpdate的更新操作时生效
[Column(Name = "modified_id", CanInsert = false, InsertValueSql = "1")]
[UpdateValueSql("1")]
public virtual int? ModifiedId { get; set; }
//更新操作忽略此字段 只在OnConflictDoUpdate的插入操作时生效
[Column(Name = "created_time", CanUpdate = false, ServerTime = DateTimeKind.Local)]
public virtual DateTime CreatedTime { get; set; }
//插入操作忽略此字段 只在OnConflictDoUpdate的更新操作时生效
[Column(Name = "modified_time", CanInsert = false, ServerTime = DateTimeKind.Local)]
public virtual DateTime? ModifiedTime { get; set; }
#endregion
}
[Table(Name = "demo_class2")]
public class DemoClass2
{
[Column(Name = "name", IsNullable = false)]
public string Name { get; set; }
[Column(Name = "desc")]
public string Description { get; set; }
#region 系统非业务基础字段
//更新操作忽略此字段 只在OnConflictDoUpdate的插入操作时生效
[Column(Name = "created_id", CanUpdate = false, InsertValueSql = "1")]
public virtual int CreatedId { get; set; }
//插入操作忽略此字段 只在OnConflictDoUpdate的更新操作时生效
[Column(Name = "modified_id", CanInsert = false, InsertValueSql = "1")]
[UpdateValueSql("1")]
public virtual int? ModifiedId { get; set; }
//更新操作忽略此字段 只在OnConflictDoUpdate的插入操作时生效
[Column(Name = "created_time", CanUpdate = false, ServerTime = DateTimeKind.Local)]
public virtual DateTime CreatedTime { get; set; }
//插入操作忽略此字段 只在OnConflictDoUpdate的更新操作时生效
[Column(Name = "modified_time", CanInsert = false, ServerTime = DateTimeKind.Local)]
public virtual DateTime? ModifiedTime { get; set; }
#endregion
}
class UpdateValueSqlAttribute : Attribute
{
public string Value { get; set; }
public UpdateValueSqlAttribute(string value) => Value = value;
}
[Fact]
public void Issues1393()
{
var fsql = g.pgsql;
//跟随 FreeSqlBuilder Build 之后初始化，批量设置实体类：
foreach (var entity in new[] { typeof(DemoClass1) })
{
var table = fsql.CodeFirst.GetTableByEntity(entity);
table.Properties.Values
.Select(a => new { Property = a, UpdateValueSql = a.GetCustomAttribute<UpdateValueSqlAttribute>()?.Value })
.Where(a => a.UpdateValueSql != null)
.ToList()
.ForEach(a =>
{
var col = table.ColumnsByCs[a.Property.Name];
col.GetType().GetProperty("DbUpdateValue").SetValue(col, a.UpdateValueSql);
});
}
var sql = fsql.Insert(Enumerable.Range(1, 5).Select(i => new DemoClass1 { Id = i, Name = $"Name{i}", Description = $"Description{i}" }))
.NoneParameter()
.OnConflictDoUpdate(a => new { a.Name })
.ToSql();
Assert.Equal(@"INSERT INTO ""demo_class1""(""id"", ""name"", ""desc"", ""created_id"", ""created_time"") VALUES(1, 'Name1', 'Description1', 1, current_timestamp), (2, 'Name2', 'Description2', 1, current_timestamp), (3, 'Name3', 'Description3', 1, current_timestamp), (4, 'Name4', 'Description4', 1, current_timestamp), (5, 'Name5', 'Description5', 1, current_timestamp)
ON CONFLICT(""name"") DO UPDATE SET
""name"" = EXCLUDED.""name"",
""desc"" = EXCLUDED.""desc"",
""modified_id"" = 1,
""modified_time"" = current_timestamp", sql);
sql = fsql.Insert(Enumerable.Range(1, 5).Select(i => new DemoClass2 { Name = $"Name{i}", Description = $"Description{i}", ModifiedId = 1 }))
.NoneParameter()
.OnConflictDoUpdate(a => new { a.Name })
.ToSql();
Assert.Equal(@"INSERT INTO ""demo_class2""(""name"", ""desc"", ""created_id"", ""created_time"") VALUES('Name1', 'Description1', 1, current_timestamp), ('Name2', 'Description2', 1, current_timestamp), ('Name3', 'Description3', 1, current_timestamp), ('Name4', 'Description4', 1, current_timestamp), ('Name5', 'Description5', 1, current_timestamp)
ON CONFLICT(""name"") DO UPDATE SET
""name"" = EXCLUDED.""name"",
""desc"" = EXCLUDED.""desc"",
""modified_id"" = CASE EXCLUDED.""name""
WHEN 'Name1' THEN 1
WHEN 'Name2' THEN 1
WHEN 'Name3' THEN 1
WHEN 'Name4' THEN 1
WHEN 'Name5' THEN 1 END::int4,
""modified_time"" = current_timestamp", sql);
//sql = g.pgsql.Insert(data)
// .NoneParameter()
// .OnConflictDoUpdate(a => new { a.Name })
// .UpdateColumns()
// .ToSql();
}
[Fact]
public void ExecuteAffrows()
{