feat: Add ability to close all connections attached to a driver

Author: exaby73

src/Basic/Driver.php

@@ -57,4 +57,9 @@ public static function create(string|UriInterface $uri, ?DriverConfiguration $co
 
         return new self($driver);
     }
+
+    public function closeConnections(): void
+    {
+        $this->driver->closeConnections();
+    }
 }

src/Bolt/BoltDriver.php

@@ -109,4 +109,9 @@ public function verifyConnectivity(?SessionConfiguration $config = null): bool
 
         return true;
     }
+
+    public function closeConnections(): void
+    {
+        $this->pool->close();
+    }
 }

src/Bolt/ConnectionPool.php

@@ -46,8 +46,12 @@ public function __construct(
         private readonly ?Neo4jLogger $logger
     ) {}
 
-    public static function create(UriInterface $uri, AuthenticateInterface $auth, DriverConfiguration $conf, SemaphoreInterface $semaphore): self
-    {
+    public static function create(
+        UriInterface $uri,
+        AuthenticateInterface $auth,
+        DriverConfiguration $conf,
+        SemaphoreInterface $semaphore
+    ): self {
         return new self(
             $semaphore,
             BoltFactory::create($conf->getLogger()),
@@ -165,4 +169,9 @@ private function returnAnyAvailableConnection(SessionConfiguration $config): ?Co
 
         return null;
     }
+
+    public function close(): void
+    {
+        $this->activeConnections = [];
+    }
 }

src/Contracts/ConnectionPoolInterface.php

@@ -44,4 +44,9 @@ public function acquire(SessionConfiguration $config): Generator;
      * Releases a connection back to the pool.
      */
     public function release(ConnectionInterface $connection): void;
+
+    /**
+     * Closes all connections in the pool.
+     */
+    public function close(): void;
 }

src/Contracts/DriverInterface.php

@@ -38,4 +38,9 @@ public function createSession(?SessionConfiguration $config = null): SessionInte
      * Returns true if the driver can make a valid connection with the server.
      */
     public function verifyConnectivity(?SessionConfiguration $config = null): bool;
+
+    /**
+     * Closes all connections in the pool.
+     */
+    public function closeConnections(): void;
 }

src/Http/HttpConnectionPool.php

@@ -127,4 +127,9 @@ public function release(ConnectionInterface $connection): void
     {
         // Nothing to release in the current HTTP Protocol implementation
     }
+
+    public function close(): void
+    {
+        // Nothing to close in the current HTTP Protocol implementation
+    }
 }

src/Http/HttpDriver.php

@@ -198,4 +198,9 @@ private function tsxUrl(SessionConfiguration $config): Resolvable
             return str_replace('{databaseName}', $database, $tsx);
         });
     }
+
+    public function closeConnections(): void
+    {
+        // Nothing to close in the current HTTP Protocol implementation
+    }
 }

src/Neo4j/Neo4jConnectionPool.php

@@ -233,4 +233,13 @@ private function createKey(ConnectionRequestData $data, ?SessionConfiguration $c
             ':',
         ], '|', $key);
     }
+
+    public function close(): void
+    {
+        foreach (self::$pools as $pool) {
+            $pool->close();
+        }
+        self::$pools = [];
+        $this->cache->clear();
+    }
 }

src/Neo4j/Neo4jDriver.php

@@ -111,4 +111,9 @@ public function verifyConnectivity(?SessionConfiguration $config = null): bool
 
         return true;
     }
+
+    public function closeConnections(): void
+    {
+        $this->pool->close();
+    }
 }