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

修复构建多常量参数语法异常问题

Browse Source
This commit is contained in:
Kenton 2025-04-17 17:08:57 +08:00
parent 686c3d249f
commit 501deb47cf
2 changed files with 46 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
Src/Asp.NetCore2/DB2CoreTest/CURD/CodeFirst.cs
View File

File diff suppressed because one or more lines are too long

39
Src/Asp.NetCore2/SqlSugar.Db2Core/DB2/SqlBuilder/DB2QueryBuilder.cs
View File

@ -8,9 +8,9 @@ namespace SqlSugar.DB2
public partial class DB2QueryBuilder : QueryBuilder
{
/// <summary>
/// 静态常量标志
/// 静态常量正则表达式
/// </summary>
private string[] constanFlag = new string[] { "@const", "@constan" };
private const string constRegex = "@.*const.*";
#region Sql Template
public override string PageTempalte
@ -153,23 +153,28 @@ namespace SqlSugar.DB2
{
var selectParams = selectSql.Split(",").Select(p =>
{
if (!constanFlag.Any(t => p.ToLower().Contains(t))) return p;
var parameter = this.Parameters.FirstOrDefault(t => p.Contains(t.ParameterName));
if (parameter == null) return p;
if (!Regex.IsMatch(p, constRegex)) return p;
var parameterItems = this.Parameters.Where(t => p.Contains(t.ParameterName)).ToList();
if (!parameterItems.Any()) return p;
var dbType = DB2DbBind.MappingDbTypesConst.FirstOrDefault(t => parameter.DbType == t.Value);
if (default(KeyValuePair<string, System.Data.DbType>).Equals(dbType)) return p;
var parameterized = string.Empty;
if (dbType.Value == System.Data.DbType.String)
var paramSql = p;
parameterItems.ForEach(parameter =>
{
var size = string.IsNullOrEmpty(parameter.Value?.ToString()) ? 1 : System.Text.Encoding.Default.GetBytes(parameter.Value?.ToString()).Length;;
parameterized = $"{dbType.Key.ToUpper()}({size})";
}
else
{
parameterized = $"{dbType.Key.ToUpper()}";
}
return p.Replace(parameter.ParameterName, $" CAST({parameter.ParameterName} AS {parameterized}) ");
var dbType = DB2DbBind.MappingDbTypesConst.FirstOrDefault(t => parameter.DbType == t.Value);
if (default(KeyValuePair<string, System.Data.DbType>).Equals(dbType)) return;
var parameterized = string.Empty;
if (dbType.Value == System.Data.DbType.String)
{
var size = string.IsNullOrEmpty(parameter.Value?.ToString()) ? 1 : System.Text.Encoding.Default.GetBytes(parameter.Value?.ToString()).Length;
parameterized = $"{dbType.Key.ToUpper()}({size})";
}
else
{
parameterized = $"{dbType.Key.ToUpper()}";
}
paramSql = paramSql.Replace(parameter.ParameterName, $" CAST({parameter.ParameterName} AS {parameterized}) ");
});
return paramSql;
});
return string.Join(",", selectParams);
}