kiri-core/System/Pool/ObjectPool.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);
	}

}