lsm/SqlSugar
1
0
Fork 0
mirror of https://gitee.com/dotnetchina/SqlSugar.git synced 2025-11-08 18:34:55 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update QuestDb

Browse Source
This commit is contained in:
sunkaixuan
2022-07-30 17:23:40 +08:00
parent 1d33cb2bfd
commit 89482f3125
3 changed files with 52 additions and 147 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
Src/Asp.NetCore2/SqlSugar/Realization/QuestDB/DbMaintenance/QuestDBDbMaintenance.cs
View File

@@ -12,7 +12,7 @@ namespace SqlSugar
{ {
get get
{ {
throw new NotSupportedException("GetDataBase"); return CreateDataBaseSql;
} }
} }
protected override string GetColumnInfosByTableNameSql protected override string GetColumnInfosByTableNameSql
@@ -37,7 +37,7 @@ namespace SqlSugar
{ {
get get
{ {
throw new NotSupportedException("GetView"); return "select * from (select 1 as id) t where id=0";
} }
} }
#endregion #endregion
@@ -47,7 +47,7 @@ namespace SqlSugar
{ {
get get
{ {
return ""; return "select * from (select 1 as id) t where id=0";
} }
} }
protected override string AddPrimaryKeySql protected override string AddPrimaryKeySql

2
Src/Asp.NetCore2/SqlSugar/Realization/QuestDB/SqlBuilder/QuestDBInsertBuilder.cs
View File

@@ -80,7 +80,7 @@ namespace SqlSugar
object value = null; object value = null;
if (it.Value is DateTime) if (it.Value is DateTime)
{ {
value = ((DateTime)it.Value).ToString("O"); return $" cast('{it.Value.ObjToDate().ToString("yyyy-MM-dd HH:mm:ss.fff")}' as date)";
} }
else else
{ {

191
Src/Asp.NetCore2/SqlSugar/Realization/QuestDB/SqlBuilder/QuestDBUpdateBuilder.cs
View File

@@ -2,39 +2,41 @@
using System.Collections.Generic; using System.Collections.Generic;
using System.Linq; using System.Linq;
using System.Text; using System.Text;
using System.Text.RegularExpressions;
namespace SqlSugar namespace SqlSugar
{ {
public class QuestDBUpdateBuilder : UpdateBuilder public class QuestDBUpdateBuilder : UpdateBuilder
{ {
public override string SqlTemplateBatch protected override string TomultipleSqlString(List<IGrouping<int, DbColumnInfo>> groupList)
{ {
get StringBuilder sb = new StringBuilder();
int i = 0;
sb.AppendLine(string.Join("\r\n", groupList.Select(t =>
{ {
return @"UPDATE {1} {2} SET {0} FROM ${{0}} "; var updateTable = string.Format("UPDATE {0} SET", base.GetTableNameStringNoWith);
} var setValues = string.Join(",", t.Where(s => !s.IsPrimarykey).Select(m => GetOracleUpdateColums(i, m, false)).ToArray());
} var pkList = t.Where(s => s.IsPrimarykey).ToList();
public override string SqlTemplateJoin List<string> whereList = new List<string>();
{ foreach (var item in pkList)
get {
{ var isFirst = pkList.First() == item;
return @" (VALUES var whereString = "";
{0} whereString += GetOracleUpdateColums(i, item, true);
whereList.Add(whereString);
) AS T ({2}) WHERE {1} }
"; i++;
} return string.Format("{0} {1} WHERE {2};", updateTable, setValues, string.Join(" AND", whereList));
}).ToArray()));
return sb.ToString();
} }
public override string SqlTemplateBatchUnion private string GetOracleUpdateColums(int i, DbColumnInfo m, bool iswhere)
{ {
get return string.Format("\"{0}\"={1}", m.DbColumnName.ToUpper(), FormatValue(i, m.DbColumnName, m.Value, iswhere));
{
return ",";
}
} }
public object FormatValue(object value,string name,int i,DbColumnInfo columnInfo) public object FormatValue(int i, string name, object value, bool iswhere)
{ {
if (value == null) if (value == null)
{ {
@@ -42,27 +44,29 @@ namespace SqlSugar
} }
else else
{ {
var type = value.GetType(); var type = UtilMethods.GetUnderType(value.GetType());
if (type == UtilConstants.DateType||columnInfo.IsArray||columnInfo.IsJson) if (type == UtilConstants.DateType && iswhere == false)
{
var date = value.ObjToDate();
if (date < UtilMethods.GetMinDate(this.Context.CurrentConnectionConfig))
{
date = UtilMethods.GetMinDate(this.Context.CurrentConnectionConfig);
}
if (this.Context.CurrentConnectionConfig?.MoreSettings?.DisableMillisecond == true)
{
return "'" + date.ToString("yyyy-MM-dd HH:mm:ss") + "'";
}
else
{
return "'" + date.ToString("yyyy-MM-dd HH:mm:ss.fff") + "'";
}
}
else if (type == UtilConstants.DateType && iswhere)
{ {
var parameterName = this.Builder.SqlParameterKeyWord + name + i; var parameterName = this.Builder.SqlParameterKeyWord + name + i;
var paramter = new SugarParameter(parameterName, value); this.Parameters.Add(new SugarParameter(parameterName, value));
if (columnInfo.IsJson)
{
paramter.IsJson = true;
}
if (columnInfo.IsArray)
{
paramter.IsArray = true;
}
this.Parameters.Add(paramter);
return parameterName; return parameterName;
} }
else if (type == UtilConstants.ByteArrayType)
{
string bytesString = "0x" + BitConverter.ToString((byte[])value);
return bytesString;
}
else if (type.IsEnum()) else if (type.IsEnum())
{ {
if (this.Context.CurrentConnectionConfig.MoreSettings?.TableEnumIsString == true) if (this.Context.CurrentConnectionConfig.MoreSettings?.TableEnumIsString == true)
@@ -74,6 +78,12 @@ namespace SqlSugar
return Convert.ToInt64(value); return Convert.ToInt64(value);
} }
} }
else if (type == UtilConstants.ByteArrayType)
{
var parameterName = this.Builder.SqlParameterKeyWord + name + i;
this.Parameters.Add(new SugarParameter(parameterName, value));
return parameterName;
}
else if (type == UtilConstants.BoolType) else if (type == UtilConstants.BoolType)
{ {
return value.ObjToBool() ? "1" : "0"; return value.ObjToBool() ? "1" : "0";
@@ -88,110 +98,5 @@ namespace SqlSugar
} }
} }
} }
protected override string TomultipleSqlString(List<IGrouping<int, DbColumnInfo>> groupList)
{
Check.Exception(PrimaryKeys == null || PrimaryKeys.Count == 0, " Update List<T> need Primary key");
int pageSize = 200;
int pageIndex = 1;
int totalRecord = groupList.Count;
int pageCount = (totalRecord + pageSize - 1) / pageSize;
StringBuilder batchUpdateSql = new StringBuilder();
while (pageCount >= pageIndex)
{
StringBuilder updateTable = new StringBuilder();
string setValues = string.Join(",", groupList.First().Where(it => it.IsPrimarykey == false && (it.IsIdentity == false || (IsOffIdentity && it.IsIdentity))).Select(it =>
{
if (SetValues.IsValuable())
{
var setValue = SetValues.Where(sv => sv.Key == Builder.GetTranslationColumnName(it.DbColumnName));
if (setValue != null && setValue.Any())
{
return setValue.First().Value;
}
}
var result = string.Format("{0}=T.{0}", Builder.GetTranslationColumnName(it.DbColumnName));
return result;
}));
string tempColumnValue = string.Join(",", groupList.First().Select(it =>
{
if (SetValues.IsValuable())
{
var setValue = SetValues.Where(sv => sv.Key == Builder.GetTranslationColumnName(it.DbColumnName));
if (setValue != null && setValue.Any())
{
return setValue.First().Value;
}
}
var result = Builder.GetTranslationColumnName(it.DbColumnName);
return result;
}));
batchUpdateSql.AppendFormat(SqlTemplateBatch.ToString(), setValues, GetTableNameStringNoWith, TableWithString);
int i = 0;
var tableColumnList = this.Context.DbMaintenance.GetColumnInfosByTableName(GetTableNameStringNoWith);
foreach (var columns in groupList.Skip((pageIndex - 1) * pageSize).Take(pageSize).ToList())
{
var isFirst = i == 0;
if (!isFirst)
{
updateTable.Append(SqlTemplateBatchUnion);
}
updateTable.Append("\r\n (" + string.Join(",", columns.Select(it =>
{
var columnInfo = tableColumnList.FirstOrDefault(x => x.DbColumnName.Equals(it.DbColumnName, StringComparison.OrdinalIgnoreCase));
var dbType = columnInfo?.DataType;
if (dbType == null) {
var typeName = it.PropertyType.Name.ToLower();
if (typeName == "int32")
typeName = "int";
if (typeName == "int64")
typeName = "long";
if (typeName == "int16")
typeName = "short";
if (typeName == "boolean")
typeName = "bool";
var isAnyType = PostgreSQLDbBind.MappingTypesConst.Where(x => x.Value.ToString().ToLower() == typeName).Any();
if (isAnyType)
{
dbType = PostgreSQLDbBind.MappingTypesConst.Where(x => x.Value.ToString().ToLower() == typeName).FirstOrDefault().Key;
}
else {
dbType = "varchar";
}
}
return string.Format("CAST({0} AS {1})", FormatValue(it.Value,it.DbColumnName,i,it), dbType);
})) + ")");
++i;
}
pageIndex++;
updateTable.Append("\r\n");
string whereString = null;
if (this.WhereValues.HasValue())
{
foreach (var item in WhereValues)
{
var isFirst = whereString == null;
whereString += (isFirst ? null : " AND ");
whereString += item;
}
}
else if (PrimaryKeys.HasValue())
{
foreach (var item in PrimaryKeys)
{
var isFirst = whereString == null;
whereString += (isFirst ? null : " AND ");
whereString += string.Format("{0}.{1}=T.{1}", GetTableNameStringNoWith, Builder.GetTranslationColumnName(item));
}
}
var format = string.Format(SqlTemplateJoin, updateTable, whereString, tempColumnValue);
batchUpdateSql.Replace("${0}", format);
batchUpdateSql.Append(";");
}
return batchUpdateSql.ToString();
}
} }
} }