fix(auth-service): ping to establish db connection

codingben · May 6, 2023 · af31304 · af31304
1 parent 0ab6dce
commit af31304
Showing 1 changed file with 21 additions and 2 deletions.
diff --git a/src/auth-service/Authenticator.Infrastructure/MongoRepository/MongoDatabaseProvider.cs b/src/auth-service/Authenticator.Infrastructure/MongoRepository/MongoDatabaseProvider.cs
@@ -1,4 +1,6 @@
-using MongoDB.Driver;
+using System;
+using MongoDB.Driver;
+using MongoDB.Bson;
 
 namespace Authenticator.Infrastructure.MongoRepository
 {
@@ -13,8 +15,25 @@ public MongoDatabaseProvider(string url)
         {
             var mongoUrl = new MongoUrl(url);
             var mongoClient = new MongoClient(mongoUrl);
-
             database = mongoClient.GetDatabase(mongoUrl.DatabaseName);
+
+            Ping(url);
+        }
+
+        /// <summary>
+        /// Force the connection to be established.
+        /// </summary>
+        private void Ping(string url)
+        {
+            try
+            {
+                var pingCommand = new BsonDocument("ping", 1);
+                database.RunCommand<BsonDocument>(pingCommand);
+            }
+            catch (Exception exception)
+            {
+                throw new Exception($"Failed to connect to MongoDB server at {url}. {exception.Message}", exception);
+            }
         }
 
         public IMongoDatabase Provide()