lsm/SqlSugar
1
0
Fork 0
mirror of https://gitee.com/dotnetchina/SqlSugar.git synced 2025-09-22 20:13:41 +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
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -10,26 +10,23 @@ namespace MongoDb.Ado.data
{
public class MongoDbConnection : DbConnection
{
private static readonly Dictionary<string, MongoClient> _clientCache = new Dictionary<string, MongoClient>(StringComparer.OrdinalIgnoreCase);
private static readonly object _lock = new object();
private readonly string _originalConnectionString;
private MongoClient _client;
private IMongoDatabase _database;
private ConnectionState _state = ConnectionState.Closed;
private string _databaseName;
public override string ConnectionString
{
get => _originalConnectionString;
set => throw new NotSupportedException("Setting ConnectionString after construction is not supported.");
}
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;
}
}