Esempio n. 1
0
        /// <summary>获取Storage连接
        /// </summary>
        public IConnection GetStorageConnection(ConnectionAddress connectionAddress)
        {
            if (!_storageConnectionPools.TryGetValue(connectionAddress, out IConnectionPool connectionPool))
            {
                lock (_storageSyncObject)
                {
                    if (!_storageConnectionPools.TryGetValue(connectionAddress, out connectionPool))
                    {
                        var connectionPoolOption = new ConnectionPoolOption()
                        {
                            ConnectionAddress          = connectionAddress,
                            ConnectionLifeTime         = _option.ConnectionLifeTime,
                            ConnectionConcurrentThread = _option.ConnectionConcurrentThread,
                            MaxConnection = _option.StorageMaxConnection,
                            ScanTimeoutConnectionInterval = _option.ScanTimeoutConnectionInterval
                        };

                        connectionPool = _connectionPoolFactory.CreateConnectionPool(connectionPoolOption);
                        if (!_storageConnectionPools.TryAdd(connectionAddress, connectionPool))
                        {
                            _logger.LogWarning("Fail to add connection pool to storage connection pools! ConnectionAddress:{0}", connectionAddress);
                        }
                    }
                }
            }

            if (connectionPool == null)
            {
                throw new ArgumentException($"Can't find any connection pools for {connectionAddress}");
            }
            return(connectionPool.GetConnection());
        }
Esempio n. 2
0
        /// <summary>Get tracker connection
        /// </summary>
        public IConnection GetTrackerConnection()
        {
            var rd                = new Random();
            var index             = rd.Next(_trackerConnectionAddresses.Count);
            var connectionAddress = _trackerConnectionAddresses[index];

            if (!_trackerConnectionPools.TryGetValue(connectionAddress, out IConnectionPool connectionPool))
            {
                lock (_trackerSyncObject)
                {
                    if (!_trackerConnectionPools.TryGetValue(connectionAddress, out connectionPool))
                    {
                        var connectionPoolOption = new ConnectionPoolOption()
                        {
                            ConnectionAddress          = connectionAddress,
                            ConnectionLifeTime         = _option.ConnectionLifeTime,
                            ConnectionConcurrentThread = _option.ConnectionConcurrentThread,
                            MaxConnection = _option.TrackerMaxConnection,
                            ScanTimeoutConnectionInterval = _option.ScanTimeoutConnectionInterval
                        };

                        connectionPool = _connectionPoolFactory.CreateConnectionPool(connectionPoolOption);
                        if (!_trackerConnectionPools.TryAdd(connectionAddress, connectionPool))
                        {
                            _logger.LogWarning("Fail to add connection pool to tracker connection pools! ConnectionAddress:{0}", connectionAddress);
                        }
                    }
                }
            }

            return(connectionPool.GetConnection());
        }
        public IConnectionPool CreateConnectionPool(ConnectionPoolOption option)
        {
            using (var scope = _serviceProvider.CreateScope())
            {
                var injectOption = scope.ServiceProvider.GetService <ConnectionPoolOption>();
                injectOption.ConnectionAddress             = option.ConnectionAddress;
                injectOption.MaxConnection                 = option.MaxConnection;
                injectOption.ConnectionConcurrentThread    = option.ConnectionConcurrentThread;
                injectOption.ConnectionLifeTime            = option.ConnectionLifeTime;
                injectOption.ScanTimeoutConnectionInterval = option.ScanTimeoutConnectionInterval;

                return(scope.ServiceProvider.GetService <IConnectionPool>());
            }
        }
        public DefaultConnectionPool(ILogger <DefaultConnectionPool> logger, ConnectionPoolOption option, IConnectionBuilder connectionFactory)
        {
            _logger            = logger;
            _option            = option;
            _connectionFactory = connectionFactory;

            Name             = Guid.NewGuid().ToString();
            _connectionCount = 0;
            _cts             = new CancellationTokenSource();

            _maxConnectionCount = _option.MaxConnection;
            _semaphoreSlim      = new SemaphoreSlim(_option.ConnectionConcurrentThread);
            _connectionStack    = new ConcurrentStack <IConnection>();
            _connectionDict     = new ConcurrentDictionary <string, IConnection>();
        }