mirror of
https://github.com/nsnail/FreeSql.git
synced 2025-05-02 23:22:51 +08:00
315 lines
10 KiB
C#
315 lines
10 KiB
C#
using FreeSql.Internal.ObjectPool;
|
||
using System;
|
||
using System.Collections.Concurrent;
|
||
using System.Collections.Generic;
|
||
using System.Data;
|
||
using System.Data.Common;
|
||
#if MicrosoftData
|
||
using Microsoft.Data.Sqlite;
|
||
#else
|
||
using System.Data.SQLite;
|
||
#endif
|
||
using System.Linq;
|
||
using System.Text;
|
||
using System.Text.RegularExpressions;
|
||
using System.Threading.Tasks;
|
||
|
||
namespace FreeSql.Sqlite
|
||
{
|
||
|
||
class SqliteConnectionPool : ObjectPool<DbConnection>
|
||
{
|
||
|
||
internal Action availableHandler;
|
||
internal Action unavailableHandler;
|
||
|
||
public SqliteConnectionPool(string name, string connectionString, Action availableHandler, Action unavailableHandler) : base(null)
|
||
{
|
||
this.availableHandler = availableHandler;
|
||
this.unavailableHandler = unavailableHandler;
|
||
policy = new SqliteConnectionPoolPolicy
|
||
{
|
||
_pool = this,
|
||
Name = name
|
||
};
|
||
this.Policy = policy;
|
||
policy.ConnectionString = connectionString;
|
||
}
|
||
|
||
public void Return(Object<DbConnection> obj, Exception exception, bool isRecreate = false)
|
||
{
|
||
#if MicrosoftData
|
||
if (exception != null && exception is SqliteException)
|
||
#else
|
||
if (exception != null && exception is SQLiteException)
|
||
#endif
|
||
{
|
||
try { if (obj.Value.Ping() == false) obj.Value.OpenAndAttach(policy.Attaches); } catch { base.SetUnavailable(exception); }
|
||
}
|
||
base.Return(obj, isRecreate);
|
||
}
|
||
|
||
internal SqliteConnectionPoolPolicy policy;
|
||
}
|
||
|
||
class SqliteConnectionPoolPolicy : IPolicy<DbConnection>
|
||
{
|
||
|
||
internal SqliteConnectionPool _pool;
|
||
public string Name { get; set; } = "Sqlite SQLiteConnection 对象池";
|
||
public int PoolSize { get; set; } = 100;
|
||
public TimeSpan SyncGetTimeout { get; set; } = TimeSpan.FromSeconds(10);
|
||
public TimeSpan IdleTimeout { get; set; } = TimeSpan.Zero;
|
||
public int AsyncGetCapacity { get; set; } = 10000;
|
||
public bool IsThrowGetTimeoutException { get; set; } = true;
|
||
public bool IsAutoDisposeWithSystem { get; set; } = true;
|
||
public int CheckAvailableInterval { get; set; } = 5;
|
||
public int Weight { get; set; } = 1;
|
||
public string[] Attaches = new string[0];
|
||
|
||
private string _connectionString;
|
||
public string ConnectionString
|
||
{
|
||
get => _connectionString;
|
||
set
|
||
{
|
||
_connectionString = value ?? "";
|
||
|
||
var pattern = @"Max\s*pool\s*size\s*=\s*(\d+)";
|
||
Match m = Regex.Match(_connectionString, pattern, RegexOptions.IgnoreCase);
|
||
if (m.Success)
|
||
{
|
||
PoolSize = int.Parse(m.Groups[1].Value);
|
||
_connectionString = Regex.Replace(_connectionString, pattern, "", RegexOptions.IgnoreCase);
|
||
}
|
||
|
||
pattern = @"Connection\s*LifeTime\s*=\s*(\d+)";
|
||
m = Regex.Match(_connectionString, pattern, RegexOptions.IgnoreCase);
|
||
if (m.Success)
|
||
{
|
||
IdleTimeout = TimeSpan.FromSeconds(int.Parse(m.Groups[1].Value));
|
||
_connectionString = Regex.Replace(_connectionString, pattern, "", RegexOptions.IgnoreCase);
|
||
}
|
||
|
||
var minPoolSize = 0;
|
||
pattern = @"Min\s*pool\s*size\s*=\s*(\d+)";
|
||
m = Regex.Match(_connectionString, pattern, RegexOptions.IgnoreCase);
|
||
if (m.Success)
|
||
{
|
||
minPoolSize = int.Parse(m.Groups[1].Value);
|
||
_connectionString = Regex.Replace(_connectionString, pattern, "", RegexOptions.IgnoreCase);
|
||
}
|
||
|
||
var att = Regex.Split(_connectionString, @"Pooling\s*=\s*", RegexOptions.IgnoreCase);
|
||
if (att.Length == 2)
|
||
{
|
||
var idx = att[1].IndexOf(';');
|
||
_connectionString = string.Concat(att[0], idx == -1 ? "" : att[1].Substring(idx));
|
||
}
|
||
|
||
att = Regex.Split(_connectionString, @"Attachs\s*=\s*", RegexOptions.IgnoreCase);
|
||
if (att.Length == 2)
|
||
{
|
||
var idx = att[1].IndexOf(';');
|
||
Attaches = (idx == -1 ? att[1] : att[1].Substring(0, idx)).Split(',');
|
||
_connectionString = string.Concat(att[0], idx == -1 ? "" : att[1].Substring(idx));
|
||
}
|
||
|
||
if (_connectionString.ToLower().Contains(":memory:"))
|
||
{
|
||
//内存模式
|
||
PoolSize = 1;
|
||
}
|
||
|
||
#if ns20
|
||
minPoolSize = 1;
|
||
#endif
|
||
FreeSql.Internal.CommonUtils.PrevReheatConnectionPool(_pool, minPoolSize);
|
||
|
||
}
|
||
}
|
||
|
||
public bool OnCheckAvailable(Object<DbConnection> obj)
|
||
{
|
||
if (obj.Value.State == ConnectionState.Closed) obj.Value.OpenAndAttach(Attaches);
|
||
return obj.Value.Ping(true);
|
||
}
|
||
|
||
public DbConnection OnCreate()
|
||
{
|
||
#if MicrosoftData
|
||
var conn = new SqliteConnection(_connectionString);
|
||
#else
|
||
var conn = new SQLiteConnection(_connectionString);
|
||
#endif
|
||
return conn;
|
||
}
|
||
|
||
public void OnDestroy(DbConnection obj)
|
||
{
|
||
if (obj.State != ConnectionState.Closed) obj.Close();
|
||
obj.Dispose();
|
||
}
|
||
|
||
public void OnGet(Object<DbConnection> obj)
|
||
{
|
||
|
||
if (_pool.IsAvailable)
|
||
{
|
||
if (obj.Value == null)
|
||
{
|
||
if (_pool.SetUnavailable(new Exception("连接字符串错误，或者检查项目属性 > 生成 > 目标平台：x86 | x64")) == true)
|
||
throw new Exception($"【{this.Name}】连接字符串错误，请检查。或者检查项目属性 > 生成 > 目标平台：x86 | x64");
|
||
return;
|
||
}
|
||
|
||
if (obj.Value.State != ConnectionState.Open || DateTime.Now.Subtract(obj.LastReturnTime).TotalSeconds > 60 && obj.Value.Ping() == false)
|
||
{
|
||
|
||
try
|
||
{
|
||
obj.Value.OpenAndAttach(Attaches);
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
if (_pool.SetUnavailable(ex) == true)
|
||
throw new Exception($"【{this.Name}】Block access and wait for recovery: {ex.Message}");
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
#if net40
|
||
#else
|
||
async public Task OnGetAsync(Object<DbConnection> obj)
|
||
{
|
||
|
||
if (_pool.IsAvailable)
|
||
{
|
||
if (obj.Value == null)
|
||
{
|
||
if (_pool.SetUnavailable(new Exception("连接字符串错误")) == true)
|
||
throw new Exception($"【{this.Name}】连接字符串错误，请检查。");
|
||
return;
|
||
}
|
||
|
||
if (obj.Value.State != ConnectionState.Open || DateTime.Now.Subtract(obj.LastReturnTime).TotalSeconds > 60 && (await obj.Value.PingAsync()) == false)
|
||
{
|
||
|
||
try
|
||
{
|
||
await obj.Value.OpenAndAttachAsync(Attaches);
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
if (_pool.SetUnavailable(ex) == true)
|
||
throw new Exception($"【{this.Name}】Block access and wait for recovery: {ex.Message}");
|
||
}
|
||
}
|
||
}
|
||
}
|
||
#endif
|
||
|
||
public void OnGetTimeout()
|
||
{
|
||
|
||
}
|
||
|
||
public void OnReturn(Object<DbConnection> obj)
|
||
{
|
||
//if (obj?.Value != null && obj.Value.State != ConnectionState.Closed) try { obj.Value.Close(); } catch { }
|
||
}
|
||
|
||
public void OnAvailable()
|
||
{
|
||
_pool.availableHandler?.Invoke();
|
||
}
|
||
|
||
public void OnUnavailable()
|
||
{
|
||
_pool.unavailableHandler?.Invoke();
|
||
}
|
||
}
|
||
static class DbConnectionExtensions
|
||
{
|
||
|
||
static DbCommand PingCommand(DbConnection conn)
|
||
{
|
||
var cmd = conn.CreateCommand();
|
||
cmd.CommandTimeout = 5;
|
||
cmd.CommandText = "select 1";
|
||
return cmd;
|
||
}
|
||
public static bool Ping(this DbConnection that, bool isThrow = false)
|
||
{
|
||
try
|
||
{
|
||
using (var cmd = PingCommand(that))
|
||
{
|
||
cmd.ExecuteNonQuery();
|
||
}
|
||
return true;
|
||
}
|
||
catch
|
||
{
|
||
if (that.State != ConnectionState.Closed) try { that.Close(); } catch { }
|
||
if (isThrow) throw;
|
||
return false;
|
||
}
|
||
}
|
||
public static void OpenAndAttach(this DbConnection that, string[] attach)
|
||
{
|
||
that.Open();
|
||
|
||
if (attach?.Any() == true)
|
||
{
|
||
var sb = new StringBuilder();
|
||
foreach (var att in attach)
|
||
sb.Append($"attach database [{att}] as [{att.Split('/', '\\').Last().Split('.').First()}];\r\n");
|
||
|
||
var cmd = that.CreateCommand();
|
||
cmd.CommandText = sb.ToString();
|
||
cmd.ExecuteNonQuery();
|
||
cmd.Dispose();
|
||
}
|
||
}
|
||
|
||
#if net40
|
||
#else
|
||
async public static Task<bool> PingAsync(this DbConnection that, bool isThrow = false)
|
||
{
|
||
try
|
||
{
|
||
using (var cmd = PingCommand(that))
|
||
{
|
||
await cmd.ExecuteNonQueryAsync();
|
||
}
|
||
return true;
|
||
}
|
||
catch
|
||
{
|
||
if (that.State != ConnectionState.Closed) try { that.Close(); } catch { }
|
||
if (isThrow) throw;
|
||
return false;
|
||
}
|
||
}
|
||
async public static Task OpenAndAttachAsync(this DbConnection that, string[] attach)
|
||
{
|
||
await that.OpenAsync();
|
||
|
||
if (attach?.Any() == true)
|
||
{
|
||
var sb = new StringBuilder();
|
||
foreach (var att in attach)
|
||
sb.Append($"attach database [{att}] as [{att.Split('/', '\\').Last().Split('.').First()}];\r\n");
|
||
|
||
var cmd = that.CreateCommand();
|
||
cmd.CommandText = sb.ToString();
|
||
await cmd.ExecuteNonQueryAsync();
|
||
cmd.Dispose();
|
||
}
|
||
}
|
||
#endif
|
||
}
|
||
}
|