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

Update core

Browse Source
This commit is contained in:
sunkaixuna
2021-11-20 19:04:51 +08:00
parent 48cd5570f7
commit bd5ec286e3
3 changed files with 149 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
Src/Asp.NetCore2/SqlSeverTest/SqlSugar/Abstract/FastestProvider/Private.cs
View File

@@ -22,7 +22,7 @@ namespace SqlSugar
case DbType.Oracle:
break;
case DbType.PostgreSQL:
break;
return new PostgreSQLFastBuilder(this.entityInfo);
case DbType.Dm:
break;
case DbType.Kdbndp:
@@ -97,14 +97,14 @@ namespace SqlSugar
{
value = Convert.ToDateTime("1900-01-01");
}
else if (columnInfo.IsJson)
{
columnInfo.IsJson = true;
}
else if (columnInfo.IsArray)
{
columnInfo.IsArray = true;
}
//else if (columnInfo.IsJson)
//{
// columnInfo.IsJson = true;
//}
//else if (columnInfo.IsArray)
//{
// columnInfo.IsArray = true;
//}
else if (columnInfo.UnderType.IsEnum() )
{
value = Convert.ToInt64(value);

123
Src/Asp.NetCore2/SqlSeverTest/SqlSugar/Realization/PostgreSQL/SqlBuilder/PostgreSQLFastBuilder.cs Normal file
View File

@@ -0,0 +1,123 @@
using Npgsql;
using NpgsqlTypes;
using System;
using System.Collections.Generic;
using System.Data;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
namespace SqlSugar
{
public class PostgreSQLFastBuilder : FastBuilder, IFastBuilder
{
public static Dictionary<string , NpgsqlDbType> PgSqlType = UtilMethods.EnumToDictionary<NpgsqlDbType>();
private EntityInfo entityInfo;
public PostgreSQLFastBuilder(EntityInfo entityInfo)
{
this.entityInfo = entityInfo;
}
public override string UpdateSql { get; set; } = @"UPDATE {1} SET {0} FROM {2} AS TE WHERE {3}
";
//public virtual async Task<int> UpdateByTempAsync(string tableName, string tempName, string[] updateColumns, string[] whereColumns)
//{
// Check.ArgumentNullException(!updateColumns.Any(), "update columns count is 0");
// Check.ArgumentNullException(!whereColumns.Any(), "where columns count is 0");
// var sets = string.Join(",", updateColumns.Select(it => $"TM.{it}=TE.{it}"));
// var wheres = string.Join(",", whereColumns.Select(it => $"TM.{it}=TE.{it}"));
// string sql = string.Format(UpdateSql, sets, tableName, tempName, wheres);
// return await this.Context.Ado.ExecuteCommandAsync(sql);
//}
public async Task<int> ExecuteBulkCopyAsync(DataTable dt)
{
List<string> lsColNames = new List<string>();
for (int i = 0; i < dt.Columns.Count; i++)
{
lsColNames.Add($"\"{dt.Columns[i].ColumnName}\"");
}
string copyString = $"COPY {dt.TableName} ( {string.Join(",", lsColNames) } ) FROM STDIN (FORMAT BINARY)";
NpgsqlConnection conn = (NpgsqlConnection)this.Context.Ado.Connection;
var columns = this.Context.DbMaintenance.GetColumnInfosByTableName(this.entityInfo.DbTableName);
try
{
var identityColumnInfo = this.entityInfo.Columns.FirstOrDefault(it => it.IsIdentity);
if (identityColumnInfo!=null)
{
throw new Exception("PgSql bulkcopy no support identity");
}
BulkCopy(dt, copyString, conn, columns);
}
catch (Exception ex)
{
base.CloseDb();
throw ex;
}
return await Task.FromResult(dt.Rows.Count);
}
private static void BulkCopy(DataTable dt, string copyString, NpgsqlConnection conn, List<DbColumnInfo> columns)
{
if (conn.State == ConnectionState.Closed)
conn.Open();
using (var writer = conn.BeginBinaryImport(copyString))
{
foreach (DataRow row in dt.Rows)
{
writer.StartRow();
foreach (DataColumn kvp in dt.Columns)
{
var first = columns.FirstOrDefault(it => it.DbColumnName.EqualCase(kvp.ColumnName));
var key = first.DataType.ToLower();
if (PgSqlType.ContainsKey(key))
{
WriterRow(writer, row, first);
}
else if (first.DataType.First() == '_')
{
var type = PgSqlType[key.Substring(1)];
writer.Write(row[kvp.ColumnName], NpgsqlDbType.Array | type);
}
else
{
writer.Write(row[kvp.ColumnName]);
}
}
}
writer.Complete();
}
}
private static void WriterRow(NpgsqlBinaryImporter writer, DataRow row, DbColumnInfo colInfo)
{
if (PgSqlType.ContainsKey(colInfo.DataType.ToLower()))
{
if (row[colInfo.DbColumnName] == null || row[colInfo.DbColumnName] == DBNull.Value)
{
writer.Write(DBNull.Value, NpgsqlDbType.Integer);
}
else
{
writer.Write(row[colInfo.DbColumnName], PgSqlType[colInfo.DataType.ToLower()]);
}
}
}
public override async Task<int> UpdateByTempAsync(string tableName, string tempName, string[] updateColumns, string[] whereColumns)
{
var sqlquerybulder= this.Context.Queryable<object>().SqlBuilder;
Check.ArgumentNullException(!updateColumns.Any(), "update columns count is 0");
Check.ArgumentNullException(!whereColumns.Any(), "where columns count is 0");
var sets = string.Join(",", updateColumns.Select(it => $"{sqlquerybulder.GetTranslationColumnName(it)}=TE.{sqlquerybulder.GetTranslationColumnName(it)}"));
var wheres = string.Join(",", whereColumns.Select(it => $"{tableName}.{sqlquerybulder.GetTranslationColumnName(it)}=TE.{sqlquerybulder.GetTranslationColumnName(it)}"));
string sql = string.Format(UpdateSql, sets, tableName, tempName, wheres);
return await this.Context.Ado.ExecuteCommandAsync(sql);
}
public override async Task CreateTempAsync<T>(DataTable dt)
{
await this.Context.Queryable<T>().Where(it => false).AS(dt.TableName).Select(" * into temp mytemptable").ToListAsync();
dt.TableName = "mytemptable";
}
}
}

17
Src/Asp.NetCore2/SqlSeverTest/SqlSugar/Utilities/UtilMethods.cs
View File

@@ -411,5 +411,22 @@ namespace SqlSugar
}
}
}
public static Dictionary<string, T> EnumToDictionary<T>()
{
Dictionary<string, T> dic = new Dictionary<string, T>();
if (!typeof(T).IsEnum)
{
return dic;
}
string desc = string.Empty;
foreach (var item in Enum.GetValues(typeof(T)))
{
var key = item.ToString().ToLower();
if (!dic.ContainsKey(key))
dic.Add(key, (T)item);
}
return dic;
}
}
}