add WorkerMemoryUsageSubscriber

src/Messenger/WorkerMemoryUsageSubscriber.php

@@ -0,0 +1,67 @@
+<?php
+
+declare(strict_types=1);
+
+namespace WonderNetwork\SlimKernel\Messenger;
+
+use Psr\Log\LoggerInterface;
+use Symfony\Component\EventDispatcher\EventSubscriberInterface;
+use Symfony\Component\Messenger\Event\WorkerRunningEvent;
+use WonderNetwork\SlimKernel\System\RealSystem;
+use WonderNetwork\SlimKernel\System\System;
+
+final class WorkerMemoryUsageSubscriber implements EventSubscriberInterface {
+    private int $memoryUsage;
+    private readonly System $system;
+
+    /**
+     * @return iterable<string,string>
+     */
+    public static function getSubscribedEvents(): iterable {
+        return [
+            WorkerRunningEvent::class => 'onWorkerRunning',
+        ];
+    }
+
+    public function __construct(
+        private readonly LoggerInterface $logger,
+        private readonly int $cutoff = 1024,
+        System $system = null,
+    ) {
+        $this->system = $system ?? new RealSystem();
+        $this->memoryUsage = $this->system->memoryUsage();
+    }
+
+    public function onWorkerRunning(): void {
+        $currentUsage = $this->system->memoryUsage();
+
+        $difference = abs($this->memoryUsage - $currentUsage);
+
+        if ($difference < $this->cutoff) {
+            return;
+        }
+
+        $this->logger->debug(
+            "Memory usage changed: {current} ({sign}{difference})",
+            [
+                'current' => $this->formatBytes($currentUsage),
+                'sign' => ($this->memoryUsage > $currentUsage) ? "-" : "+",
+                'difference' => $this->formatBytes($difference),
+            ],
+        );
+
+        $this->memoryUsage = $currentUsage;
+    }
+
+    private function formatBytes(int $bytes): string {
+        $units = ['B', 'KB', 'MB', 'GB', 'TB'];
+
+        $bytes = max($bytes, 0);
+        $pow = floor(($bytes ? log($bytes) : 0) / log(1024));
+        $pow = (int) min($pow, count($units) - 1);
+
+        $bytes /= 1024 ** $pow;
+
+        return sprintf("%s %s", round($bytes, 2), $units[$pow]);
+    }
+}

src/System/RealSystem.php

@@ -0,0 +1,11 @@
+<?php
+
+declare(strict_types=1);
+
+namespace WonderNetwork\SlimKernel\System;
+
+final readonly class RealSystem implements System {
+    public function memoryUsage(): int {
+        return memory_get_usage();
+    }
+}

src/System/System.php

@@ -0,0 +1,9 @@
+<?php
+
+declare(strict_types=1);
+
+namespace WonderNetwork\SlimKernel\System;
+
+interface System {
+    public function memoryUsage(): int;
+}

tests/Messenger/WorkerMemoryUsageSubscriberTest.php

@@ -0,0 +1,46 @@
+<?php
+
+declare(strict_types=1);
+
+namespace WonderNetwork\SlimKernel\Messenger;
+
+use Monolog\Handler\TestHandler;
+use Monolog\Logger;
+use PHPUnit\Framework\TestCase;
+use WonderNetwork\SlimKernel\System\FakeSystem;
+
+final class WorkerMemoryUsageSubscriberTest extends TestCase {
+    public function testReportsAboveCutoff(): void {
+        $buffer = new TestHandler();
+        $logger = new Logger('some');
+        $logger->pushHandler($buffer);
+
+        $system = new FakeSystem();
+
+        $sut = new WorkerMemoryUsageSubscriber(
+            logger: $logger,
+            cutoff: 10 * 1024,
+            system: $system,
+        );
+        $sut->onWorkerRunning();
+        self::assertEmpty($buffer->getRecords());
+
+        $system->setMemoryUsage(10 * 1024 - 1);
+        $sut->onWorkerRunning();
+        self::assertCount(0, $buffer->getRecords());
+
+        $system->setMemoryUsage(10 * 1024);
+        $sut->onWorkerRunning();
+        self::assertCount(1, $buffer->getRecords());
+        self::assertSame([
+            'current' => '10 KB',
+            'sign' => '+',
+            'difference' => '10 KB',
+        ], $buffer->getRecords()[0]->context);
+
+        $sut->onWorkerRunning();
+        $system->setMemoryUsage(12 * 1024);
+        $sut->onWorkerRunning();
+        self::assertCount(1, $buffer->getRecords());
+    }
+}

tests/System/FakeSystem.php

@@ -0,0 +1,17 @@
+<?php
+
+declare(strict_types=1);
+
+namespace WonderNetwork\SlimKernel\System;
+
+final class FakeSystem implements System {
+    private int $memoryUsage = 0;
+
+    public function setMemoryUsage(int $memoryUsage): void {
+        $this->memoryUsage = $memoryUsage;
+    }
+
+    public function memoryUsage(): int {
+        return $this->memoryUsage;
+    }
+}