105 lines
1.8 KiB
PHP
105 lines
1.8 KiB
PHP
<?php
|
|
|
|
|
|
namespace Snowflake\Pool;
|
|
|
|
|
|
use Exception;
|
|
use ReflectionException;
|
|
use Snowflake\Event;
|
|
use Snowflake\Exception\ComponentException;
|
|
use Snowflake\Exception\NotFindClassException;
|
|
use Snowflake\Snowflake;
|
|
use Swoole\Coroutine;
|
|
|
|
/**
|
|
* Class ObjectPool
|
|
* @package Snowflake\Pool
|
|
*/
|
|
class ObjectPool extends \Snowflake\Abstracts\Pool
|
|
{
|
|
|
|
private array $_waitRecover = [];
|
|
|
|
|
|
/**
|
|
* ObjectPool constructor.
|
|
* @param array $config
|
|
* @throws ComponentException
|
|
* @throws Exception
|
|
*/
|
|
public function __construct($config = [])
|
|
{
|
|
$this->max = 5000;
|
|
|
|
listen(Event::SYSTEM_RESOURCE_RELEASES, [$this, 'destruct']);
|
|
|
|
parent::__construct($config);
|
|
}
|
|
|
|
|
|
/**
|
|
* @param array $config
|
|
* @param callable $construct
|
|
* @return mixed
|
|
* @throws Exception
|
|
*/
|
|
public function load(mixed $config, callable $construct): mixed
|
|
{
|
|
if (is_object($config)) {
|
|
return $config;
|
|
}
|
|
return $this->getFromChannel($name = md5($config), [$config, $construct]);
|
|
}
|
|
|
|
|
|
/**
|
|
* @param string $name
|
|
* @param mixed $config
|
|
* @return mixed
|
|
*/
|
|
public function createClient(string $name, mixed $config): mixed
|
|
{
|
|
return call_user_func($config[1]);
|
|
}
|
|
|
|
|
|
/**
|
|
* @param string $name
|
|
* @param $object
|
|
*/
|
|
public function release(string $name, mixed $object)
|
|
{
|
|
if (!isset($this->_waitRecover[$name])) {
|
|
$this->_waitRecover[$name] = [];
|
|
}
|
|
$this->_waitRecover[$name][] = $object;
|
|
}
|
|
|
|
|
|
/**
|
|
* 清理等待回收的对象
|
|
*/
|
|
public function destruct()
|
|
{
|
|
$this->async_create(function ($scope) {
|
|
if (empty($scope->_waitRecover)) {
|
|
return;
|
|
}
|
|
foreach ($scope->_waitRecover as $name => $value) {
|
|
if (empty($value)) {
|
|
continue;
|
|
}
|
|
foreach ($value as $object) {
|
|
if (method_exists($object, 'clean')) {
|
|
$object->clean();
|
|
}
|
|
$scope->push($name, $object);
|
|
}
|
|
unset($scope->_waitRecover[$name]);
|
|
}
|
|
}, $this);
|
|
}
|
|
|
|
}
|