From 21accea1bc2b437054bcc4020f8ceb3daef53b2e Mon Sep 17 00:00:00 2001 From: sunkaixuan <610262374@qq.com> Date: Mon, 5 May 2025 11:34:24 +0800 Subject: [PATCH] Update Mongodb --- .../MongoDbTest/OrmTest/Models/Order.cs | 8 +- .../MongoDbTest/OrmTest/OrmTest.cs | 29 ++- .../SqlBuilder/MongoDbUpdateBuilder.cs | 244 ++++-------------- .../SqlSugar.MongoDbCore/MongoDbBase.cs | 12 + 4 files changed, 88 insertions(+), 205 deletions(-) create mode 100644 Src/Asp.NetCore2/SqlSugar.MongoDbCore/MongoDbBase.cs diff --git a/Src/Asp.NetCore2/MongoDbTest/OrmTest/Models/Order.cs b/Src/Asp.NetCore2/MongoDbTest/OrmTest/Models/Order.cs index a01d56f2f..2e09d35a2 100644 --- a/Src/Asp.NetCore2/MongoDbTest/OrmTest/Models/Order.cs +++ b/Src/Asp.NetCore2/MongoDbTest/OrmTest/Models/Order.cs @@ -1,4 +1,5 @@ using SqlSugar; +using SqlSugar.MongoDbCore; using System; using System.Collections.Generic; using System.Linq; @@ -7,11 +8,8 @@ using System.Text; namespace MongoDbTest { - public class OrderInfo - { - [SugarColumn(IsPrimaryKey = true, IsOnlyIgnoreInsert =true)] - public string Id { get; set; } - + public class OrderInfo:MongoDbBase + { public string Name { get; set; } public decimal Price { get; set; } [SugarColumn(IsNullable = true)] diff --git a/Src/Asp.NetCore2/MongoDbTest/OrmTest/OrmTest.cs b/Src/Asp.NetCore2/MongoDbTest/OrmTest/OrmTest.cs index e396c5d16..b0907edef 100644 --- a/Src/Asp.NetCore2/MongoDbTest/OrmTest/OrmTest.cs +++ b/Src/Asp.NetCore2/MongoDbTest/OrmTest/OrmTest.cs @@ -8,20 +8,41 @@ using System.Threading.Tasks; namespace MongoDbTest { - public class OrmTest + public class OrmTest { - public static void Init() + public static void Init() { var db = DbHelper.GetNewDb(); db.Insertable(new OrderInfo() { CreateTime = DateTime.Now, Name = "a", Price = 1 }) .ExecuteCommand(); - var ids= db.Insertable(new List(){ + var ids = db.Insertable(new List(){ new OrderInfo() { CreateTime = DateTime.Now, Name = "a1", Price = 2 }, new OrderInfo() { CreateTime = DateTime.Now, Name = "a2", Price = 3 }}) .ExecuteReturnPkList(); - var delrow=db.Deleteable(new OrderInfo() { Id= ids.Last() }) + + var updateRow = db.Updateable(new List() + { + new OrderInfo() { Id = ids.First(),Name="a3"}, + new OrderInfo() { Id = ids.Last(),Name="a4"} + }) + .ExecuteCommand(); + + var delrow = db.Deleteable(new OrderInfo() { Id = ids.Last() }) .ExecuteCommand(); + + + //测试生成SQL性能 + TestSqlBuilder(db); + } + + private static void TestSqlBuilder(SqlSugar.SqlSugarClient db) + { + for (int i = 0; i < 10000; i++) + { + db.Insertable(new OrderInfo() { CreateTime = DateTime.Now, Name = "a", Price = 1 }) + .ToSql(); + } } } } diff --git a/Src/Asp.NetCore2/SqlSugar.MongoDbCore/MongoDb/SqlBuilder/MongoDbUpdateBuilder.cs b/Src/Asp.NetCore2/SqlSugar.MongoDbCore/MongoDb/SqlBuilder/MongoDbUpdateBuilder.cs index 53ea4212b..c34769eb5 100644 --- a/Src/Asp.NetCore2/SqlSugar.MongoDbCore/MongoDb/SqlBuilder/MongoDbUpdateBuilder.cs +++ b/Src/Asp.NetCore2/SqlSugar.MongoDbCore/MongoDb/SqlBuilder/MongoDbUpdateBuilder.cs @@ -1,7 +1,9 @@ -using System; +using MongoDB.Bson; +using System; using System.Collections.Generic; using System.Linq; using System.Text; +using System.Text.Json.Nodes; namespace SqlSugar.MongoDb { @@ -33,217 +35,67 @@ namespace SqlSugar.MongoDb return ","; } } - - public object FormatValue(object value,string name,int i,DbColumnInfo columnInfo) + protected override string ToSingleSqlString(List> groupList) { - if (value == null) - { - return "NULL"; - } - else - { - var type =UtilMethods.GetUnderType(value.GetType()); - if (type == UtilConstants.DateType||columnInfo.IsArray||columnInfo.IsJson) - { - var parameterName = this.Builder.SqlParameterKeyWord + name + i; - var paramter = new SugarParameter(parameterName, value); - if (columnInfo.IsJson) - { - paramter.IsJson = true; - } - if (columnInfo.IsArray) - { - paramter.IsArray = true; - } - this.Parameters.Add(paramter); - return parameterName; - } - else if (type == UtilConstants.DateTimeOffsetType) - { - return FormatDateTimeOffset(value); - } - else if (type == UtilConstants.ByteArrayType) - { - string bytesString = "0x" + BitConverter.ToString((byte[])value); - return bytesString; - } - else if (type.IsEnum()) - { - if (this.Context.CurrentConnectionConfig.MoreSettings?.TableEnumIsString == true) - { - return value.ToSqlValue(); - } - else - { - return Convert.ToInt64(value); - } - } - else if (type == UtilConstants.BoolType) - { - return value.ObjToBool() ? "1" : "0"; - } - else if (type == UtilConstants.StringType || type == UtilConstants.ObjType) - { - return "'" + value.ToString().ToSqlFilter() + "'"; - } - else - { - return "'" + value.ToString() + "'"; - } - } + var result = BuildUpdateMany(groupList, this.TableName); + return result; } - protected override string TomultipleSqlString(List> groupList) { - Check.Exception(PrimaryKeys == null || PrimaryKeys.Count == 0, " Update List need Primary key"); - int pageSize = 200; - int pageIndex = 1; - int totalRecord = groupList.Count; - int pageCount = (totalRecord + pageSize - 1) / pageSize; - StringBuilder batchUpdateSql = new StringBuilder(); - while (pageCount >= pageIndex) - { - StringBuilder updateTable = new StringBuilder(); - string setValues = string.Join(",", groupList.First().Where(it => it.IsPrimarykey == false && (it.IsIdentity == false || (IsOffIdentity && it.IsIdentity))).Select(it => - { - if (SetValues.IsValuable()) - { - var setValue = SetValues.Where(sv => sv.Key == Builder.GetTranslationColumnName(it.DbColumnName)); - if (setValue != null && setValue.Any()) - { - return setValue.First().Value; - } - } - var result = string.Format("{0}=T.{0}", Builder.GetTranslationColumnName(it.DbColumnName)); - return result; - })); - string tempColumnValue = string.Join(",", groupList.First().Select(it => - { - if (SetValues.IsValuable()) - { - var setValue = SetValues.Where(sv => sv.Key == Builder.GetTranslationColumnName(it.DbColumnName)); - if (setValue != null && setValue.Any()) - { - return setValue.First().Value; - } - } - var result = Builder.GetTranslationColumnName(it.DbColumnName); - return result; - })); - batchUpdateSql.AppendFormat(SqlTemplateBatch.ToString(), setValues, GetTableNameStringNoWith, TableWithString); - int i = 0; - var tableColumnList = this.Context.DbMaintenance.GetColumnInfosByTableName(GetTableNameStringNoWith); - - foreach (var columns in groupList.Skip((pageIndex - 1) * pageSize).Take(pageSize).ToList()) - { - var isFirst = i == 0; - if (!isFirst) - { - updateTable.Append(SqlTemplateBatchUnion); - } - updateTable.Append("\r\n (" + string.Join(",", columns.Select(it => - { - var columnInfo = tableColumnList.FirstOrDefault(x => x.DbColumnName.Equals(it.DbColumnName, StringComparison.OrdinalIgnoreCase)); - var dbType = columnInfo?.DataType; - if (dbType == null) { - var typeName = it.PropertyType.Name.ToLower(); - if (columnInfo==null&&it.PropertyType.IsEnum) - { - if (this.Context.CurrentConnectionConfig?.MoreSettings?.TableEnumIsString!=true) - { - typeName = "int"; - } - } - if (typeName == "int32") - typeName = "int"; - if (typeName == "int64") - typeName = "long"; - if (typeName == "int16") - typeName = "short"; - if (typeName == "boolean") - typeName = "bool"; - - var isAnyType = MongoDbDbBind.MappingTypesConst.Where(x => x.Value.ToString().ToLower() == typeName).Any(); - if (isAnyType) - { - dbType = MongoDbDbBind.MappingTypesConst.Where(x => x.Value.ToString().ToLower() == typeName).FirstOrDefault().Key; - } - else { - dbType = "varchar"; - } - } - return string.Format("CAST({0} AS {1})", base.GetDbColumn(it,FormatValue(it.Value,it.DbColumnName,i,it)), dbType); - - })) + ")"); - ++i; - } - pageIndex++; - updateTable.Append("\r\n"); - string whereString = null; - if (this.WhereValues.HasValue()) - { - foreach (var item in WhereValues) - { - var isFirst = whereString == null; - whereString += (isFirst ? null : " AND "); - whereString += item; - } - } - else if (PrimaryKeys.HasValue()) - { - foreach (var item in PrimaryKeys) - { - var isFirst = whereString == null; - whereString += (isFirst ? null : " AND "); - whereString += string.Format("{0}.{1}=T.{1}", GetTableNameStringNoWith, Builder.GetTranslationColumnName(item)); - } - } - var format = string.Format(SqlTemplateJoin, updateTable, whereString, tempColumnValue); - batchUpdateSql.Replace("${0}", format); - batchUpdateSql.Append(";"); - } - batchUpdateSql = GetBatchUpdateSql(batchUpdateSql); - return batchUpdateSql.ToString(); + var result = BuildUpdateMany(groupList,this.TableName); + return result; } - - private StringBuilder GetBatchUpdateSql(StringBuilder batchUpdateSql) + public string BuildUpdateMany(List> groupList, string tableName) { - if (ReSetValueBySqlExpListType == null && ReSetValueBySqlExpList != null) + var operations = new List(); + List pks = this.PrimaryKeys; + + foreach (var group in groupList) { - var result = batchUpdateSql.ToString(); - foreach (var item in ReSetValueBySqlExpList) + var filter = new BsonDocument(); + var setDoc = new BsonDocument(); + + foreach (var col in group) { - var dbColumnName = item.Value.DbColumnName; - if (item.Value.Type == ReSetValueBySqlExpListModelType.List) + + if (col.IsPrimarykey || pks.Contains(col.DbColumnName)) { - result = result.Replace($"{dbColumnName}=T.{dbColumnName}", $"{dbColumnName}={GetTableNameString}.{dbColumnName}{item.Value.Sql}T.{dbColumnName}"); + filter[col.DbColumnName] = BsonValue.Create(ObjectId.Parse(col.Value?.ToString())); ; } else { - result = result.Replace($"{dbColumnName}=T.{dbColumnName}", $"{dbColumnName}={item.Value.Sql.Replace(dbColumnName, $"{Builder.GetTranslationColumnName(this.TableName)}.{dbColumnName}")}"); + var bsonValue = BsonValue.Create(col.Value); + setDoc[col.DbColumnName] = bsonValue; } - batchUpdateSql = new StringBuilder(result); } + + var update = new BsonDocument + { + { "$set", setDoc } + }; + + var entry = new BsonDocument + { + { "filter", filter }, + { "update", update } + }; + + string json = entry.ToJson(new MongoDB.Bson.IO.JsonWriterSettings + { + OutputMode = MongoDB.Bson.IO.JsonOutputMode.Shell // JSON标准格式，带双引号 + }); + + operations.Add(json); } - return batchUpdateSql; - } - protected override string GetJoinUpdate(string columnsString, ref string whereString) - { - var formString = $" {Builder.GetTranslationColumnName(this.TableName)} AS {Builder.GetTranslationColumnName(this.ShortName)} "; - var joinString = ""; - foreach (var item in this.JoinInfos) - { - whereString += " AND "+item.JoinWhere; - joinString += $"\r\n FROM {Builder.GetTranslationColumnName(item.TableName)} {Builder.GetTranslationColumnName(item.ShortName)} "; - } - var tableName = formString + "\r\n "; - columnsString = columnsString.Replace(Builder.GetTranslationColumnName(this.ShortName)+".","")+joinString; - return string.Format(SqlTemplate, tableName, columnsString, whereString); - } - public override string FormatDateTimeOffset(object value) - { - return "'" + ((DateTimeOffset)value).ToString("o") + "'"; + var sb = new StringBuilder(); + sb.Append($"updateMany {tableName} [ "); + sb.Append(string.Join(", ", operations)); + sb.Append(" ]"); + + return sb.ToString(); } + + } } diff --git a/Src/Asp.NetCore2/SqlSugar.MongoDbCore/MongoDbBase.cs b/Src/Asp.NetCore2/SqlSugar.MongoDbCore/MongoDbBase.cs new file mode 100644 index 000000000..5e3fa19e2 --- /dev/null +++ b/Src/Asp.NetCore2/SqlSugar.MongoDbCore/MongoDbBase.cs @@ -0,0 +1,12 @@ +using System; +using System.Collections.Generic; +using System.Text; + +namespace SqlSugar.MongoDbCore +{ + public class MongoDbBase + { + [SugarColumn(IsPrimaryKey =true,IsOnlyIgnoreInsert =true,ColumnName ="_id")] + public string Id { get; set; } + } +}