Skip to content

Commit 16b0fd9

Browse files
HypeMCHypeMC
committed
[Console] Add AlarmableCommandInterface and console.alarm event
1 parent cf3dfb6 commit 16b0fd9

File tree

12 files changed

+579
-25
lines changed

12 files changed

+579
-25
lines changed

‎src/Symfony/Bundle/FrameworkBundle/Console/Application.php

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,8 +12,10 @@
1212
namespace Symfony\Bundle\FrameworkBundle\Console;
1313

1414
use Symfony\Component\Console\Application as BaseApplication;
15+
use Symfony\Component\Console\Command\AlarmableCommandInterface;
1516
use Symfony\Component\Console\Command\Command;
1617
use Symfony\Component\Console\Command\ListCommand;
18+
use Symfony\Component\Console\Command\TraceableAlarmableCommand;
1719
use Symfony\Component\Console\Command\TraceableCommand;
1820
use Symfony\Component\Console\Debug\CliRequest;
1921
use Symfony\Component\Console\Input\InputInterface;
@@ -112,7 +114,10 @@ protected function doRunCommand(Command $command, InputInterface $input, OutputI
112114

113115
(new SymfonyStyle($input, $output))->warning('The "--profile" option needs the profiler integration. Try enabling the "framework.profiler" option.');
114116
} else {
115-
$command = new TraceableCommand($command, $container->get('debug.stopwatch'));
117+
$command = $command instanceof AlarmableCommandInterface
118+
? new TraceableAlarmableCommand($command, $container->get('debug.stopwatch'))
119+
: new TraceableCommand($command, $container->get('debug.stopwatch'))
120+
;
116121

117122
$requestStack = $container->get('.virtual_request_stack');
118123
$requestStack->push(new CliRequest($command));

‎src/Symfony/Component/Console/Application.php

Lines changed: 49 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111

1212
namespace Symfony\Component\Console;
1313

14+
use Symfony\Component\Console\Command\AlarmableCommandInterface;
1415
use Symfony\Component\Console\Command\Command;
1516
use Symfony\Component\Console\Command\CompleteCommand;
1617
use Symfony\Component\Console\Command\DumpCompletionCommand;
@@ -22,6 +23,7 @@
2223
use Symfony\Component\Console\Completion\CompletionInput;
2324
use Symfony\Component\Console\Completion\CompletionSuggestions;
2425
use Symfony\Component\Console\Completion\Suggestion;
26+
use Symfony\Component\Console\Event\ConsoleAlarmEvent;
2527
use Symfony\Component\Console\Event\ConsoleCommandEvent;
2628
use Symfony\Component\Console\Event\ConsoleErrorEvent;
2729
use Symfony\Component\Console\Event\ConsoleSignalEvent;
@@ -97,7 +99,7 @@ public function __construct(
9799
$this->defaultCommand = 'list';
98100
if (\defined('SIGINT') && SignalRegistry::isSupported()) {
99101
$this->signalRegistry = new SignalRegistry();
100-
$this->signalsToDispatchEvent = [\SIGINT, \SIGQUIT, \SIGTERM, \SIGUSR1, \SIGUSR2];
102+
$this->signalsToDispatchEvent = [\SIGINT, \SIGQUIT, \SIGTERM, \SIGUSR1, \SIGUSR2, \SIGALRM];
101103
}
102104
}
103105

@@ -973,7 +975,18 @@ protected function doRunCommand(Command $command, InputInterface $input, OutputI
973975
}
974976
}
975977

978+
// bind before getAlarmInterval() and the console.command event, so the method and listeners have access to input options/arguments
979+
try {
980+
$command->mergeApplicationDefinition();
981+
$input->bind($command->getDefinition());
982+
} catch (ExceptionInterface) {
983+
// ignore invalid options/arguments for now, to allow the event listeners to customize the InputDefinition
984+
}
985+
976986
$commandSignals = $command instanceof SignalableCommandInterface ? $command->getSubscribedSignals() : [];
987+
if ($command instanceof AlarmableCommandInterface && \defined('SIGALRM') && SignalRegistry::isSupported() && !\in_array(\SIGALRM, $commandSignals, true)) {
988+
$commandSignals[] = \SIGALRM;
989+
}
977990
if ($commandSignals || $this->dispatcher && $this->signalsToDispatchEvent) {
978991
if (!$this->signalRegistry) {
979992
throw new RuntimeException('Unable to subscribe to signal events. Make sure that the "pcntl" extension is installed and that "pcntl_*" functions are not disabled by your php.ini\'s "disable_functions" directive.');
@@ -990,19 +1003,34 @@ protected function doRunCommand(Command $command, InputInterface $input, OutputI
9901003
if ($this->dispatcher) {
9911004
// We register application signals, so that we can dispatch the event
9921005
foreach ($this->signalsToDispatchEvent as $signal) {
993-
$event = new ConsoleSignalEvent($command, $input, $output, $signal);
994-
995-
$this->signalRegistry->register($signal, function ($signal) use ($event, $command, $commandSignals) {
996-
$this->dispatcher->dispatch($event, ConsoleEvents::SIGNAL);
997-
$exitCode = $event->getExitCode();
1006+
$signalEvent = new ConsoleSignalEvent($command, $input, $output, $signal);
1007+
$alarmEvent = \SIGALRM === $signal ? new ConsoleAlarmEvent($command, $input, $output) : null;
1008+
1009+
$this->signalRegistry->register($signal, function ($signal) use ($signalEvent, $alarmEvent, $command, $commandSignals, $input, $output) {
1010+
$this->dispatcher->dispatch($signalEvent, ConsoleEvents::SIGNAL);
1011+
$exitCode = $signalEvent->getExitCode();
1012+
1013+
if (null !== $alarmEvent) {
1014+
if (false !== $exitCode) {
1015+
$alarmEvent->setExitCode($exitCode);
1016+
} else {
1017+
$alarmEvent->abortExit();
1018+
}
1019+
$this->dispatcher->dispatch($alarmEvent, ConsoleEvents::ALARM);
1020+
$exitCode = $alarmEvent->getExitCode();
1021+
}
9981022

1023+
if (\SIGALRM === $signal && $command instanceof AlarmableCommandInterface) {
1024+
$exitCode = $command->handleAlarm($exitCode);
1025+
$this->signalRegistry->scheduleAlarm($command->getAlarmInterval($input));
1026+
}
9991027
// If the command is signalable, we call the handleSignal() method
1000-
if (\in_array($signal, $commandSignals, true)) {
1028+
elseif (\in_array($signal, $commandSignals, true)) {
10011029
$exitCode = $command->handleSignal($signal, $exitCode);
10021030
}
10031031

10041032
if (false !== $exitCode) {
1005-
$event = new ConsoleTerminateEvent($command, $event->getInput(), $event->getOutput(), $exitCode, $signal);
1033+
$event = new ConsoleTerminateEvent($command, $input, $output, $exitCode, $signal);
10061034
$this->dispatcher->dispatch($event, ConsoleEvents::TERMINATE);
10071035

10081036
exit($event->getExitCode());
@@ -1015,26 +1043,29 @@ protected function doRunCommand(Command $command, InputInterface $input, OutputI
10151043
}
10161044

10171045
foreach ($commandSignals as $signal) {
1018-
$this->signalRegistry->register($signal, function (int $signal) use ($command): void {
1019-
if (false !== $exitCode = $command->handleSignal($signal)) {
1046+
$this->signalRegistry->register($signal, function (int $signal) use ($command, $input): void {
1047+
if (\SIGALRM === $signal && $command instanceof AlarmableCommandInterface) {
1048+
$exitCode = $command->handleAlarm();
1049+
$this->signalRegistry->scheduleAlarm($command->getAlarmInterval($input));
1050+
} else {
1051+
$exitCode = $command->handleSignal($signal);
1052+
}
1053+
1054+
if (false !== $exitCode) {
10201055
exit($exitCode);
10211056
}
10221057
});
10231058
}
1059+
1060+
if ($command instanceof AlarmableCommandInterface) {
1061+
$this->signalRegistry->scheduleAlarm($command->getAlarmInterval($input));
1062+
}
10241063
}
10251064

10261065
if (null === $this->dispatcher) {
10271066
return $command->run($input, $output);
10281067
}
10291068

1030-
// bind before the console.command event, so the listeners have access to input options/arguments
1031-
try {
1032-
$command->mergeApplicationDefinition();
1033-
$input->bind($command->getDefinition());
1034-
} catch (ExceptionInterface) {
1035-
// ignore invalid options/arguments for now, to allow the event listeners to customize the InputDefinition
1036-
}
1037-
10381069
$event = new ConsoleCommandEvent($command, $input, $output);
10391070
$e = null;
10401071

‎src/Symfony/Component/Console/CHANGELOG.md

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,11 @@
11
CHANGELOG
22
=========
33

4+
7.2
5+
---
6+
7+
* Add `AlarmableCommandInterface` and `console.alarm` event
8+
49
7.1
510
---
611

‎src/Symfony/Component/Console/Command/AlarmableCommandInterface.php

Lines changed: 34 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,34 @@
1+
<?php
2+
3+
/*
4+
* This file is part of the Symfony package.
5+
*
6+
* (c) Fabien Potencier <fabien@symfony.com>
7+
*
8+
* For the full copyright and license information, please view the LICENSE
9+
* file that was distributed with this source code.
10+
*/
11+
12+
namespace Symfony\Component\Console\Command;
13+
14+
use Symfony\Component\Console\Input\InputInterface;
15+
16+
/**
17+
* Interface for commands that listen to SIGALRM signals.
18+
*/
19+
interface AlarmableCommandInterface
20+
{
21+
/**
22+
* The method will be called before the command is run and subsequently on each SIGALRM signal.
23+
*
24+
* @return int The alarm interval in seconds
25+
*/
26+
public function getAlarmInterval(InputInterface $input): int;
27+
28+
/**
29+
* The method will be called when the application is signaled with SIGALRM.
30+
*
31+
* @return int|false The exit code to return or false to continue the normal execution
32+
*/
33+
public function handleAlarm(int|false $previousExitCode = 0): int|false;
34+
}

‎src/Symfony/Component/Console/Command/TraceableAlarmableCommand.php

Lines changed: 57 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,57 @@
1+
<?php
2+
3+
/*
4+
* This file is part of the Symfony package.
5+
*
6+
* (c) Fabien Potencier <fabien@symfony.com>
7+
*
8+
* For the full copyright and license information, please view the LICENSE
9+
* file that was distributed with this source code.
10+
*/
11+
12+
namespace Symfony\Component\Console\Command;
13+
14+
use Symfony\Component\Console\Input\InputInterface;
15+
use Symfony\Component\Stopwatch\Stopwatch;
16+
17+
/**
18+
* @internal
19+
*
20+
* @property Command&AlarmableCommandInterface $command
21+
*/
22+
final class TraceableAlarmableCommand extends TraceableCommand implements AlarmableCommandInterface
23+
{
24+
public function __construct(Command&AlarmableCommandInterface $command, Stopwatch $stopwatch)
25+
{
26+
parent::__construct($command, $stopwatch);
27+
}
28+
29+
public function getSubscribedSignals(): array
30+
{
31+
$commandSignals = parent::getSubscribedSignals();
32+
33+
if (!\in_array(\SIGALRM, $commandSignals, true)) {
34+
$commandSignals[] = \SIGALRM;
35+
}
36+
37+
return $commandSignals;
38+
}
39+
40+
public function getAlarmInterval(InputInterface $input): int
41+
{
42+
return $this->command->getAlarmInterval($input);
43+
}
44+
45+
public function handleAlarm(false|int $previousExitCode = 0): int|false
46+
{
47+
$event = $this->stopwatch->start($this->getName().'.handle_alarm');
48+
49+
$exit = $this->command->handleAlarm($previousExitCode);
50+
51+
$event->stop();
52+
53+
$this->recordHandledSignal(\SIGALRM, $event);
54+
55+
return $exit;
56+
}
57+
}

‎src/Symfony/Component/Console/Command/TraceableCommand.php

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -21,13 +21,14 @@
2121
use Symfony\Component\Console\Output\ConsoleOutputInterface;
2222
use Symfony\Component\Console\Output\OutputInterface;
2323
use Symfony\Component\Stopwatch\Stopwatch;
24+
use Symfony\Component\Stopwatch\StopwatchEvent;
2425

2526
/**
2627
* @internal
2728
*
2829
* @author Jules Pietri <jules@heahprod.com>
2930
*/
30-
final class TraceableCommand extends Command implements SignalableCommandInterface
31+
class TraceableCommand extends Command implements SignalableCommandInterface
3132
{
3233
public readonly Command $command;
3334
public int $exitCode;
@@ -48,7 +49,7 @@ final class TraceableCommand extends Command implements SignalableCommandInterfa
4849

4950
public function __construct(
5051
Command $command,
51-
private readonly Stopwatch $stopwatch,
52+
protected readonly Stopwatch $stopwatch,
5253
) {
5354
if ($command instanceof LazyCommand) {
5455
$command = $command->getCommand();
@@ -103,6 +104,13 @@ public function handleSignal(int $signal, int|false $previousExitCode = 0): int|
103104

104105
$event->stop();
105106

107+
$this->recordHandledSignal($signal, $event);
108+
109+
return $exit;
110+
}
111+
112+
protected function recordHandledSignal(int $signal, StopwatchEvent $event): void
113+
{
106114
if (!isset($this->handledSignals[$signal])) {
107115
$this->handledSignals[$signal] = [
108116
'handled' => 0,
@@ -117,8 +125,6 @@ public function handleSignal(int $signal, int|false $previousExitCode = 0): int|
117125
$this->handledSignals[$signal]['memory'],
118126
$event->getMemory() >> 20
119127
);
120-
121-
return $exit;
122128
}
123129

124130
/**

‎src/Symfony/Component/Console/ConsoleEvents.php

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111

1212
namespace Symfony\Component\Console;
1313

14+
use Symfony\Component\Console\Event\ConsoleAlarmEvent;
1415
use Symfony\Component\Console\Event\ConsoleCommandEvent;
1516
use Symfony\Component\Console\Event\ConsoleErrorEvent;
1617
use Symfony\Component\Console\Event\ConsoleSignalEvent;
@@ -40,6 +41,14 @@ final class ConsoleEvents
4041
*/
4142
public const SIGNAL = 'console.signal';
4243

44+
/**
45+
* The ALARM event allows you to perform some actions
46+
* after the command received a SIGALRM signal.
47+
*
48+
* @Event("Symfony\Component\Console\Event\ConsoleAlarmEvent")
49+
*/
50+
public const ALARM = 'console.alarm';
51+
4352
/**
4453
* The TERMINATE event allows you to attach listeners after a command is
4554
* executed by the console.
@@ -67,6 +76,7 @@ final class ConsoleEvents
6776
ConsoleCommandEvent::class => self::COMMAND,
6877
ConsoleErrorEvent::class => self::ERROR,
6978
ConsoleSignalEvent::class => self::SIGNAL,
79+
ConsoleAlarmEvent::class => self::ALARM,
7080
ConsoleTerminateEvent::class => self::TERMINATE,
7181
];
7282
}

‎src/Symfony/Component/Console/Event/ConsoleAlarmEvent.php

Lines changed: 47 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,47 @@
1+
<?php
2+
3+
/*
4+
* This file is part of the Symfony package.
5+
*
6+
* (c) Fabien Potencier <fabien@symfony.com>
7+
*
8+
* For the full copyright and license information, please view the LICENSE
9+
* file that was distributed with this source code.
10+
*/
11+
12+
namespace Symfony\Component\Console\Event;
13+
14+
use Symfony\Component\Console\Command\Command;
15+
use Symfony\Component\Console\Input\InputInterface;
16+
use Symfony\Component\Console\Output\OutputInterface;
17+
18+
final class ConsoleAlarmEvent extends ConsoleEvent
19+
{
20+
public function __construct(
21+
Command $command,
22+
InputInterface $input,
23+
OutputInterface $output,
24+
private int|false $exitCode = 0,
25+
) {
26+
parent::__construct($command, $input, $output);
27+
}
28+
29+
public function setExitCode(int $exitCode): void
30+
{
31+
if ($exitCode < 0 || $exitCode > 255) {
32+
throw new \InvalidArgumentException('Exit code must be between 0 and 255.');
33+
}
34+
35+
$this->exitCode = $exitCode;
36+
}
37+
38+
public function abortExit(): void
39+
{
40+
$this->exitCode = false;
41+
}
42+
43+
public function getExitCode(): int|false
44+
{
45+
return $this->exitCode;
46+
}
47+
}

‎src/Symfony/Component/Console/SignalRegistry/SignalRegistry.php

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -54,4 +54,12 @@ public function handle(int $signal): void
5454
$signalHandler($signal, $hasNext);
5555
}
5656
}
57+
58+
/**
59+
* @internal
60+
*/
61+
public function scheduleAlarm(int $seconds): void
62+
{
63+
pcntl_alarm($seconds);
64+
}
5765
}

0 commit comments

Comments
 (0)
pFad - Phonifier reborn

Pfad - The Proxy pFad of © 2024 Garber Painting. All rights reserved.

Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.


Alternative Proxies:

Alternative Proxy

pFad Proxy

pFad v3 Proxy

pFad v4 Proxy