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

TDengine BulkCopy

Browse Source
This commit is contained in:
sunkaixuan
2025-04-22 17:34:16 +08:00
parent 878042ba50
commit d4560619d0
1 changed files with 31 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -48,38 +48,42 @@ namespace SqlSugar.TDengine
}
public async Task BulkInsertToTDengine(TDengineConnection conn, string tableName, DataTable table, bool isTran, string[] tagColumns)
{
string insertSql=string.Empty;
try
{
if (tagColumns != null && tagColumns.Length > 0)
{
string tags = string.Join(", ", tagColumns.Select(tag => FormatValue(tag)));
foreach (var item in tagColumns)
{
table.Columns.Remove(item);
}
// Build the column names and value placeholders
var valuePlaceholdersList = table.Rows.Cast<DataRow>().Select(row =>
{
var values = row.ItemArray.Select(item => FormatValue(item)).ToList();
return $"({string.Join(", ", values)})";
});
var valuePlaceholders = string.Join(", ", valuePlaceholdersList);
// 排除标签列，确保数据列不包含标签列
var columnNames = string.Join(", ", table.Columns.Cast<DataColumn>()
.Select(c => c.ColumnName));
// 生成插入语句，USING 部分包含标签列，VALUES 部分只包含数据列
insertSql = $"INSERT INTO {tableName} USING {tableName} TAGS({tags}) ({columnNames}) VALUES {valuePlaceholders}";
}
else
{
// Build the column names and value placeholders
var valuePlaceholdersList = table.Rows.Cast<DataRow>().Select(row =>
{
var values = row.ItemArray.Select(item => FormatValue(item)).ToList();
// If there are tags, move them to the beginning of the values list and adjust insert syntax
if (tagColumns != null && tagColumns.Length > 0)
{
foreach (var tag in tagColumns)
{
int index = table.Columns.IndexOf(tag);
values.Insert(0, FormatValue(row[index]));
}
}
return $"({string.Join(", ", values)})";
}).ToList();
var valuePlaceholders = string.Join(", ", valuePlaceholdersList);
// Check if tagColumns is provided and adjust the SQL statement accordingly
string insertSql;
if (tagColumns != null && tagColumns.Length > 0)
{
// Construct SQL with tags included
var columnNames = string.Join(", ", tagColumns.Concat(table.Columns.Cast<DataColumn>().Where(c => !tagColumns.Contains(c.ColumnName)).Select(c => c.ColumnName)));
insertSql = $"INSERT INTO {tableName} ({columnNames}) VALUES {valuePlaceholders}";
}
else
{
// Construct SQL without tags
var columnNames = string.Join(", ", table.Columns.Cast<DataColumn>().Select(c => c.ColumnName));
insertSql = $"INSERT INTO {tableName} ({columnNames}) VALUES {valuePlaceholders}";
@@ -88,9 +92,9 @@ namespace SqlSugar.TDengine
// Execute the command asynchronously
await this.Context.Ado.ExecuteCommandAsync(insertSql);
}
catch
catch(Exception ex)
{
throw;
throw new Exception(ex.Message +"\r\n"+ insertSql);
}
finally
{