MaintenanceModeEnabler.php

Go to the documentation of this file.

<?php
namespace Magento\Framework\App\Console;

use Magento\Framework\App\MaintenanceMode;
use Symfony\Component\Console\Output\OutputInterface;

class MaintenanceModeEnabler
{
    private $maintenanceMode;

    private $skipDisableMaintenanceMode;

    public function __construct(MaintenanceMode $maintenanceMode)
    {
        $this->maintenanceMode = $maintenanceMode;
    }

    private function enableMaintenanceMode(OutputInterface $output)
    {
        if ($this->maintenanceMode->isOn()) {
            $this->skipDisableMaintenanceMode = true;
            $output->writeln('<info>Maintenance mode already enabled</info>');
            return;
        }

        $this->maintenanceMode->set(true);
        $this->skipDisableMaintenanceMode = false;
        $output->writeln('<info>Enabling maintenance mode</info>');
    }

    private function disableMaintenanceMode(OutputInterface $output)
    {
        if ($this->skipDisableMaintenanceMode) {
            $output->writeln('<info>Skipped disabling maintenance mode</info>');
            return;
        }

        $this->maintenanceMode->set(false);
        $output->writeln('<info>Disabling maintenance mode</info>');
    }

    public function executeInMaintenanceMode(callable $task, OutputInterface $output, bool $holdMaintenanceOnFailure)
    {
        $this->enableMaintenanceMode($output);

        try {
            $result = call_user_func($task);
        } catch (\Throwable $e) {
            if (!$holdMaintenanceOnFailure) {
                $this->disableMaintenanceMode($output);
            }
            throw $e;
        }

        $this->disableMaintenanceMode($output);
        return $result;
    }
}