/// <summary>
/// 创建一个连接，并压入队列
/// </summary>
/// <param name="pollCapacity"></param>
/// <param name="isForceCreate"></param>
/// <returns></returns>
private INTCPConnection CreatOneConnectionToPool(bool polling, int pollCapacity, bool isForceCreate = false)
{
INTCPConnection driver = null;//SingleConnectionCable
if (isForceCreate == false && _hasInitDriverCount > this._config.PoolingMaxSize)
{
return(driver);
//throw new NTcpException("超过最大连接池设置的数目!", ErrorCode.OverPoolingSize);
}
string address = _config.Address;
int port = _config.Port;
lock (_locker)
{
driver = CreatNewConnection(address, port, polling, pollCapacity);
if (null != driver)
{
//尝试打开驱动连接
driver.Connect(_config.TimeOut * 1000, true);
}
_driverQueue.Enqueue(driver);
if (isForceCreate == false)
{
_hasInitDriverCount += 1;
}
}
return(driver);
}
/// <summary>
/// 释放连接到池子
/// </summary>
/// <param name="driver"></param>
public void ReleaseToPool(INTCPConnection driver)
{
if (null != driver)
{
this._driverQueue.Enqueue(driver);
autoEvent.Set();
}
}
/// <summary>
/// 创建新的连接对象
/// </summary>
/// <param name="address"></param>
/// <param name="port"></param>
/// <param name="isPolling"></param>
/// <param name="pollCapacity"></param>
/// <returns></returns>
public static INTCPConnection CreatNewConnection(string address, int port, bool isPolling, int pollCapacity)
{
INTCPConnection driver = null;
var remoteEndPoint = new IPEndPoint(IPAddress.Parse(address), port);
if (isPolling == true)
{
//基于连接池的多窗口连接
driver = new SingleConnectionCable(remoteEndPoint, pollCapacity);
}
else
{
//单连接
driver = new SingleConnection(remoteEndPoint);
}
return(driver);
}
/// <summary>
/// 关闭连接
/// </summary>
public void Close()
{
if (this.State != ConnectionState.Closed && driver != null)
{
//扔回连接池中
var currentSettings = new WebCrawlerConnection {
Address = this.IPAddress, Port = this.Port
};
SoapTcpPool pool = SoapTcpPool.GetPool(currentSettings);
if (null != pool)
{
pool.ReleaseToPool(this.driver);
this.driver = null;
}
else
{
this.driver.Close();//如果没有在池中 那么直接关闭对象
}
}
}
/// <summary>
/// 打开连接
/// </summary>
public void Open()
{
if (null != this.driver && State == ConnectionState.Open)
{
return;
}
try
{
var currentSettings = new WebCrawlerConnection {
Address = this.IPAddress,
Port = this.Port,
TimeOut = this.TimeOut,
Pooling = this.Pooling,
PoolingMinSize = this.PoolingMinSize,
PoolingMaxSize = this.PoolingMaxSize,
};
SoapTcpPool pool = SoapTcpPool.GetPool(currentSettings);
if (null != pool)
{
driver = pool.GetConnection();
}
if (driver == null)
{
driver = SoapTcpPool.CreatNewConnection(this.IPAddress, this.Port, this.Pooling, this.PoolingMinSize);
}
if (!driver.Connected)
{
driver.Connect(this.TimeOut * 1000);
}
this.State = ConnectionState.Open;
}
catch (Exception ex)
{
this.State = ConnectionState.Closed;
throw ex;
}
}
/// <summary>
/// 从连接池 获取一个连接对象
/// </summary>
/// <returns></returns>
public INTCPConnection GetConnection()
{
INTCPConnection driver = null;
if (null == _driverQueue)
{
throw new Exception("未能正确初始化连接池！请检查配置！");
}
//检测是否为空
if (_driverQueue.IsEmpty)
{
this.CreatOneConnectionToPool(this._config.Pooling, this._config.PoolingMinSize);
}
try
{
int fullTimeOut = _config.TimeOut * 1000;
int timeOut = fullTimeOut;
DateTime start = DateTime.Now;
while (timeOut > 0)
{
this._driverQueue.TryDequeue(out driver);
if (driver != null)
{
//等待可用的连接
if (driver.Connected == true)
{
autoEvent.Set();
break;
}
}
// We have no tickets right now, lets wait for one.
if (!autoEvent.WaitOne(timeOut, false))
{
break;
}
timeOut = fullTimeOut - (int)DateTime.Now.Subtract(start).TotalMilliseconds;
}
//如果未能从池中检索出可用的连接，那么尝试创建新的连接
if (null == driver)
{
var driverTryNew = CreatOneConnectionToPool(this._config.Pooling, this._config.PoolingMinSize);
if (null == driverTryNew)
{
throw new Exception("连接池最大连接池数目已经被消耗完毕！未能正确获取连接对象！请及时关闭连接！");
}
return(driverTryNew);
}
}
catch (NTcpException tcpEx)
{
//一旦内部故障 失败 ，捕获异常
if (tcpEx.Code == ErrorCode.Disconnected && null != driver)
{
this._hasInitDriverCount -= 1;//打开阈值开关 将当前驱动的引用 置为最新的连接实例
driver = CreatOneConnectionToPool(this._config.Pooling, this._config.PoolingMinSize);
}
}
catch (Exception ex)
{
throw ex;
}
return(driver);
}
