chore: update mongodb uri config

jamebal · Mar 19, 2024 · 80db2a7 · 80db2a7
1 parent 9482561
commit 80db2a7
Showing 1 changed file with 17 additions and 1 deletion.
diff --git a/src/main/java/com/jmal/clouddisk/config/MongoClientConfig.java b/src/main/java/com/jmal/clouddisk/config/MongoClientConfig.java
@@ -10,7 +10,9 @@
 import org.springframework.core.env.Environment;
 import org.springframework.data.mongodb.config.AbstractMongoClientConfiguration;
 
+import java.net.InetAddress;
 import java.net.URI;
+import java.net.UnknownHostException;
 
 import static java.util.Collections.singletonList;
 
@@ -43,6 +45,19 @@ public boolean autoIndexCreation() {
     return true;
   }
 
+  private String domainResolver(String domainName) {
+    if (StrUtil.isBlank(domainName)) {
+      return "localhost";
+    }
+    try {
+      InetAddress address = InetAddress.getByName(domainName);
+      return address.getHostAddress();
+    } catch (UnknownHostException e) {
+      log.error(e.getMessage(), e);
+    }
+    return "localhost";
+  }
+
   @Override
   protected void configureClientSettings(@NotNull MongoClientSettings.Builder builder) {
 
@@ -63,7 +78,8 @@ protected void configureClientSettings(@NotNull MongoClientSettings.Builder buil
     String finalHost = host;
     builder.applyToClusterSettings(settings -> {
       settings.hosts(singletonList(
-              new ServerAddress(finalHost, Integer.parseInt(finalPort))));
+              new ServerAddress(domainResolver(finalHost), Integer.parseInt(finalPort))));
     });
   }
+
 }