weiai/FreeSql
1
0
Fork 0
mirror of https://github.com/dotnetcore/FreeSql.git synced 2026-02-14 20:30:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

处理超级表查询/参数化问题

Browse Source
This commit is contained in:
d4ilys
2024-11-25 14:13:51 +08:00
parent 2f2ec05fd4
commit 592247c0ac
1 changed files with 56 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

64
Providers/FreeSql.Provider.TDengine/TDengineProvider.cs
View File

@@ -1,7 +1,13 @@
using FreeSql.Internal.CommonProvider;
using System;
using System.Data.Common;
using System.Linq;
using System.Reflection;
using System.Threading;
using FreeSql.Provider.TDengine.Attributes;
using FreeSql.TDengine.Curd;
using Newtonsoft.Json.Linq;
using TDengine.Data.Client;
namespace FreeSql.TDengine
{
@@ -10,7 +16,6 @@ namespace FreeSql.TDengine
public TDengineProvider(string masterConnectionString, string[] slaveConnectionString,
Func<DbConnection> connectionFactory = null)
{
this.InternalCommonUtils = new TDengineUtils(this);
this.InternalCommonExpression = new TDengineExpression(this.InternalCommonUtils);
this.Ado = new TDengineAdo(this.InternalCommonUtils, masterConnectionString, slaveConnectionString,
@@ -18,31 +23,74 @@ namespace FreeSql.TDengine
this.Aop = new AopProvider();
this.DbFirst = new TDengineDbFirst(this, this.InternalCommonUtils, this.InternalCommonExpression);
this.CodeFirst = new TDengineCodeFirst(this, this.InternalCommonUtils, this.InternalCommonExpression);
//处理超级表查询问题
this.Aop.ConfigEntityProperty += (s, e) =>
{
if (e.Property.ReflectedType == null) return;
if (e.Property.ReflectedType.BaseType == null) return;
var propertyInfo = e.Property.ReflectedType.BaseType.GetProperty(e.Property.Name);
if (propertyInfo == null) return;
if (propertyInfo.GetCustomAttribute(typeof(TDengineTagAttribute)) != null)
e.ModifyResult.IsIgnore = true;
};
//处理参数化
this.Aop.CommandBefore += (_, e) =>
{
if (e.Command.Parameters.Count <= 0) return;
var dengineParameters = new TDengineParameter[e.Command.Parameters.Count];
e.Command.Parameters.CopyTo(dengineParameters, 0);
var cmdText = e.Command.CommandText;
var isChanged = false;
foreach (var parameter in dengineParameters.OrderByDescending(a => a.ParameterName.Length))
{
var idx = cmdText.IndexOf(parameter.ParameterName, StringComparison.Ordinal);
if (idx != -1)
{
isChanged = true;
cmdText =
$"{cmdText.Substring(0, idx)}?{cmdText.Substring(idx + parameter.ParameterName.Length)}";
}
}
if (isChanged) e.Command.CommandText = cmdText;
};
}
public override ISelect<T1> CreateSelectProvider<T1>(object dywhere) => new TDengineSelect<T1>(this, this.InternalCommonUtils, this.InternalCommonExpression, dywhere);
public override ISelect<T1> CreateSelectProvider<T1>(object dywhere) =>
new TDengineSelect<T1>(this, this.InternalCommonUtils, this.InternalCommonExpression, dywhere);
public override IInsert<T1> CreateInsertProvider<T1>() =>
new TDengineInsert<T1>(this, this.InternalCommonUtils, this.InternalCommonExpression);
public override IInsert<T1> CreateInsertProvider<T1>() => new TDengineInsert<T1>(this, this.InternalCommonUtils, this.InternalCommonExpression);
public override IUpdate<T1> CreateUpdateProvider<T1>(object dywhere)
{
throw new NotImplementedException();
throw new NotImplementedException($"FreeSql.Provider.TDengine {CoreStrings.S_Not_Implemented_Feature}");
}
public override IDelete<T1> CreateDeleteProvider<T1>(object dywhere)
{
throw new NotImplementedException();
return new TDengineDelete<T1>(this, this.InternalCommonUtils, this.InternalCommonExpression, dywhere);
}
public override IInsertOrUpdate<T1> CreateInsertOrUpdateProvider<T1>()
{
throw new NotImplementedException();
throw new NotImplementedException($"FreeSql.Provider.TDengine {CoreStrings.S_Not_Implemented_Feature}");
}
~TDengineProvider() => this.Dispose();
int _disposeCounter;
public override void Dispose()
{
throw new NotImplementedException();
if (Interlocked.Increment(ref _disposeCounter) != 1) return;
(this.Ado as AdoProvider)?.Dispose();
}
}
}