lsm/SqlSugar
1
0
Fork 0
mirror of https://gitee.com/dotnetchina/SqlSugar.git synced 2025-06-28 13:34:32 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Upddate mongodb

Browse Source
This commit is contained in:
sunkaixuan 2025-06-10 17:46:57 +08:00
parent 6afa9f7539
commit b41076f0b0
12 changed files with 34 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
Src/Asp.NetCore2/MongoDb.Ado.data/ExecuteDbDataReaderItemsAsync/DbDataReaderFactoryAsync.cs
View File

@ -7,6 +7,7 @@ using System.Data;
using System.Data.Common;
using System.Net.Http.Headers;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
namespace MongoDb.Ado.data
@ -18,16 +19,16 @@ namespace MongoDb.Ado.data
{ "find", new QueryFindHandlerAsync() },
{ "aggregate", new QueryAggregateHandlerAsync() },
};
public async Task<DbDataReader> HandleAsync(string operation, IMongoCollection<BsonDocument> collection, string json)
public async Task<DbDataReader> HandleAsync(string operation, IMongoCollection<BsonDocument> collection, string json, CancellationToken cancellationToken)
{
MongoDbMethodUtils.ValidateOperation(operation);
var doc = MongoDB.Bson.Serialization.BsonSerializer.Deserialize<BsonValue>(json);
DbDataReaderFactoryAsync.Items.TryGetValue(operation, out var handler);
if (handler == null)
{
await ExecuteHandlerFactoryAsync.HandlerAsync(operation, json, collection);
await ExecuteHandlerFactoryAsync.HandlerAsync(operation, json, collection, cancellationToken);
return new DataTable().CreateDataReader();
}
}
return await handler.HandlerAsync(collection, doc);
}

4
Src/Asp.NetCore2/MongoDb.Ado.data/ExecuteNonQueryItemsAsync/DeleteHandlerAsync.cs
View File

@ -3,18 +3,20 @@ using MongoDB.Driver;
using System;
using System.Collections.Generic;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
namespace MongoDb.Ado.data
{
public class DeleteHandlerAsync : IMongoOperationHandlerAsync
{
public CancellationToken token { get; set; }
public string operation { get; set; }
public async Task<int> HandleAsync(IMongoCollection<BsonDocument> collection, string json)
{
var doc = BsonDocument.Parse(json);
var filter = doc["filter"].AsBsonDocument;
var result =await collection.DeleteOneAsync(filter);
var result =await collection.DeleteOneAsync(filter,token);
return (int)result.DeletedCount;
}
}

4
Src/Asp.NetCore2/MongoDb.Ado.data/ExecuteNonQueryItemsAsync/DeleteManyHandlerAsync.cs
View File

@ -5,11 +5,13 @@ using System;
using System.Collections.Generic;
using System.Text;
using System.Threading.Tasks;
using System.Threading;
namespace MongoDb.Ado.data
{
public class DeleteManyHandlerAsync : IMongoOperationHandlerAsync
{
public CancellationToken token { get; set; }
public string operation { get; set; }
public async Task<int> HandleAsync(IMongoCollection<BsonDocument> collection, string json)
{
@ -18,7 +20,7 @@ namespace MongoDb.Ado.data
foreach (var doc in documents)
{
var filter = doc["filter"].AsBsonDocument;
var result =await collection.DeleteManyAsync(filter);
var result =await collection.DeleteManyAsync(filter,token);
total += (int)result.DeletedCount;
}
return total;

4
Src/Asp.NetCore2/MongoDb.Ado.data/ExecuteNonQueryItemsAsync/ExecuteHandlerFactoryAsync.cs
View File

@ -4,6 +4,7 @@ using MongoDB.Driver;
using System;
using System.Collections.Generic;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
namespace MongoDb.Ado.data
@ -22,7 +23,7 @@ namespace MongoDb.Ado.data
};
public static Task<int> HandlerAsync(string operation, string json, IMongoCollection<BsonDocument> collection)
public static Task<int> HandlerAsync(string operation, string json, IMongoCollection<BsonDocument> collection,CancellationToken cancellationToken)
{
MongoDbMethodUtils.ValidateOperation(operation);
var handlers = ExecuteHandlerFactoryAsync.Items;
@ -30,6 +31,7 @@ namespace MongoDb.Ado.data
if (!handlers.TryGetValue(operation, out var handler))
throw new NotSupportedException($"不支持的操作类型: {operation}");
handler.operation = operation;
handler.token = cancellationToken;
return handler.HandleAsync(collection, json);
}

3
Src/Asp.NetCore2/MongoDb.Ado.data/ExecuteNonQueryItemsAsync/IMongoOperationHandler.cs
View File

@ -3,6 +3,7 @@ using MongoDB.Driver;
using System;
using System.Collections.Generic;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
namespace MongoDb.Ado.data
@ -10,6 +11,8 @@ namespace MongoDb.Ado.data
public interface IMongoOperationHandlerAsync
{
string operation { get; set; }
CancellationToken token { get; set; }
Task<int> HandleAsync(IMongoCollection<BsonDocument> collection, string json);
}
}

2
Src/Asp.NetCore2/MongoDb.Ado.data/ExecuteNonQueryItemsAsync/InsertHandlerAsync.cs
View File

@ -3,12 +3,14 @@ using MongoDB.Driver;
using System;
using System.Collections.Generic;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
namespace MongoDb.Ado.data
{
public class InsertHandlerAsync : IMongoOperationHandlerAsync
{
public CancellationToken token { get; set; }
public string operation { get; set; }
public async Task<int> HandleAsync(IMongoCollection<BsonDocument> collection, string json)
{

2
Src/Asp.NetCore2/MongoDb.Ado.data/ExecuteNonQueryItemsAsync/InsertManyHandlerAsync.cs
View File

@ -5,11 +5,13 @@ using System;
using System.Collections.Generic;
using System.Text;
using System.Threading.Tasks;
using System.Threading;
namespace MongoDb.Ado.data
{
public class InsertManyHandlerAsync : IMongoOperationHandlerAsync
{
public CancellationToken token { get; set; }
public string operation { get; set; }
public async Task<int> HandleAsync(IMongoCollection<BsonDocument> collection, string json)
{

4
Src/Asp.NetCore2/MongoDb.Ado.data/ExecuteNonQueryItemsAsync/NonFindHandlerAsync.cs
View File

@ -3,17 +3,19 @@ using MongoDB.Driver;
using System;
using System.Collections.Generic;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
namespace MongoDb.Ado.data
{
public class NonFindHandlerAsync : IMongoOperationHandlerAsync
{
public CancellationToken token { get; set; }
public string operation { get; set; }
public async Task<int> HandleAsync(IMongoCollection<BsonDocument> collection, string json)
{
using (var dr = await new DbDataReaderFactoryAsync().HandleAsync(operation, collection, json))
using (var dr = await new DbDataReaderFactoryAsync().HandleAsync(operation, collection, json,token))
{
if (dr.Read())
{

4
Src/Asp.NetCore2/MongoDb.Ado.data/ExecuteNonQueryItemsAsync/UpdateHandler.cs
View File

@ -3,19 +3,21 @@ using MongoDB.Driver;
using System;
using System.Collections.Generic;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
namespace MongoDb.Ado.data
{
public class UpdateHandlerAsync : IMongoOperationHandlerAsync
{
public CancellationToken token { get; set; }
public string operation { get; set; }
public async Task<int> HandleAsync(IMongoCollection<BsonDocument> collection, string json)
{
var doc = BsonDocument.Parse(json);
var filter = doc["filter"].AsBsonDocument;
var update = doc["update"].AsBsonDocument;
var result =await collection.UpdateOneAsync(filter, update);
var result =await collection.UpdateOneAsync(filter, update,null,token);
return (int)result.ModifiedCount;
}
}

2
Src/Asp.NetCore2/MongoDb.Ado.data/ExecuteNonQueryItemsAsync/UpdateManyHandler.cs
View File

@ -5,11 +5,13 @@ using System;
using System.Collections.Generic;
using System.Text;
using System.Threading.Tasks;
using System.Threading;
namespace MongoDb.Ado.data
{
public class UpdateManyHandlerAsync : IMongoOperationHandlerAsync
{
public CancellationToken token { get; set; }
public string operation { get; set; }
public async Task<int> HandleAsync(IMongoCollection<BsonDocument> collection, string json)
{

5
Src/Asp.NetCore2/MongoDb.Ado.data/ExecuteScalarItemsAsync/ExecuteScalarHandlerAsync.cs
View File

@ -4,15 +4,16 @@ using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
namespace MongoDb.Ado.data
{
public class ExecuteScalarHandlerAsync
{
public async Task<object> HandleAsync(string operation, IMongoCollection<BsonDocument> collection, string json)
public async Task<object> HandleAsync(string operation, IMongoCollection<BsonDocument> collection, string json,CancellationToken cancellationToken)
{
using (var dbReader = await new DbDataReaderFactoryAsync().HandleAsync(operation, collection, json))
using (var dbReader = await new DbDataReaderFactoryAsync().HandleAsync(operation, collection, json,cancellationToken))
{
if (dbReader.Read())
{

6
Src/Asp.NetCore2/MongoDb.Ado.data/MongoDbCommand.cs
View File

@ -83,19 +83,19 @@ namespace MongoDb.Ado.data
{
var (operation, collectionName, json) = ParseCommand(_commandText);
var collection = GetCollection(collectionName);
return ExecuteHandlerFactoryAsync.HandlerAsync(operation, json, collection);
return ExecuteHandlerFactoryAsync.HandlerAsync(operation, json, collection, cancellationToken);
}
public override Task<object> ExecuteScalarAsync(CancellationToken cancellationToken)
{
var (operation, collectionName, json) = ParseCommand(_commandText);
var collection = GetCollection(collectionName);
return new ExecuteScalarHandlerAsync().HandleAsync(operation, collection, json);
return new ExecuteScalarHandlerAsync().HandleAsync(operation, collection, json, cancellationToken);
}
protected override Task<DbDataReader> ExecuteDbDataReaderAsync(CommandBehavior behavior,CancellationToken cancellationToken)
{
var (operation, collectionName, json) = ParseCommand(_commandText);
var collection = GetCollection(collectionName);
return new DbDataReaderFactoryAsync().HandleAsync(operation, collection, json);
return new DbDataReaderFactoryAsync().HandleAsync(operation, collection, json, cancellationToken);
}