PidConsumerManager.php

Go to the documentation of this file.

<?php
namespace Magento\MessageQueue\Model\Cron\ConsumersRunner;

use Magento\Framework\App\Filesystem\DirectoryList;
use Magento\Framework\Filesystem\Directory\WriteInterface;
use Magento\Framework\Filesystem\Directory\ReadInterface;
use Magento\Framework\Filesystem;
use Magento\Framework\Exception\FileSystemException;

class PidConsumerManager
{
    const PID_FILE_EXT = '.pid';

    private $filesystem;

    public function __construct(Filesystem $filesystem)
    {
        $this->filesystem = $filesystem;
    }

    public function isRun($pidFilePath)
    {
        $pid = $this->getPid($pidFilePath);
        if ($pid) {
            if (function_exists('posix_getpgid')) {
                return (bool) posix_getpgid($pid);
            } else {
                return $this->checkIsProcessExists($pid);
            }
        }

        return false;
    }

    private function checkIsProcessExists($pid)
    {
        if (!function_exists('exec')) {
            throw new \RuntimeException('Function exec is not available');
        }

        exec(escapeshellcmd('ps -p ' . $pid), $output, $code);

        $code = (int) $code;

        switch ($code) {
            case 0:
                return true;
                break;
            case 1:
                return false;
                break;
            default:
                throw new \RuntimeException('Exec returned non-zero code', $code);
                break;
        }
    }

    public function getPid($pidFilePath)
    {
        $directory = $this->filesystem->getDirectoryRead(DirectoryList::VAR_DIR);

        if ($directory->isExist($pidFilePath)) {
            return (int) $directory->readFile($pidFilePath);
        }

        return 0;
    }

    public function savePid($pidFilePath)
    {
        $directory = $this->filesystem->getDirectoryWrite(DirectoryList::VAR_DIR);
        $directory->writeFile($pidFilePath, function_exists('posix_getpid') ? posix_getpid() : getmypid(), 'w');
    }
}