lsm/SqlSugar
1
0
Fork 0
mirror of https://gitee.com/dotnetchina/SqlSugar.git synced 2025-05-03 20:27:56 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add Mongo DB

Browse Source
This commit is contained in:
sunkaixuan 2025-04-26 19:34:30 +08:00
parent 4adeb10cf2
commit 04fa1793b4
3 changed files with 126 additions and 41 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

119
Src/Asp.NetCore2/MongoDb.Ado.data/MongoDbConnection.cs
View File

@ -10,26 +10,23 @@ namespace MongoDb.Ado.data
{
public class MongoDbConnection : DbConnection
{
private readonly string _originalConnectionString;
private MongoClient _client;
private IMongoDatabase _database;
private ConnectionState _state = ConnectionState.Closed;
private string _databaseName;
private static readonly Dictionary<string, MongoClient> _clientCache = new Dictionary<string, MongoClient>(StringComparer.OrdinalIgnoreCase);
private static readonly object _lock = new object();
public override string ConnectionString
{
get => _originalConnectionString;
set => throw new NotSupportedException("Setting ConnectionString after construction is not supported.");
}
private readonly string _originalConnectionString;
private IMongoDatabase _database;
private string _databaseName;
private ConnectionState _state = ConnectionState.Closed;
public override string Database => _databaseName;
public override string DataSource => _client?.Settings?.Server?.ToString() ?? "";
public override string ServerVersion => "MongoDB_" + (_client?.Cluster?.Description?.ClusterId.ToString() ?? "Unknown");
public override ConnectionState State => _state;
public override string ConnectionString { get => throw new NotImplementedException(); set => throw new NotImplementedException(); }
private MongoClient _client;
public MongoDbConnection(string connectionString)
{
_originalConnectionString = connectionString;
@ -38,29 +35,84 @@ namespace MongoDb.Ado.data
private void ParseAndConnect(string connStr)
{
string mongoConnStr;
if (connStr.TrimStart().StartsWith("mongodb://", StringComparison.OrdinalIgnoreCase))
{
var mongoUrl = new MongoUrl(connStr);
_client = new MongoClient(mongoUrl);
_databaseName = mongoUrl.DatabaseName ?? "test";
_database = _client.GetDatabase(_databaseName);
return;
mongoConnStr = connStr;
}
else
{
string queryParams = string.Empty; // 用来存储查询参数部分
var dict = ParsePgStyleConnectionString(connStr);
var host = dict.GetValueOrDefault("Host", "localhost");
var port = dict.GetValueOrDefault("Port", "27017");
_databaseName = dict.GetValueOrDefault("Database", "test");
var username = dict.GetValueOrDefault("Username", "");
var password = dict.GetValueOrDefault("Password", "");
// 如果连接字符串以 "mongodb://" 开头
if (connStr.TrimStart().StartsWith("mongodb://", StringComparison.OrdinalIgnoreCase))
{
mongoConnStr = connStr;
var mongoConnStr = string.IsNullOrEmpty(username)
? $"mongodb://{host}:{port}"
: $"mongodb://{Uri.EscapeDataString(username)}:{Uri.EscapeDataString(password)}@{host}:{port}/{_databaseName}";
// 提取查询参数
var uri = new Uri(mongoConnStr);
var query = uri.Query;
if (!string.IsNullOrEmpty(query))
{
queryParams = query;
}
}
else
{
// 解析以 PostgreSQL 风格的连接字符串
var dict = ParsePgStyleConnectionString(connStr);
var host = dict.GetValueOrDefault("Host", "localhost");
var port = dict.GetValueOrDefault("Port", "27017");
_databaseName = dict.GetValueOrDefault("Database", "");
var username = dict.GetValueOrDefault("Username", "");
var password = dict.GetValueOrDefault("Password", "");
var mongoUrlParsed = new MongoUrl(mongoConnStr);
_client = new MongoClient(mongoUrlParsed);
_database = _client.GetDatabase(_databaseName);
mongoConnStr = string.IsNullOrEmpty(username)
? $"mongodb://{host}:{port}/{_databaseName}"
: $"mongodb://{Uri.EscapeDataString(username)}:{Uri.EscapeDataString(password)}@{host}:{port}/{_databaseName}";
// 提取查询参数（如果有）
if (dict.ContainsKey("ReplicaSet"))
{
queryParams += $"?replicaSet={dict["ReplicaSet"]}";
}
if (dict.ContainsKey("AuthSource"))
{
if (!string.IsNullOrEmpty(queryParams))
{
queryParams += "&";
}
queryParams += $"authSource={dict["AuthSource"]}";
}
}
mongoConnStr = mongoConnStr + queryParams;
_client = GetOrCreateClient(mongoConnStr);
if (_databaseName == null)
{
var mongoUrl = new MongoUrl(mongoConnStr);
_databaseName = mongoUrl.DatabaseName ?? "test";
}
_database = _client.GetDatabase(_databaseName);
}
}
private static MongoClient GetOrCreateClient(string connectionString)
{
if (_clientCache.TryGetValue(connectionString, out var client))
return client;
lock (_lock)
{
if (_clientCache.TryGetValue(connectionString, out client))
return client;
client = new MongoClient(connectionString);
_clientCache[connectionString] = client;
return client;
}
}
private Dictionary<string, string> ParsePgStyleConnectionString(string connStr)
@ -72,13 +124,14 @@ namespace MongoDb.Ado.data
public override void Open()
{
// MongoClient 实际上在操作集合时才连接，我们只改变状态
_state = ConnectionState.Open;
}
public override void Close()
{
_state = ConnectionState.Closed;
// 注意：MongoClient 不需要 Dispose，它内部自己管理连接池！
// 所以这里不用处理 _client.Dispose()，否则会出大问题
}
protected override DbCommand CreateDbCommand()
@ -97,13 +150,13 @@ namespace MongoDb.Ado.data
protected override DbTransaction BeginDbTransaction(IsolationLevel isolationLevel)
{
throw new NotSupportedException("MongoDB does not support transactions via DbTransaction.");
return null;
}
public IMongoDatabase GetDatabase() => _database;
public MongoClient GetClient() => _client;
public override string ToString() => _originalConnectionString;
}
}

33
Src/Asp.NetCore2/MongoDbTest/AdoTest.cs
View File

@ -12,17 +12,34 @@ namespace MongoDbTest
public class AdoTest
{
public static void Init()
{
MongoClientTest();
MongoDbConnectionTest();
}
private static void MongoDbConnectionTest()
{
var db= new MongoDbConnection(DbHelper.SqlSugarConnectionString);
var database= db.GetDatabase();
var collections = database.GetCollection<BsonDocument>("b");
// 插入一个文档，MongoDB 会创建数据库和集合
var document = new BsonDocument { { "name", "bbbbbb" }, { "age", 30 } };
collections.InsertOne(document);
var list = collections.AsQueryable<BsonDocument>().ToList();
}
private static void MongoClientTest()
{
//开发中
var client = new MongoClient("");
var database = client.GetDatabase("test");
var client = new MongoClient(DbHelper.ConnectionString);
var database = client.GetDatabase("SqlSugarDb");
// 获取当前数据库中的所有集合
var collections = database.ListCollections();
MongoDbConnection connection = new MongoDbConnection("");
connection.Open();
connection.Close();
var collections = database.GetCollection<BsonDocument>("a");
// 插入一个文档，MongoDB 会创建数据库和集合
var document = new BsonDocument { { "name", "aaaa" }, { "age", 30 } };
collections.InsertOne(document);
var list = collections.AsQueryable<BsonDocument>().ToList();
}
}
}

15
Src/Asp.NetCore2/MongoDbTest/DbHelper.cs Normal file
View File

@ -0,0 +1,15 @@
using MongoDB.Driver.Core.Configuration;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
namespace MongoDbTest
{
public class DbHelper
{
public static string ConnectionString= "mongodb://mongouser:Huangxin%40123@localhost:27018/SqlSugarDb?replicaSet=cmgo-7d07e4w1_0&authSource=admin";
public static string SqlSugarConnectionString = "host=localhost;Port=27018;Database=SqlSugarDb;Username=mongouser;Password=Huangxin@123;replicaSet=cmgo-7d07e4w1_0&authSource=admin";
}
}