close conn if auth fails (#144)

vapor · Sep 23, 2019 · 2cdd00a · 2cdd00a
1 parent 74d49fd
commit 2cdd00a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/Sources/PostgresKit/PostgresConnectionSource.swift b/Sources/PostgresKit/PostgresConnectionSource.swift
@@ -23,7 +23,10 @@ public struct PostgresConnectionSource: ConnectionPoolSource {
                 username: self.configuration.username,
                 database: self.configuration.database,
                 password: self.configuration.password
-            ).map { conn }
+            ).flatMapError { error in
+                return conn.close()
+                    .flatMapThrowing { throw error }
+            }.map { conn }
         }
     }
 }