Fix Swoole auto worker count in containers (#1101)

* Fix Swoole auto worker count in containers

* formatting

* formatting

---------

Co-authored-by: Taylor Otwell <taylor@laravel.com>

Author: mtawil

src/Swoole/SwooleExtension.php

@@ -2,10 +2,22 @@
 
 namespace Laravel\Octane\Swoole;
 
+use Closure;
 use Swoole\Process;
 
 class SwooleExtension
 {
+    /**
+     * Create a new Swoole extension helper.
+     */
+    public function __construct(
+        protected ?Closure $isReadable = null,
+        protected ?Closure $fileGetContents = null,
+    ) {
+        $this->isReadable ??= static fn (string $path): bool => is_readable($path);
+        $this->fileGetContents ??= static fn (string $path): string|false => @file_get_contents($path);
+    }
+
     /**
      * Determine if the Swoole extension is installed.
      */
@@ -41,6 +53,12 @@ public function setProcessName(string $appName, string $processName): void
      */
     public function cpuCount(): int
     {
+        $cgroupCpuCount = $this->containerCpuCount();
+
+        if ($cgroupCpuCount !== null) {
+            return $cgroupCpuCount;
+        }
+
         if (function_exists('swoole_cpu_num')) {
             return swoole_cpu_num();
         }
@@ -51,4 +69,45 @@ public function cpuCount(): int
 
         return 1;
     }
+
+    /**
+     * Get the CPU count from the container's cgroup CPU quota, if available.
+     */
+    protected function containerCpuCount(): ?int
+    {
+        // cgroups v2...
+        if (($this->isReadable)('/sys/fs/cgroup/cpu.max')) {
+            $cpuMax = ($this->fileGetContents)('/sys/fs/cgroup/cpu.max');
+
+            if ($cpuMax !== false) {
+                $parts = preg_split('/\s+/', trim($cpuMax));
+                $quota = $parts[0] ?? null;
+                $period = isset($parts[1]) ? (int) $parts[1] : 0;
+
+                if ($quota !== 'max' && $period > 0) {
+                    return (int) max(1, ceil((int) $quota / $period));
+                }
+            }
+        }
+
+        // cgroups v1...
+        $quotaFile = '/sys/fs/cgroup/cpu/cpu.cfs_quota_us';
+        $periodFile = '/sys/fs/cgroup/cpu/cpu.cfs_period_us';
+
+        if (($this->isReadable)($quotaFile) && ($this->isReadable)($periodFile)) {
+            $quota = ($this->fileGetContents)($quotaFile);
+            $period = ($this->fileGetContents)($periodFile);
+
+            if ($quota !== false && $period !== false) {
+                $quota = (int) trim($quota);
+                $period = (int) trim($period);
+
+                if ($quota > 0 && $period > 0) {
+                    return (int) max(1, ceil($quota / $period));
+                }
+            }
+        }
+
+        return null;
+    }
 }

tests/SwooleExtensionTest.php

@@ -14,4 +14,82 @@ public function test_cpu_count()
 
         $this->assertTrue($cpuCount > 0);
     }
+
+    public function test_container_cpu_count_reads_cgroup_v2_quota()
+    {
+        $extension = $this->fakeExtensionWithFiles([
+            '/sys/fs/cgroup/cpu.max' => '200000 100000',
+        ]);
+
+        $this->assertSame(2, $extension->readContainerCpuCount());
+    }
+
+    public function test_container_cpu_count_reads_cgroup_v1_quota()
+    {
+        $extension = $this->fakeExtensionWithFiles([
+            '/sys/fs/cgroup/cpu/cpu.cfs_quota_us' => '250000',
+            '/sys/fs/cgroup/cpu/cpu.cfs_period_us' => '100000',
+        ]);
+
+        $this->assertSame(3, $extension->readContainerCpuCount());
+    }
+
+    public function test_container_cpu_count_returns_null_when_unlimited_or_missing()
+    {
+        $extension = $this->fakeExtensionWithFiles([
+            '/sys/fs/cgroup/cpu.max' => 'max 100000',
+        ]);
+
+        $this->assertNull($extension->readContainerCpuCount());
+    }
+
+    public function test_container_cpu_count_ignores_invalid_v2_data()
+    {
+        $extension = $this->fakeExtensionWithFiles([
+            '/sys/fs/cgroup/cpu.max' => '200000',
+        ]);
+
+        $this->assertNull($extension->readContainerCpuCount());
+    }
+
+    public function test_cpu_count_respects_cgroup_v2_limit()
+    {
+        $extension = $this->getMockBuilder(SwooleExtension::class)
+            ->onlyMethods(['containerCpuCount'])
+            ->getMock();
+
+        $extension->method('containerCpuCount')->willReturn(2);
+
+        $this->assertSame(2, $extension->cpuCount());
+    }
+
+    public function test_cpu_count_falls_back_when_no_cgroup_limit()
+    {
+        $extension = $this->getMockBuilder(SwooleExtension::class)
+            ->onlyMethods(['containerCpuCount'])
+            ->getMock();
+
+        $extension->method('containerCpuCount')->willReturn(null);
+
+        $this->assertTrue($extension->cpuCount() >= 1);
+    }
+
+    protected function fakeExtensionWithFiles(array $files): object
+    {
+        return new class($files) extends SwooleExtension
+        {
+            public function __construct(private array $files)
+            {
+                parent::__construct(
+                    isReadable: fn (string $path): bool => array_key_exists($path, $this->files),
+                    fileGetContents: fn (string $path): string|false => $this->files[$path] ?? false,
+                );
+            }
+
+            public function readContainerCpuCount(): ?int
+            {
+                return $this->containerCpuCount();
+            }
+        };
+    }
 }