news_producer/vendor/hyperf/pool/src/SimplePool/PoolFactory.php

<?php

declare(strict_types=1);
/**
 * This file is part of Hyperf.
 *
 * @link     https://www.hyperf.io
 * @document https://hyperf.wiki
 * @contact  group@hyperf.io
 * @license  https://github.com/hyperf/hyperf/blob/master/LICENSE
 */

namespace Hyperf\Pool\SimplePool;

use Psr\Container\ContainerInterface;

use function Hyperf\Support\make;

class PoolFactory
{
    /**
     * @var Pool[]
     */
    protected array $pools = [];

    protected array $configs = [];

    public function __construct(protected ContainerInterface $container)
    {
    }

    public function addConfig(Config $config)
    {
        $this->configs[$config->getName()] = $config;
        return $this;
    }

    public function get(string $name, callable $callback, array $option = []): Pool
    {
        if (! $this->hasConfig($name)) {
            $config = new Config($name, $callback, $option);
            $this->addConfig($config);
        }

        $config = $this->getConfig($name);

        if (! isset($this->pools[$name])) {
            $this->pools[$name] = make(Pool::class, [
                'callback' => $config->getCallback(),
                'option' => $config->getOption(),
            ]);
        }

        return $this->pools[$name];
    }

    public function getPoolNames(): array
    {
        return array_keys($this->pools);
    }

    protected function hasConfig(string $name): bool
    {
        return isset($this->configs[$name]);
    }

    protected function getConfig(string $name): Config
    {
        return $this->configs[$name];
    }
}