1
0
mirror of https://github.com/danog/parallel.git synced 2024-12-04 02:27:55 +01:00
parallel/lib/Worker/AutoloadingWorkerFactory.php

72 lines
2.2 KiB
PHP
Raw Normal View History

<?php
namespace Amp\Parallel\Worker;
use Amp\Parallel\Context\Parallel;
use Amp\Parallel\Context\Thread;
/**
* The built-in worker factory type.
*/
final class AutoloadingWorkerFactory implements WorkerFactory
{
/** @var string */
private $autoloadPath;
/** @var string */
private $className;
/**
* @param string $autoloadFilePath Path to custom autoloader.
* @param string $envClassName Name of class implementing \Amp\Parallel\Worker\Environment to instigate in each
* worker. Defaults to \Amp\Parallel\Worker\BasicEnvironment.
*
* @throws \Error If the given class name does not exist or does not implement \Amp\Parallel\Worker\Environment.
*/
public function __construct(string $autoloadFilePath, string $envClassName = BasicEnvironment::class)
{
if (!\file_exists($autoloadFilePath)) {
throw new \Error(\sprintf("No file found at autoload path given '%s'", $autoloadFilePath));
}
if (!\class_exists($envClassName)) {
throw new \Error(\sprintf("Invalid environment class name '%s'", $envClassName));
}
if (!\is_subclass_of($envClassName, Environment::class)) {
throw new \Error(\sprintf(
"The class '%s' does not implement '%s'",
$envClassName,
Environment::class
));
}
$this->autoloadPath = $autoloadFilePath;
$this->className = $envClassName;
}
/**
* {@inheritdoc}
*
* The type of worker created depends on the extensions available. If multi-threading is enabled, a WorkerThread
* will be created. If threads are not available a WorkerProcess will be created.
*/
public function create(): Worker
{
if (Parallel::isSupported()) {
return new WorkerParallel($this->className, $this->autoloadPath);
}
if (Thread::isSupported()) {
return new WorkerThread($this->className, $this->autoloadPath);
}
return new WorkerProcess(
$this->className,
[],
\getenv("AMP_PHP_BINARY") ?: (\defined("AMP_PHP_BINARY") ? \AMP_PHP_BINARY : null),
$this->autoloadPath
);
}
}