fix: refactor func

LLCFreedom-Space · Jun 5, 2024 · 36d9b5c · 36d9b5c
1 parent da5e558
commit 36d9b5c
Showing 1 changed file with 5 additions and 9 deletions.
diff --git a/Sources/HealthChecks/MongoHealthChecks/MongoRequest.swift b/Sources/HealthChecks/MongoHealthChecks/MongoRequest.swift
@@ -41,14 +41,10 @@ public final class MongoRequest: MongoRequestSendable {
     /// - Parameter url: `String`
     /// - Returns: `String`
     public func getConnection(by url: String) async throws -> String {
-        if "\(app.mongoCluster?.connectionState)" == "disconnect" || "\(app.mongoCluster?.connectionState)" == "closed" || "\(app.mongoCluster?.connectionState)" == "nil" {
-        app.mongoCluster = try await MongoCluster(connectingTo: ConnectionSettings(url))
-            let connection = "\(app.mongoCluster?.connectionState ?? .disconnected)"
-            app.logger.info("--------------\(app.mongoCluster?.connectionState)")
-            return connection
-        } else {
-            app.logger.info("-----disconnect---------\(app.mongoCluster?.connectionState)")
-            return "disconnect"
-        }
+        await app.mongoCluster?.disconnect()
+        app.mongoCluster = try? MongoCluster(lazyConnectingTo: ConnectionSettings(url))
+        let connection = "\(app.mongoCluster?.connectionState ?? .disconnected)"
+        await app.mongoCluster?.disconnect()
+        return connection
     }
 }