Map output of concurrecy calls to the index of the input

src/Illuminate/Concurrency/ForkDriver.php

@@ -17,8 +17,15 @@ class ForkDriver implements Driver
      */
     public function run(Closure|array $tasks): array
     {
+        $tasks = Arr::wrap($tasks);
+
+        $keys = array_keys($tasks);
+        $values = array_values($tasks);
+
         /** @phpstan-ignore class.notFound */
-        return Fork::new()->run(...Arr::wrap($tasks));
+        $results = Fork::new()->run(...$values);
+
+        return array_combine($keys, $results);
     }
 
     /**

src/Illuminate/Concurrency/ProcessDriver.php

@@ -31,14 +31,14 @@ public function run(Closure|array $tasks): array
         $command = Application::formatCommandString('invoke-serialized-closure');
 
         $results = $this->processFactory->pool(function (Pool $pool) use ($tasks, $command) {
-            foreach (Arr::wrap($tasks) as $task) {
-                $pool->path(base_path())->env([
+            foreach (Arr::wrap($tasks) as $key => $task) {
+                $pool->as($key)->path(base_path())->env([
                     'LARAVEL_INVOKABLE_CLOSURE' => serialize(new SerializableClosure($task)),
                 ])->command($command);
             }
         })->start()->wait();
 
-        return $results->collect()->map(function ($result) {
+        return $results->collect()->mapWithKeys(function ($result, $key) {
             $result = json_decode($result->output(), true);
 
             if (! $result['successful']) {
@@ -47,7 +47,7 @@ public function run(Closure|array $tasks): array
                 );
             }
 
-            return unserialize($result['result']);
+            return [$key => unserialize($result['result'])];
         })->all();
     }
 

tests/Integration/Concurrency/ConcurrencyTest.php

@@ -1,7 +1,8 @@
 <?php
 
-namespace Illuminate\Tests\Integration\Console;
+namespace Illuminate\Tests\Integration\Concurrency;
 
+use Illuminate\Support\Facades\Concurrency;
 use Orchestra\Testbench\TestCase;
 use PHPUnit\Framework\Attributes\RequiresOperatingSystem;
 
@@ -36,4 +37,37 @@ public function testWorkCanBeDistributed()
         $this->assertEquals(2, $first);
         $this->assertEquals(4, $second);
     }
+
+    public function testOutputIsMappedToArrayInput()
+    {
+        $input = [
+            'first' => fn () => 1 + 1,
+            'second' => fn () => 2 + 2,
+        ];
+
+        $processOutput = Concurrency::driver('process')->run($input);
+
+        $this->assertIsArray($processOutput);
+        $this->assertArrayHasKey('first', $processOutput);
+        $this->assertArrayHasKey('second', $processOutput);
+
+        $syncOutput = Concurrency::driver('sync')->run($input);
+
+        $this->assertIsArray($syncOutput);
+        $this->assertArrayHasKey('first', $syncOutput);
+        $this->assertArrayHasKey('second', $syncOutput);
+
+        /** As of now, the spatie/fork package is not included by default.
+         * $forkOutput = Concurrency::driver('fork')->run([
+         * 'first' => fn() => 1 + 1,
+         * 'second' => fn() => 2 + 2,
+         * ]);.
+         *
+         * $this->assertIsArray($forkOutput);
+         * $this->assertArrayHasKey('first', $forkOutput);
+         * $this->assertArrayHasKey('second', $forkOutput);
+         * $this->assertEquals(2, $forkOutput['first']);
+         * $this->assertEquals(4, $forkOutput['second']);
+         */
+    }
 }