ProcessManager.php 2.25 KB
Newer Older
Ketan's avatar
Ketan committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107
<?php
/**
 * Copyright © Magento, Inc. All rights reserved.
 * See COPYING.txt for license details.
 */

namespace Magento\Deploy\Model;

class ProcessManager
{
    /** @var Process[] */
    private $processes = [];

    /**
     * @var ProcessFactory
     */
    private $processFactory;

    /**
     * ProcessManager constructor.
     * @param ProcessFactory $processFactory
     */
    public function __construct(ProcessFactory $processFactory)
    {
        $this->processFactory = $processFactory;
    }

    /**
     * Forks the currently running process.
     *
     * @param callable $handler
     *
     * @return Process
     * @throws \RuntimeException
     * @SuppressWarnings(PHPMD.ExitExpression)
     */
    public function fork(callable $handler)
    {
        $process = $this->createProcess($handler);
        $pid = pcntl_fork();

        if ($pid === -1) {
            throw new \RuntimeException('Unable to fork a new process');
        }

        if ($pid) {
            $process->setPid($pid);
            $this->processes[$pid] = $process;
            return $process;
        }

        // process child process
        $this->processes = [];
        $process->setPid(getmypid());
        $process->run();

        exit(0);
    }

    /**
     * @return Process[]
     */
    public function getProcesses()
    {
        return $this->processes;
    }

    /**
     * @param Process $process
     * @return void
     */
    public function delete(Process $process)
    {
        unset($this->processes[$process->getPid()]);
    }

    /**
     * @param callable $handler
     * @return Process
     */
    private function createProcess(callable $handler)
    {
        return $this->processFactory->create(['handler' => $handler]);
    }

    /**
     * Protect against zombie process
     * @return void
     * @SuppressWarnings(PHPMD.UnusedLocalVariable)
     */
    private function freeResources()
    {
        foreach ($this->processes as $process) {
            if (pcntl_waitpid($process->getPid(), $status) === -1) {
                throw new \RuntimeException('Error while waiting for process '. $process->getPid());
            }
        }
    }

    /**
     * Free resources
     */
    public function __destruct()
    {
        $this->freeResources();
    }
}