lsm/SqlSugar
1
0
Fork 0
mirror of https://gitee.com/dotnetchina/SqlSugar.git synced 2025-04-05 17:37:58 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update questdb bulkcopy

Browse Source
This commit is contained in:
sunkaixuan 2024-04-20 16:43:35 +08:00
parent e756b928fc
commit ec918466bc
2 changed files with 42 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
Src/Asp.NetCore2/SqlSugar.QuestDb.RestApi/CsvHelperEnumToIntConverter.cs Normal file
View File

@ -0,0 +1,34 @@
using CsvHelper.Configuration;
using CsvHelper.TypeConversion;
using CsvHelper;
using System;
using System.Collections.Generic;
using System.Text;
namespace SqlSugar
{
public class CsvHelperEnumToIntConverter : ITypeConverter
{
public string ConvertToString(object value, IWriterRow row, MemberMapData memberMapData)
{
if (value == null)
{
return "null";
}
else if (value is Enum enumValue)
{
return (Convert.ToInt32(enumValue)).ToString();
}
throw new NotSupportedException();
}
public object ConvertFromString(string text, IReaderRow row, MemberMapData memberMapData)
{
if (int.TryParse(text, out int intValue))
{
return text;
}
throw new NotSupportedException();
}
}
}

10
Src/Asp.NetCore2/SqlSugar.QuestDb.RestApi/QuestDbRestAPI.cs
View File

@ -197,12 +197,18 @@ namespace SqlSugar
private void CsvCreating<T>(CsvWriter csv) where T : class, new()
{
var entityColumns = db.EntityMaintenance.GetEntityInfo<T>().Columns;
if (entityColumns.Any(it => it.IsIgnore))
if (entityColumns.Any(it => it.IsIgnore||it.UnderType?.IsEnum==true))
{
var customMap = new DefaultClassMap<T>();
foreach (var item in entityColumns.Where(it => !it.IsIgnore))
{
customMap.Map(typeof(T), item.PropertyInfo).Name(item.PropertyName);
var memberMap = customMap.Map(typeof(T), item.PropertyInfo).Name(item.PropertyName);
if (item.UnderType?.IsEnum==true
&&item.SqlParameterDbType==null
&&db.CurrentConnectionConfig?.MoreSettings?.TableEnumIsString!=true)
{
memberMap.TypeConverter<CsvHelperEnumToIntConverter>();
}
}
csv.Context.RegisterClassMap(customMap);
}