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-06-28 08:51:23 +08:00
parent 32c5e7a37f
commit 1ac0d7c715
1 changed files with 119 additions and 89 deletions
Download Patch File Download Diff File Expand all files Collapse all files

208
Src/Asp.NetCore2/SqlSugar.MongoDbCore/MongoDb/SqlBuilder/MongoDbQueryBuilder.cs
View File

@@ -55,89 +55,41 @@ namespace SqlSugar.MongoDb
List<string> operations = new List<string>(); List<string> operations = new List<string>();
var sb = new StringBuilder(); var sb = new StringBuilder();
#region Where #region Joins
foreach (var item in this.WhereInfos)
{
// 去除开头的 WHERE 或 AND（忽略大小写和空格）
string trimmed = item.TrimStart();
if (trimmed.StartsWith("WHERE", StringComparison.OrdinalIgnoreCase))
trimmed = trimmed.Substring(5).TrimStart();
else if (trimmed.StartsWith("AND", StringComparison.OrdinalIgnoreCase))
trimmed = trimmed.Substring(3).TrimStart();
if (MongoDbExpTools.IsFieldNameJson(trimmed))
{
var outerDoc = BsonDocument.Parse(trimmed);
trimmed = outerDoc[UtilConstants.FieldName].AsString;
operations.Add(trimmed);
}
else
{
// item 是 JSON 格式字符串，直接包进 $match
operations.Add($"{{ \"$match\": {trimmed} }}");
}
}
#endregion #endregion
#region Page ProcessWhereConditions(operations);
var skip = this.Skip;
var take = this.Take;
// 处理 skip 和 take
if (this.Skip.HasValue)
{
operations.Add($"{{ \"$skip\": {this.Skip.Value} }}");
}
if (this.Take.HasValue)
{
operations.Add($"{{ \"$limit\": {this.Take.Value} }}");
}
#endregion
#region OrderBy ProcessPagination(operations);
var order = this.GetOrderByString;
var orderByString = this.GetOrderByString?.Trim();
if (orderByString == "ORDER BY NOW()")
orderByString = null;
if (!string.IsNullOrEmpty(orderByString) && orderByString.StartsWith("ORDER BY ", StringComparison.OrdinalIgnoreCase))
{
order = order.Substring("ORDER BY ".Length).Trim();
var sortDoc = new BsonDocument(); ProcessOrderByConditions(operations);
foreach (var str in order.Split(","))
{ ProcessSelectConditions(operations);
int lastSpace = str.LastIndexOf(' ');
string jsonPart = str.Substring(0, lastSpace).Trim(); ProcessGroupByConditions(operations);
string directionPart = str.Substring(lastSpace + 1).Trim().ToUpper();
if (str.EndsWith("}")) sb.Append($"aggregate {this.GetTableNameString} ");
{ sb.Append("[");
jsonPart = str; sb.Append(string.Join(", ", operations));
directionPart = "ASC"; sb.Append("]");
}
var bson = MongoDB.Bson.Serialization.BsonSerializer.Deserialize<BsonDocument>(jsonPart); return sb.ToString();
if (bson.Contains(UtilConstants.FieldName)) }
{
var field = bson[UtilConstants.FieldName].AsString; private void ProcessSelectConditions(List<string> operations)
var direction = directionPart == "DESC" ? -1 : 1; {
sortDoc[field] = direction;
}
}
if (sortDoc.ElementCount > 0)
{
operations.Add($"{{ \"$sort\": {sortDoc.ToJson()} }}");
}
}
#endregion
#region Select #region Select
if (this.SelectValue is Expression expression) if (this.SelectValue is Expression expression)
{ {
var dos=MongoNestedTranslator.Translate(expression, new MongoNestedTranslatorContext() { var dos = MongoNestedTranslator.Translate(expression, new MongoNestedTranslatorContext()
{
context = this.Context, context = this.Context,
resolveType=ResolveExpressType.SelectSingle, resolveType = ResolveExpressType.SelectSingle,
queryBuilder=this queryBuilder = this
}); });
if (MongoDbExpTools.IsFieldNameJson(dos)) if (MongoDbExpTools.IsFieldNameJson(dos))
{ {
dos[UtilConstants.FieldName] = "$"+ dos[UtilConstants.FieldName]; dos[UtilConstants.FieldName] = "$" + dos[UtilConstants.FieldName];
dos.Add(new BsonElement("_id", "0")); dos.Add(new BsonElement("_id", "0"));
} }
else if (dos.ElementCount > 0 && dos.GetElement(0).Name.StartsWith("$")) else if (dos.ElementCount > 0 && dos.GetElement(0).Name.StartsWith("$"))
@@ -147,12 +99,15 @@ namespace SqlSugar.MongoDb
dos.Clear(); dos.Clear();
dos.Add(UtilConstants.FieldName, funcDoc); dos.Add(UtilConstants.FieldName, funcDoc);
dos.Add(new BsonElement("_id", "0")); dos.Add(new BsonElement("_id", "0"));
} }
var json = dos.ToJson(UtilMethods.GetJsonWriterSettings()); var json = dos.ToJson(UtilMethods.GetJsonWriterSettings());
operations.Add($"{{\"$project\": {json} }}"); operations.Add($"{{\"$project\": {json} }}");
} }
#endregion #endregion
}
private void ProcessGroupByConditions(List<string> operations)
{
#region GroupBy #region GroupBy
if (this.GroupByValue.HasValue()) if (this.GroupByValue.HasValue())
{ {
@@ -166,19 +121,19 @@ namespace SqlSugar.MongoDb
var selectItem = match.Groups[1].Value; var selectItem = match.Groups[1].Value;
selectItems.Add(selectItem); selectItems.Add(selectItem);
} }
var jsonPart = "["+Regex.Split(this.GroupByValue, UtilConstants.ReplaceCommaKey) var jsonPart = "[" + Regex.Split(this.GroupByValue, UtilConstants.ReplaceCommaKey)
.First() .First()
.TrimEnd('(') .TrimEnd('(')
.Replace("GROUP BY ", "")+"]"; .Replace("GROUP BY ", "") + "]";
var fieldNames = new List<string>(); var fieldNames = new List<string>();
var bsonArray = MongoDB.Bson.Serialization.BsonSerializer.Deserialize<BsonArray>(jsonPart); var bsonArray = MongoDB.Bson.Serialization.BsonSerializer.Deserialize<BsonArray>(jsonPart);
foreach (BsonDocument bson in bsonArray) foreach (BsonDocument bson in bsonArray)
{ {
if (bson.Contains(UtilConstants.FieldName)) if (bson.Contains(UtilConstants.FieldName))
{ {
var field = bson[UtilConstants.FieldName].AsString; var field = bson[UtilConstants.FieldName].AsString;
operations[operations.Count - 1] = operations[operations.Count - 1].Replace($"\"${field}\"", $"\"$_id.{field}\""); operations[operations.Count - 1] = operations[operations.Count - 1].Replace($"\"${field}\"", $"\"$_id.{field}\"");
fieldNames.Add(field); fieldNames.Add(field);
} }
} }
// 构造 _id 部分：支持多字段形式 // 构造 _id 部分：支持多字段形式
@@ -203,17 +158,92 @@ namespace SqlSugar.MongoDb
groupFields.Add(element.Name, element.Value); groupFields.Add(element.Name, element.Value);
} }
} }
operations.Insert(operations.Count-1, groupDoc.ToJson(UtilMethods.GetJsonWriterSettings())); operations.Insert(operations.Count - 1, groupDoc.ToJson(UtilMethods.GetJsonWriterSettings()));
} }
#endregion #endregion
}
sb.Append($"aggregate {this.GetTableNameString} "); private void ProcessOrderByConditions(List<string> operations)
sb.Append("["); {
sb.Append(string.Join(", ", operations)); #region OrderBy
sb.Append("]"); var order = this.GetOrderByString;
var orderByString = this.GetOrderByString?.Trim();
if (orderByString == "ORDER BY NOW()")
orderByString = null;
if (!string.IsNullOrEmpty(orderByString) && orderByString.StartsWith("ORDER BY ", StringComparison.OrdinalIgnoreCase))
{
order = order.Substring("ORDER BY ".Length).Trim();
var sortDoc = new BsonDocument();
foreach (var str in order.Split(","))
{
int lastSpace = str.LastIndexOf(' ');
string jsonPart = str.Substring(0, lastSpace).Trim();
string directionPart = str.Substring(lastSpace + 1).Trim().ToUpper();
if (str.EndsWith("}"))
{
jsonPart = str;
directionPart = "ASC";
}
var bson = MongoDB.Bson.Serialization.BsonSerializer.Deserialize<BsonDocument>(jsonPart);
if (bson.Contains(UtilConstants.FieldName))
{
var field = bson[UtilConstants.FieldName].AsString;
var direction = directionPart == "DESC" ? -1 : 1;
sortDoc[field] = direction;
}
}
if (sortDoc.ElementCount > 0)
{
operations.Add($"{{ \"$sort\": {sortDoc.ToJson()} }}");
}
}
#endregion
}
private void ProcessPagination(List<string> operations)
{
#region Page
var skip = this.Skip;
var take = this.Take;
// 处理 skip 和 take
if (this.Skip.HasValue)
{
operations.Add($"{{ \"$skip\": {this.Skip.Value} }}");
}
if (this.Take.HasValue)
{
operations.Add($"{{ \"$limit\": {this.Take.Value} }}");
}
#endregion
}
private void ProcessWhereConditions(List<string> operations)
{
#region Where
foreach (var item in this.WhereInfos)
{
// 去除开头的 WHERE 或 AND（忽略大小写和空格）
string trimmed = item.TrimStart();
if (trimmed.StartsWith("WHERE", StringComparison.OrdinalIgnoreCase))
trimmed = trimmed.Substring(5).TrimStart();
else if (trimmed.StartsWith("AND", StringComparison.OrdinalIgnoreCase))
trimmed = trimmed.Substring(3).TrimStart();
if (MongoDbExpTools.IsFieldNameJson(trimmed))
{
var outerDoc = BsonDocument.Parse(trimmed);
trimmed = outerDoc[UtilConstants.FieldName].AsString;
operations.Add(trimmed);
}
else
{
// item 是 JSON 格式字符串，直接包进 $match
operations.Add($"{{ \"$match\": {trimmed} }}");
}
}
#endregion
}
return sb.ToString();
}
public override string ToCountSql(string sql) public override string ToCountSql(string sql)
{ {
sql=sql.TrimEnd(']'); sql=sql.TrimEnd(']');