146 lines
2.4 KiB
PHP
146 lines
2.4 KiB
PHP
<?php
|
|
|
|
|
|
namespace Snowflake\Abstracts;
|
|
|
|
|
|
use Exception;
|
|
use Swoole\Coroutine;
|
|
use Swoole\Coroutine\Channel;
|
|
|
|
/**
|
|
* Class Pool
|
|
* @package Snowflake\Snowflake\Pool
|
|
*/
|
|
abstract class Pool extends Component
|
|
{
|
|
|
|
/** @var Channel[] */
|
|
private $_items = [];
|
|
|
|
protected $max = 60;
|
|
|
|
/**
|
|
* @param $name
|
|
* @param false $isMaster
|
|
* @param int $max
|
|
*/
|
|
public function initConnections($name, $isMaster = false, $max = 60)
|
|
{
|
|
$name = $this->name($name, $isMaster);
|
|
if (isset($this->_items[$name]) &&
|
|
$this->_items[$name] instanceof Channel) {
|
|
return;
|
|
}
|
|
$this->_items[$name] = new Channel($max);
|
|
$this->max = $max;
|
|
}
|
|
|
|
/**
|
|
* @param $name
|
|
* @return mixed
|
|
* @throws Exception
|
|
*/
|
|
protected function get($name)
|
|
{
|
|
[$timeout, $connection] = $this->_items[$name]->pop();
|
|
if (!$this->checkCanUse($name, $timeout, $connection)) {
|
|
unset($client);
|
|
return [0, null];
|
|
} else {
|
|
return [$timeout, $connection];
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @param $cds
|
|
* @param false $isMaster
|
|
* @return string
|
|
*/
|
|
public function name($cds, $isMaster = false)
|
|
{
|
|
return hash('sha1', $cds . ($isMaster ? 'master' : 'slave'));
|
|
}
|
|
|
|
|
|
/**
|
|
* @param $name
|
|
* @param $time
|
|
* @param $client
|
|
* @return mixed
|
|
* 检查连接可靠性
|
|
* @throws Exception
|
|
*/
|
|
public function checkCanUse($name, $time, $client)
|
|
{
|
|
throw new Exception('Undefined system processing function.');
|
|
}
|
|
|
|
/**
|
|
* @param $name
|
|
* @throws Exception
|
|
*/
|
|
public function desc($name)
|
|
{
|
|
throw new Exception('Undefined system processing function.');
|
|
}
|
|
|
|
/**
|
|
* @param array $config
|
|
* @param $isMaster
|
|
* @throws Exception
|
|
*/
|
|
public function getConnection(array $config, $isMaster)
|
|
{
|
|
throw new Exception('Undefined system processing function.');
|
|
}
|
|
|
|
/**
|
|
* @param $name
|
|
* @return mixed
|
|
*/
|
|
public function hasItem($name)
|
|
{
|
|
return $this->size($name) > 0;
|
|
}
|
|
|
|
|
|
/**
|
|
* @param $name
|
|
* @return mixed
|
|
*/
|
|
public function size($name)
|
|
{
|
|
if (!isset($this->_items[$name])) {
|
|
return 0;
|
|
}
|
|
return $this->_items[$name]->length();
|
|
}
|
|
|
|
|
|
/**
|
|
* @param $name
|
|
* @param $client
|
|
*/
|
|
public function push($name, $client)
|
|
{
|
|
$this->_items[$name]->push([time(), $client]);
|
|
$this->debug('release connect.' . $this->hasItem($name));
|
|
}
|
|
|
|
|
|
/**
|
|
* @param $name
|
|
*/
|
|
public function clean($name)
|
|
{
|
|
if (!isset($this->_items[$name])) {
|
|
return;
|
|
}
|
|
while ([$time, $client] = $this->_items[$name]->pop(0.001)) {
|
|
unset($client);
|
|
}
|
|
}
|
|
|
|
}
|