mirror of
https://gitee.com/dotnetchina/SqlSugar.git
synced 2025-07-16 07:57:33 +08:00
46 lines
1.4 KiB
C#
46 lines
1.4 KiB
C#
using MongoDB.Bson;
|
|
using MongoDB.Driver;
|
|
using System;
|
|
using System.Collections;
|
|
using System.Collections.Generic;
|
|
using System.Data.Common;
|
|
using System.Text;
|
|
|
|
namespace MongoDb.Ado.data
|
|
{
|
|
public class QueryFindHandler : IQueryHandler
|
|
{
|
|
public HandlerContext Context { get; set; }
|
|
public DbDataReader Handler(IMongoCollection<BsonDocument> collection, BsonValue doc)
|
|
{
|
|
BsonDocument filter;
|
|
BsonDocument projection = null;
|
|
|
|
if (doc.IsBsonArray)
|
|
{
|
|
var array = doc.AsBsonArray;
|
|
filter = array.Count > 0 ? array[0].AsBsonDocument : new BsonDocument();
|
|
if (array.Count > 1)
|
|
projection = array[1].AsBsonDocument;
|
|
}
|
|
else if (doc.IsBsonDocument)
|
|
{
|
|
filter = doc.AsBsonDocument;
|
|
}
|
|
else
|
|
{
|
|
throw new ArgumentException("Invalid JSON format for MongoDB find operation.");
|
|
}
|
|
|
|
var findFluent =Context?.IsAnyServerSession==true? collection.Find(Context.ServerSession,filter): collection.Find(filter);
|
|
|
|
if (projection != null)
|
|
findFluent = findFluent.Project<BsonDocument>(projection);
|
|
|
|
var cursor = findFluent.ToList();
|
|
var result = MongoDbDataReaderHelper.ToDataReader(cursor);
|
|
return result;
|
|
}
|
|
}
|
|
}
|