chore: try tcp check

open-runtimes · Oct 22, 2024 · a787ca1 · a787ca1
1 parent 899f3d4
commit a787ca1
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/app/http.php b/app/http.php
@@ -832,8 +832,9 @@ function cleanUp(Orchestration $orchestration, Table $activeRuntimes, array $net
     ->inject('response')
     ->inject('request')
     ->inject('log')
+    ->inject('orchestration')
     ->action(
-        function (string $runtimeId, ?string $payload, string $path, string $method, mixed $headers, int $timeout, string $image, string $source, string $entrypoint, mixed $variables, float $cpus, int $memory, string $version, string $runtimeEntrypoint, bool $logging, string $restartPolicy, Table $activeRuntimes, Response $response, Request $request, Log $log) {
+        function (string $runtimeId, ?string $payload, string $path, string $method, mixed $headers, int $timeout, string $image, string $source, string $entrypoint, mixed $variables, float $cpus, int $memory, string $version, string $runtimeEntrypoint, bool $logging, string $restartPolicy, Table $activeRuntimes, Response $response, Request $request, Log $log, Orchestration $orchestration) {
             if (empty($payload)) {
                 $payload = '';
             }
@@ -1243,7 +1244,16 @@ function (string $runtimeId, ?string $payload, string $path, string $method, mix
                         throw new Exception('Function timed out during cold start.', 400);
                     }
 
-                    $online = $validator->isValid($hostname . ':' . 3000);
+                    $output = '';
+
+                    $status = $orchestration->execute(
+                        name: $runtimeName,
+                        command: ['bash', '-c', '>/dev/tcp/localhost/3000'],
+                        output: $output,
+                        timeout: $timeout
+                    );
+
+                    $online = $status === 0;
                     if ($online) {
                         break;
                     }