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

Synchronous code

Browse Source
This commit is contained in:
sunkaixuan
2025-08-24 14:36:27 +08:00
parent cde0f2d92d
commit 72a96f5f46
1 changed files with 26 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
Src/Asp.NetCore2/SqlSugar.MongoDbCore/ExpToSql/VisitorItems/MethodCallExpressionTractor.cs
View File

@@ -21,6 +21,7 @@ namespace SqlSugar.MongoDb
_context = context; _context = context;
_visitorContext = visitorContext; _visitorContext = visitorContext;
} }
public BsonValue Extract(Expression expr) public BsonValue Extract(Expression expr)
{ {
if (ExpressionTool.GetParameters(expr).Count == 0) if (ExpressionTool.GetParameters(expr).Count == 0)
@@ -46,7 +47,7 @@ namespace SqlSugar.MongoDb
{ {
return InvokeMongoMethod(name, out result, context, model); return InvokeMongoMethod(name, out result, context, model);
} }
else if (name.StartsWith("Add")) else if (IsAddMethod(name))
{ {
result = AddDateByType(name, context, model); result = AddDateByType(name, context, model);
} }
@@ -68,6 +69,7 @@ namespace SqlSugar.MongoDb
} }
} }
#region Core
private BsonValue InvokeMongoMethod(string name, out BsonValue result, MongoDbMethod context, MethodCallExpressionModel model) private BsonValue InvokeMongoMethod(string name, out BsonValue result, MongoDbMethod context, MethodCallExpressionModel model)
{ {
if (name == nameof(ToString)) if (name == nameof(ToString))
@@ -178,11 +180,6 @@ namespace SqlSugar.MongoDb
} }
} }
private static bool IsEndCondition(string name)
{
return name == "End";
}
public BsonValue BuildMongoSwitch(List<KeyValuePair<string, Expression>> ifConditions) public BsonValue BuildMongoSwitch(List<KeyValuePair<string, Expression>> ifConditions)
{ {
if (ifConditions == null || ifConditions.Count < 3) if (ifConditions == null || ifConditions.Count < 3)
@@ -221,12 +218,12 @@ namespace SqlSugar.MongoDb
if (MongoDbExpTools.GetIsMemember(returnPair.Value)) if (MongoDbExpTools.GetIsMemember(returnPair.Value))
{ {
thenExpr = UtilMethods.GetMemberName(thenExpr); thenExpr = UtilMethods.GetMemberName(thenExpr);
} }
branches.Add(new BsonDocument branches.Add(new BsonDocument
{ {
{ "case", caseExpr }, { "case", caseExpr },
{ "then", thenExpr } { "then", thenExpr }
}); });
i += 2; // 跳过 RETURN i += 2; // 跳过 RETURN
} }
else if (key == "END") else if (key == "END")
@@ -244,7 +241,7 @@ namespace SqlSugar.MongoDb
ifConditions.Last().Value, ifConditions.Last().Value,
context, context,
new ExpressionVisitorContext { IsText = true } new ExpressionVisitorContext { IsText = true }
); );
var switchDoc = new BsonDocument var switchDoc = new BsonDocument
{ {
{ "$switch", new BsonDocument { "$switch", new BsonDocument
@@ -253,11 +250,10 @@ namespace SqlSugar.MongoDb
{ "default", defaultValue } { "default", defaultValue }
} }
} }
}; };
return switchDoc; return switchDoc;
} }
private static string TransformMethodName(MethodCallExpression methodCallExpression, string name) private static string TransformMethodName(MethodCallExpression methodCallExpression, string name)
{ {
if (name == "ToDateTime") if (name == "ToDateTime")
@@ -269,7 +265,7 @@ namespace SqlSugar.MongoDb
{ {
name = "ContainsArray"; name = "ContainsArray";
} }
else if (name == "Contains" && methodCallExpression.Arguments.Count ==2 else if (name == "Contains" && methodCallExpression.Arguments.Count == 2
&& methodCallExpression.Arguments.FirstOrDefault().Type.IsArray && methodCallExpression.Arguments.FirstOrDefault().Type.IsArray
&& ExpressionTool.GetParameters(methodCallExpression.Arguments.FirstOrDefault()).Count == 0 && ExpressionTool.GetParameters(methodCallExpression.Arguments.FirstOrDefault()).Count == 0
) )
@@ -277,20 +273,32 @@ namespace SqlSugar.MongoDb
name = "ContainsArray"; name = "ContainsArray";
} }
else if (name == "Contains" && methodCallExpression.Arguments.Count == 1 else if (name == "Contains" && methodCallExpression.Arguments.Count == 1
&& methodCallExpression?.Object!=null && methodCallExpression?.Object != null
&&UtilMethods.IsCollectionOrArrayButNotByteArray(methodCallExpression.Object.Type) && UtilMethods.IsCollectionOrArrayButNotByteArray(methodCallExpression.Object.Type)
&& ExpressionTool.GetParameters(methodCallExpression?.Object).Count() >0 && ExpressionTool.GetParameters(methodCallExpression?.Object).Count() > 0
&& ExpressionTool.GetParameters(methodCallExpression.Arguments.FirstOrDefault()).Count == 0) && ExpressionTool.GetParameters(methodCallExpression.Arguments.FirstOrDefault()).Count == 0)
{ {
name = "JsonArrayAny"; name = "JsonArrayAny";
} }
return name; return name;
} }
#endregion
#region Helper
private static bool IsEndCondition(string name)
{
return name == "End";
}
private static bool IsCountJson(MethodCallExpression methodCallExpression, string name) private static bool IsCountJson(MethodCallExpression methodCallExpression, string name)
{ {
return name == "Count" && methodCallExpression?.Arguments?.FirstOrDefault() is MemberExpression m; return name == "Count" && methodCallExpression?.Arguments?.FirstOrDefault() is MemberExpression m;
}
private static bool IsAddMethod(string name)
{
return name.StartsWith("Add");
} }
#endregion
} }
} }