feat(core)!: worker mode support

packages/auth/src/Authentication/SessionAuthenticator.php

@@ -4,11 +4,10 @@
 
 namespace Tempest\Auth\Authentication;
 
-use Tempest\Container\Resettable;
 use Tempest\Http\Session\Session;
 use Tempest\Http\Session\SessionManager;
 
-final class SessionAuthenticator implements Authenticator, Resettable
+final class SessionAuthenticator implements Authenticator
 {
     public const string AUTHENTICATABLE_KEY = '#authenticatable:id';
 
@@ -77,12 +76,7 @@ public function current(): ?Authenticatable
         return $this->current;
     }
 
-    public function reset(): void
-    {
-        $this->clearCurrent();
-    }
-
-    private function clearCurrent(): void
+    public function clearCurrent(): void
     {
         $this->currentId = null;
         $this->currentClass = null;

packages/auth/src/Authentication/SessionAuthenticatorReset.php

@@ -0,0 +1,17 @@
+<?php
+
+namespace Tempest\Auth\Authentication;
+
+use Tempest\Container\Resettable;
+
+final readonly class SessionAuthenticatorReset implements Resettable
+{
+    public function __construct(
+        private SessionAuthenticator $sessionAuthenticator,
+    ) {}
+
+    public function reset(): void
+    {
+        $this->sessionAuthenticator->clearCurrent();
+    }
+}

packages/auth/tests/SessionAuthenticatorTest.php

@@ -9,7 +9,7 @@
 use Tempest\Auth\Authentication\Authenticatable;
 use Tempest\Auth\Authentication\AuthenticatableResolver;
 use Tempest\Auth\Authentication\SessionAuthenticator;
-use Tempest\Container\Resettable;
+use Tempest\Auth\Authentication\SessionAuthenticatorReset;
 use Tempest\DateTime\DateTime;
 use Tempest\Http\Session\Session;
 use Tempest\Http\Session\SessionId;
@@ -100,10 +100,9 @@ public function reset_clears_the_cached_current_authenticatable(): void
             authenticatableResolver: $resolver,
         );
 
-        $this->assertInstanceOf(Resettable::class, $authenticator);
         $this->assertSame($authenticatable, $authenticator->current());
 
-        $authenticator->reset();
+        new SessionAuthenticatorReset($authenticator)->reset();
 
         $this->assertSame($authenticatable, $authenticator->current());
         $this->assertSame(2, $resolver->resolveCalls);

packages/console/src/ConsoleApplication.php

@@ -48,7 +48,7 @@ public static function boot(
         return $container->get(ConsoleApplication::class);
     }
 
-    public function run(): never
+    public function run(): void
     {
         $exitCode = $this->container->get(ExecuteConsoleCommand::class)($this->argumentBag->getCommandName());
 

packages/core/src/DeferredTasksReset.php

@@ -0,0 +1,18 @@
+<?php
+
+namespace Tempest\Core;
+
+use Tempest\Container\Container;
+use Tempest\Container\Resettable;
+
+final readonly class DeferredTasksReset implements Resettable
+{
+    public function __construct(
+        private Container $container,
+    ) {}
+
+    public function reset(): void
+    {
+        $this->container->unregister(DeferredTasks::class);
+    }
+}

packages/core/src/FrameworkKernel.php

@@ -35,9 +35,10 @@ final class FrameworkKernel implements Kernel
     public function __construct(
         public string $root,
         /** @var DiscoveryLocation[] */
-        private array $discoveryLocations = [],
+        private readonly array $discoveryLocations = [],
         ?Container $container = null,
         ?string $internalStorage = null,
+        private readonly bool $longRunning = false,
     ) {
         $this->container = $container ?? $this->createContainer();
 
@@ -51,6 +52,7 @@ public static function boot(
         array $discoveryLocations = [],
         ?Container $container = null,
         ?string $internalStorage = null,
+        bool $longRunning = false,
     ): self {
         if (! defined('TEMPEST_START')) {
             define('TEMPEST_START', value: hrtime(as_number: true));
@@ -61,6 +63,7 @@ public static function boot(
             discoveryLocations: $discoveryLocations,
             container: $container,
             internalStorage: $internalStorage,
+            longRunning: $longRunning,
         )
             ->registerKernel()
             ->validateRoot()
@@ -98,12 +101,23 @@ public function validateRoot(): self
         return $this;
     }
 
-    public function shutdown(int|string $status = ''): never
+    public function shutdown(int|string $status = ''): void
     {
-        $this->finishDeferredTasks()
-            ->event(KernelEvent::SHUTDOWN);
+        $this->event(KernelEvent::SHUTTING_DOWN)
+            ->finishDeferredTasks();
+
+        if ($this->longRunning) {
+            $this
+                ->event(KernelEvent::RESETTING)
+                ->resetContainer()
+                ->event(KernelEvent::RESET);
+        }
+
+        $this->event(KernelEvent::SHUTDOWN);
 
-        exit($status);
+        if (! $this->longRunning) {
+            exit($status);
+        }
     }
 
     public function loadComposer(): self
@@ -236,6 +250,13 @@ public function finishDeferredTasks(): self
         return $this;
     }
 
+    public function resetContainer(): self
+    {
+        $this->container->reset();
+
+        return $this;
+    }
+
     public function event(object $event): self
     {
         if (interface_exists(EventBus::class)) {

packages/core/src/Kernel.php

@@ -23,5 +23,5 @@ public static function boot(
         ?string $internalStorage = null,
     ): self;
 
-    public function shutdown(int|string $status = ''): never;
+    public function shutdown(int|string $status = ''): void;
 }

packages/core/src/KernelEvent.php

@@ -6,6 +6,9 @@
 
 enum KernelEvent
 {
+    case SHUTTING_DOWN;
+    case RESETTING;
+    case RESET;
     case BOOTED;
     case SHUTDOWN;
 }

packages/core/src/Tempest.php

@@ -9,12 +9,17 @@
 final readonly class Tempest
 {
     /** @param \Tempest\Discovery\DiscoveryLocation[] $discoveryLocations */
-    public static function boot(?string $root = null, array $discoveryLocations = [], ?string $internalStorage = null): Container
-    {
+    public static function boot(
+        ?string $root = null,
+        array $discoveryLocations = [],
+        ?string $internalStorage = null,
+        bool $longRunning = false,
+    ): Container {
         $kernel = FrameworkKernel::boot(
             root: $root ?? getcwd(),
             discoveryLocations: $discoveryLocations,
             internalStorage: $internalStorage,
+            longRunning: $longRunning,
         );
 
         return $kernel->container;

packages/database/src/Connection/Connection.php

@@ -10,6 +10,8 @@ interface Connection
 {
     public function beginTransaction(): bool;
 
+    public function inTransaction(): bool;
+
     public function commit(): bool;
 
     public function rollback(): bool;
@@ -21,4 +23,8 @@ public function prepare(string $sql): PDOStatement;
     public function close(): void;
 
     public function connect(): void;
+
+    public function reconnect(): void;
+
+    public function ping(): bool;
 }

packages/database/src/Connection/ConnectionInitializer.php

@@ -11,14 +11,38 @@
 
 final class ConnectionInitializer implements Initializer
 {
+    /** @var Connection[] */
+    private static array $connections = [];
+
     #[Singleton]
     public function initialize(Container $container): Connection
     {
-        $databaseConfig = $container->get(DatabaseConfig::class);
+        $config = $container->get(DatabaseConfig::class);
+        $connectionKey = $this->getConnectionKey($config);
+
+        $connection = $config->usePersistentConnection
+            ? self::$connections[$connectionKey] ?? null
+            : null;
 
-        $connection = new PDOConnection($databaseConfig);
-        $connection->connect();
+        if (! $connection instanceof Connection) {
+            $connection = new PDOConnection($config);
+            $connection->connect();
+            self::$connections[$connectionKey] = $connection;
+        } elseif ($connection->ping() === false) {
+            $connection->reconnect();
+        }
 
         return $connection;
     }
+
+    private function getConnectionKey(DatabaseConfig $config): string
+    {
+        return hash('xxh128', serialize([
+            $config->dsn,
+            $config->username,
+            $config->options,
+            $config->password,
+            $config->tag,
+        ]));
+    }
 }

packages/database/src/Connection/ConnectionReset.php

@@ -0,0 +1,35 @@
+<?php
+
+namespace Tempest\Database\Connection;
+
+use Tempest\Container\Container;
+use Tempest\Container\GenericContainer;
+use Tempest\Container\Resettable;
+use Tempest\Database\Exceptions\CouldNotResetConnection;
+
+final readonly class ConnectionReset implements Resettable
+{
+    public function __construct(
+        private Container $container,
+    ) {}
+
+    public function reset(): void
+    {
+        // Manually looping over the connection singletons so that we can check whether they still have an active transaction
+        if ($this->container instanceof GenericContainer) {
+            $connections = $this->container->getSingletons(Connection::class);
+
+            foreach ($connections as $connection) {
+                if (! $connection instanceof Connection) {
+                    continue;
+                }
+
+                if ($connection->inTransaction()) {
+                    throw new CouldNotResetConnection("There's still an active transaction, make sure to close it before ending the request");
+                }
+            }
+        }
+
+        $this->container->unregister(Connection::class, tagged: true);
+    }
+}

packages/database/src/Connection/PDOConnection.php

@@ -69,6 +69,15 @@ public function prepare(string $sql): PDOStatement
         return $statement;
     }
 
+    public function inTransaction(): bool
+    {
+        if (! $this->pdo instanceof PDO) {
+            return false;
+        }
+
+        return $this->pdo->inTransaction();
+    }
+
     public function ping(): bool
     {
         try {

packages/database/src/DatabaseInitializer.php

@@ -16,8 +16,11 @@
 use Tempest\Reflection\ClassReflector;
 use UnitEnum;
 
-final readonly class DatabaseInitializer implements DynamicInitializer
+final class DatabaseInitializer implements DynamicInitializer
 {
+    /** @var Connection[] */
+    private static array $connections = [];
+
     public function canInitialize(ClassReflector $class, string|UnitEnum|null $tag): bool
     {
         return $class->getType()->matches(Database::class);
@@ -26,26 +29,43 @@ public function canInitialize(ClassReflector $class, string|UnitEnum|null $tag):
     #[Singleton]
     public function initialize(ClassReflector $class, string|UnitEnum|null $tag, Container $container): Database
     {
-        $container->singleton(
-            className: Connection::class,
-            definition: function () use ($tag, $container) {
-                $config = $container->get(DatabaseConfig::class, $tag);
+        $config = $container->get(DatabaseConfig::class, $tag);
+        $connectionKey = $this->getConnectionKey($config);
 
-                $connection = new PDOConnection($config);
-                $connection->connect();
+        $connection = $config->usePersistentConnection
+            ? self::$connections[$connectionKey] ?? null
+            : null;
 
-                return $connection;
-            },
+        if (! $connection) {
+            $connection = new PDOConnection($config);
+            $connection->connect();
+            self::$connections[$connectionKey] = $connection;
+        } elseif ($connection->ping() === false) {
+            $connection->reconnect();
+        }
+
+        $container->singleton(
+            className: Connection::class,
+            definition: $connection,
             tag: $tag,
         );
 
-        $connection = $container->get(Connection::class, $tag);
-
         return new GenericDatabase(
             connection: $connection,
             transactionManager: new GenericTransactionManager($connection),
             serializerFactory: $container->get(SerializerFactory::class),
             eventBus: $container->get(EventBus::class),
         );
     }
+
+    private function getConnectionKey(DatabaseConfig $config): string
+    {
+        return hash('xxh128', serialize([
+            $config->dsn,
+            $config->username,
+            $config->options,
+            $config->password,
+            $config->tag,
+        ]));
+    }
 }

packages/database/src/DatabaseReset.php

@@ -0,0 +1,18 @@
+<?php
+
+namespace Tempest\Database;
+
+use Tempest\Container\Container;
+use Tempest\Container\Resettable;
+
+final readonly class DatabaseReset implements Resettable
+{
+    public function __construct(
+        private Container $container,
+    ) {}
+
+    public function reset(): void
+    {
+        $this->container->unregister(Database::class, tagged: true);
+    }
+}