src/ZKEACMS/DbConnectionPool/SimpleDbConnectionPool.cs

这里只显示每个文件的开头 60 行。登录后可以解锁完整代码。

微信扫码关注公众号，代码整理好后第一时间通知你。

/* http://www.zkea.net/ 
 * Copyright (c) ZKEASOFT. All rights reserved. 
 * http://www.zkea.net/licenses */

using Easy;
using Microsoft.Data.Sqlite;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.ObjectPool;
using System;
using System.Data.Common;
using System.Data.SqlClient;
using ZKEACMS.Options;

namespace ZKEACMS.DbConnectionPool
{
    /// <summary>
    /// 仅仅对单独一个数据库提供的连接池，对多个数据库的情形不支持
    /// </summary>
    public class SimpleDbConnectionPool : IDisposable, IDbConnectionPool
    {
        private class PooledObjectPolicy : PooledObjectPolicy<DbConnection>
        {
            internal PooledObjectPolicy(int maximumRetained, Func<DbConnection> funcCreateConnection)
            {
                _maximumRetained = maximumRetained;
                _createdObjects = new DbConnection[_maximumRetained];
                _funcCreateConnection = funcCreateConnection;
            }
            internal readonly DbConnection[] _createdObjects;
            private readonly int _maximumRetained;
            private readonly Func<DbConnection> _funcCreateConnection;
            private int _numObjectsInPool;

            public override DbConnection Create()
            {
                int cnt = System.Threading.Interlocked.Increment(ref _numObjectsInPool);
                if (cnt > _maximumRetained)
                {
                    System.Threading.Interlocked.Decrement(ref _numObjectsInPool);
                    return null;
                }
                DbConnection dbConnection = _funcCreateConnection();
                _createdObjects[cnt - 1] = dbConnection;
                return dbConnection;
            }
            public override bool Return(DbConnection obj)
            {
                return true;
            }
        }

        private readonly PooledObjectPolicy _pooledObjectPolicy;

        public SimpleDbConnectionPool(Options options, DatabaseOption databaseOption)
        {
            DatabaseOption = databaseOption;
            int maximumRetained = Math.Max(options.MaximumRetained, 0);
            if (maximumRetained > 0)
            {
                _pooledObjectPolicy = new PooledObjectPolicy(maximumRetained, CreateDbConnection);