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

Update mongodb

Browse Source
This commit is contained in:
sunkaixuan
2025-05-11 14:39:59 +08:00
parent c915fd409e
commit 92b64b6260
4 changed files with 77 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
Src/Asp.NetCore2/SqlSugar.MongoDbCore/ExpToSql/VisitorItems/BinaryExpressionTranslator/CalculationOperationExpression.cs Normal file
View File

@@ -0,0 +1,42 @@
using MongoDB.Bson;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Linq.Expressions;
using System.Text;
namespace SqlSugar.MongoDb
{
public partial class BinaryExpressionTranslator
{
private BsonDocument GetCalculationOperation(BsonValue field, ExpressionType nodeType, BsonValue value, bool leftIsMember, bool rightIsMember)
{
string operation = GetCalculationType(nodeType);
if (IsStringAdd(value, operation))
return StringAddCalculation(field, value, leftIsMember, rightIsMember, out operation);
else
return GetCommonCalculation(field, value, operation);
}
private static BsonDocument GetCommonCalculation(BsonValue field, BsonValue value, string operation)
{
return new BsonDocument
{
{ field.ToString(), new BsonDocument { { operation, value } } }
};
}
private static BsonDocument StringAddCalculation(BsonValue field, BsonValue value, bool leftIsMember, bool rightIsMember, out string operation)
{
operation = "$concat";
return new BsonDocument
{
{ operation, new BsonArray { UtilMethods.GetBsonValue(leftIsMember, field), UtilMethods.GetBsonValue(rightIsMember, value) } }
};
;
}
private static bool IsStringAdd(BsonValue value, string operation)
{
return operation == "$add" && value.BsonType == BsonType.String;
}
}
}

34
Src/Asp.NetCore2/SqlSugar.MongoDbCore/ExpToSql/VisitorItems/BinaryExpressionTranslator/FieldComparisonExpression.cs → Src/Asp.NetCore2/SqlSugar.MongoDbCore/ExpToSql/VisitorItems/BinaryExpressionTranslator/ComparisonOperationExpression.cs
View File

@@ -9,14 +9,6 @@ namespace SqlSugar.MongoDb
{
public partial class BinaryExpressionTranslator
{
private BsonDocument FieldComparisonOrCalculationExpression(BinaryExpression expr)
{
OutParameters(expr, out var field, out var value, out var leftIsMember, out var rightIsMember, out var op);
return op == null
? GetCalculationOperation(field, expr.NodeType, value, leftIsMember, rightIsMember)
: GetComparisonOperation(expr, field, value, leftIsMember, rightIsMember, op);
}
private BsonDocument GetComparisonOperation(BinaryExpression expr, BsonValue field, BsonValue value, bool leftIsMember, bool rightIsMember, string op)
{
string leftValue = "";
@@ -76,31 +68,5 @@ namespace SqlSugar.MongoDb
}
}
private BsonDocument GetCalculationOperation(BsonValue field, ExpressionType nodeType, BsonValue value, bool leftIsMember, bool rightIsMember)
{
string operation = nodeType switch
{
ExpressionType.Add => "$add",
ExpressionType.Subtract => "$subtract",
ExpressionType.Multiply => "$multiply",
ExpressionType.Divide => "$divide",
ExpressionType.Modulo => "$mod",
_ => throw new NotSupportedException($"Unsupported calculation operation: {nodeType}")
};
if (operation == "$add" && value.BsonType == BsonType.String)
{
operation = "$concat";
return new BsonDocument
{
{ operation, new BsonArray { UtilMethods.GetBsonValue(leftIsMember, field), UtilMethods.GetBsonValue(rightIsMember, value) } }
};
;
}
return new BsonDocument
{
{ field.ToString(), new BsonDocument { { operation, value } } }
};
}
}
}

20
Src/Asp.NetCore2/SqlSugar.MongoDbCore/ExpToSql/VisitorItems/BinaryExpressionTranslator/FieldExpressionCase.cs Normal file
View File

@@ -0,0 +1,20 @@
using MongoDB.Bson;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Linq.Expressions;
using System.Text;
namespace SqlSugar.MongoDb
{
public partial class BinaryExpressionTranslator
{
private BsonDocument FieldComparisonOrCalculationExpression(BinaryExpression expr)
{
OutParameters(expr, out var field, out var value, out var leftIsMember, out var rightIsMember, out var op);
return op == null
? GetCalculationOperation(field, expr.NodeType, value, leftIsMember, rightIsMember)
: GetComparisonOperation(expr, field, value, leftIsMember, rightIsMember, op);
}
}
}

13
Src/Asp.NetCore2/SqlSugar.MongoDbCore/ExpToSql/VisitorItems/BinaryExpressionTranslator/Helper.cs
View File

@@ -30,5 +30,18 @@ namespace SqlSugar.MongoDb
_ => null
};
}
private static string GetCalculationType(ExpressionType nodeType)
{
return nodeType switch
{
ExpressionType.Add => "$add",
ExpressionType.Subtract => "$subtract",
ExpressionType.Multiply => "$multiply",
ExpressionType.Divide => "$divide",
ExpressionType.Modulo => "$mod",
_ => throw new NotSupportedException($"Unsupported calculation operation: {nodeType}")
};
}
}
}