Make ConnectionPool faster (#546)

Author: fabianfett

Sources/ConnectionPoolModule/ConnectionRequest.swift

@@ -21,7 +21,8 @@ public struct ConnectionRequest<Connection: PooledConnection>: ConnectionRequest
     }
 }
 
-fileprivate let requestIDGenerator = _ConnectionPoolModule.ConnectionIDGenerator()
+@usableFromInline
+let requestIDGenerator = _ConnectionPoolModule.ConnectionIDGenerator()
 
 @available(macOS 13.0, iOS 16.0, tvOS 16.0, watchOS 9.0, *)
 extension ConnectionPool where Request == ConnectionRequest<Connection> {
@@ -44,6 +45,7 @@ extension ConnectionPool where Request == ConnectionRequest<Connection> {
         )
     }
 
+    @inlinable
     public func leaseConnection() async throws -> Connection {
         let requestID = requestIDGenerator.next()
 
@@ -67,6 +69,7 @@ extension ConnectionPool where Request == ConnectionRequest<Connection> {
         return connection
     }
 
+    @inlinable
     public func withConnection<Result>(_ closure: (Connection) async throws -> Result) async throws -> Result {
         let connection = try await self.leaseConnection()
         defer { self.releaseConnection(connection) }

Sources/ConnectionPoolModule/PoolStateMachine+ConnectionGroup.swift

@@ -132,6 +132,12 @@ extension PoolStateMachine {
 
             @usableFromInline
             var info: ConnectionAvailableInfo
+
+            @inlinable
+            init(use: ConnectionUse, info: ConnectionAvailableInfo) {
+                self.use = use
+                self.info = info
+            }
         }
 
         mutating func refillConnections() -> [ConnectionRequest] {
@@ -623,7 +629,7 @@ extension PoolStateMachine {
 
         // MARK: - Private functions -
 
-        @usableFromInline
+        @inlinable
         /*private*/ func getConnectionUse(index: Int) -> ConnectionUse {
             switch index {
             case 0..<self.minimumConcurrentConnections:
@@ -637,7 +643,7 @@ extension PoolStateMachine {
             }
         }
 
-        @usableFromInline
+        @inlinable
         /*private*/ func makeAvailableConnectionContextForConnection(at index: Int, info: ConnectionAvailableInfo) -> AvailableConnectionContext {
             precondition(self.connections[index].isAvailable)
             let use = self.getConnectionUse(index: index)

Sources/ConnectionPoolModule/PoolStateMachine+ConnectionState.swift

@@ -164,7 +164,7 @@ extension PoolStateMachine {
             }
         }
 
-        @usableFromInline
+        @inlinable
         var isLeased: Bool {
             switch self.state {
             case .leased:
@@ -174,7 +174,7 @@ extension PoolStateMachine {
             }
         }
 
-        @usableFromInline
+        @inlinable
         var isConnected: Bool {
             switch self.state {
             case .idle, .leased:

Sources/ConnectionPoolTestUtils/MockRequest.swift

@@ -3,7 +3,7 @@ import _ConnectionPoolModule
 public final class MockRequest: ConnectionRequestProtocol, Hashable, Sendable {
     public typealias Connection = MockConnection
 
-    public struct ID: Hashable {
+    public struct ID: Hashable, Sendable {
         var objectID: ObjectIdentifier
 
         init(_ request: MockRequest) {