lsm/SqlSugar
1
0
Fork 0
mirror of https://gitee.com/dotnetchina/SqlSugar.git synced 2025-07-16 16:50:41 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update TDengine BulkCopy

Browse Source
This commit is contained in:
sunkaixuan 2025-04-22 20:16:19 +08:00
parent 87d90e1c4d
commit f3d3dfdac1
1 changed files with 25 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
Src/Asp.NetCore2/SqlSugar.TDengineCore/TDengine/SqlBuilder/TDengineFastBuilder.cs
View File

@ -38,15 +38,6 @@ namespace SqlSugar.TDengine
return dt.Rows.Count; return dt.Rows.Count;
} }
public static void SetTags(ISqlSugarClient db, Func<string, string> action, params string [] tagNames)
{
if (db.TempItems == null)
{
db.TempItems = new Dictionary<string, object>();
}
db.TempItems.Add(TagKey, tagNames);
db.TempItems.Add(TagKey+"action", action);
}
public async Task BulkInsertToTDengine(TDengineConnection conn, string tableName, DataTable table, bool isTran, string[] tagColumns) public async Task BulkInsertToTDengine(TDengineConnection conn, string tableName, DataTable table, bool isTran, string[] tagColumns)
{ {
@ -138,24 +129,41 @@ namespace SqlSugar.TDengine
var columnNames = string.Join(", ", table.Columns.Cast<DataColumn>() var columnNames = string.Join(", ", table.Columns.Cast<DataColumn>()
.Select(c => builder.GetTranslationColumnName(c.ColumnName))); .Select(c => builder.GetTranslationColumnName(c.ColumnName)));
var action = this.Context.TempItems[TagKey + "action"] as Func<string, string>; var action = this.Context.TempItems[TagKey + "action"] as Func<string, string,string>;
var subTableName = builder.GetTranslationColumnName(action(tagsValues)); var subTableName = builder.GetTranslationColumnName(action(tagsValues, tableName.Replace("`","")));
// sbtables.AppendLine($"CREATE TABLE {subTableName} USING {tableName} TAGS({tags});"); // sbtables.AppendLine($"CREATE TABLE {subTableName} USING {tableName} TAGS({tags});");
var sqlBuilder = sb; var sqlBuilder = sb;
var valuesList = new List<string>();
foreach (DataRow row in table.Rows) foreach (DataRow row in table.Rows)
{ {
var values = row.ItemArray.Select(item => FormatValue(item)).ToList(); var values = row.ItemArray.Select(item => FormatValue(item)).ToList();
var valuePart = string.Join(", ", values); var valuePart = $"({string.Join(", ", values)})";
string insertSql = $"INSERT INTO {subTableName} USING {tableName} TAGS({tags}) ({columnNames}) VALUES ({valuePart});"; valuesList.Add(valuePart);
}
if (valuesList.Count > 0)
{
string insertSql = $"INSERT INTO {subTableName} USING {tableName} TAGS({tags}) ({columnNames}) VALUES {string.Join(", ", valuesList)};";
sqlBuilder.AppendLine(insertSql); sqlBuilder.AppendLine(insertSql);
} }
return sqlBuilder; return sqlBuilder;
} }
public static void SetTags(ISqlSugarClient db, Func<string, string, string> action, params string[] tagNames)
{
if (db.TempItems == null)
{
db.TempItems = new Dictionary<string, object>();
}
db.TempItems.Add(TagKey, tagNames);
db.TempItems.Add(TagKey + "action", action);
}
public object FormatValue(object value) public object FormatValue(object value)
{ {
if (value == null || value == DBNull.Value) if (value == null || value == DBNull.Value)
@ -167,7 +175,7 @@ namespace SqlSugar.TDengine
var type = value.GetType(); var type = value.GetType();
if (type == UtilConstants.DateType) if (type == UtilConstants.DateType)
{ {
return Convert.ToDateTime(value).ToString("yyyy-MM-dd HH:mm:ss.ms").ToSqlValue(); return Convert.ToDateTime(value).ToString("yyyy-MM-dd HH:mm:ss.fff").ToSqlValue();
} }
else if (type == UtilConstants.ByteArrayType) else if (type == UtilConstants.ByteArrayType)
{ {