support multiple hosts

Author: InventivetalentDev

src/main/java/org/inventivetalent/mcauth/DatabaseClient.java

@@ -9,56 +9,70 @@
 
 import java.io.IOException;
 import java.util.Collections;
+import java.util.List;
 
 public class DatabaseClient {
 
-	private String          dbName;
-	private String          host;
-	private int             port;
-	private MongoCredential credential;
+    private List<ServerAddress> hosts;
 
-	private MongoClient   mongoClient;
-	private MongoDatabase mongoDatabase;
+    private String dbName;
+    private String host;
+    private int port;
+    private MongoCredential credential;
 
-	public DatabaseClient(String dbName, String host, int port, String user, char[] pass, String authDatabase) {
-		this.dbName = dbName;
-		this.host = host;
-		this.port = port;
+    private MongoClient mongoClient;
+    private MongoDatabase mongoDatabase;
+
+    public DatabaseClient(String dbName, List<ServerAddress> hosts, String user, char[] pass, String authDatabase) {
+        this.dbName = dbName;
+        this.hosts = hosts;
 		this.credential = MongoCredential.createScramSha1Credential(user, authDatabase, pass);
-	}
+    }
+
+    public DatabaseClient(String dbName, String host, int port, String user, char[] pass, String authDatabase) {
+        this.dbName = dbName;
+        this.host = host;
+        this.port = port;
+        this.credential = MongoCredential.createScramSha1Credential(user, authDatabase, pass);
+    }
 
-	public int collectionCount() {
-		int c = 0;
-		for (String ignored : db().listCollectionNames()) {
-			c++;
-		}
-		return c;
-	}
+    public int collectionCount() {
+        int c = 0;
+        for (String ignored : db().listCollectionNames()) {
+            c++;
+        }
+        return c;
+    }
 
-	public ServerAddress connect(int timeout) throws IOException {
-		if (mongoClient == null) {
-			System.out.println("Connecting to MongoDB " + this.host + ":" + this.port + "...");
-			mongoClient = new MongoClient(new ServerAddress(this.host, this.port), Collections.singletonList(this.credential), MongoClientOptions.builder().connectTimeout(timeout).build());
-		}
-		return mongoClient.getAddress();
-	}
+    public ServerAddress connect(int timeout) throws IOException {
+        if (mongoClient == null) {
+            if (hosts != null && !hosts.isEmpty()) {
+                System.out.println("Connecting to MongoDB...");
+                mongoClient = new MongoClient(this.hosts, this.credential, MongoClientOptions.builder().connectTimeout(timeout).build());
+            } else {
+                System.out.println("Connecting to MongoDB " + this.host + ":" + this.port + "...");
+                mongoClient = new MongoClient(new ServerAddress(this.host, this.port), Collections.singletonList(this.credential), MongoClientOptions.builder().connectTimeout(timeout).build());
+            }
+        }
+        return mongoClient.getAddress();
+    }
 
-	public void disconnect() throws IOException {
-		if (mongoClient != null) {
-			mongoClient.close();
-		}
-	}
+    public void disconnect() throws IOException {
+        if (mongoClient != null) {
+            mongoClient.close();
+        }
+    }
 
-	public MongoDatabase db() {
-		if (mongoDatabase == null) {
-			System.out.println("Initializing database '" + dbName + "'");
-			mongoDatabase = mongoClient.getDatabase(dbName);
-		}
-		return mongoDatabase;
-	}
+    public MongoDatabase db() {
+        if (mongoDatabase == null) {
+            System.out.println("Initializing database '" + dbName + "'");
+            mongoDatabase = mongoClient.getDatabase(dbName);
+        }
+        return mongoDatabase;
+    }
 
-	public Jongo jongo() {
-		return new Jongo(mongoClient.getDB(dbName));
-	}
+    public Jongo jongo() {
+        return new Jongo(mongoClient.getDB(dbName));
+    }
 
 }

src/main/java/org/inventivetalent/mcauth/MCAuthServer.java

@@ -1,5 +1,6 @@
 package org.inventivetalent.mcauth;
 
+import com.mongodb.ServerAddress;
 import net.md_5.bungee.api.ProxyServer;
 import net.md_5.bungee.api.ServerPing;
 import net.md_5.bungee.api.chat.TextComponent;
@@ -20,6 +21,8 @@
 import java.io.IOException;
 import java.io.InputStream;
 import java.nio.file.Files;
+import java.util.ArrayList;
+import java.util.List;
 import java.util.Random;
 
 public class MCAuthServer extends Plugin implements Listener {
@@ -52,12 +55,21 @@ public void onEnable() {
 			throw new RuntimeException("Failed to load config", e);
 		}
 
-		databaseClient = new DatabaseClient(config.getString("mongodb.database"), config.getString("mongodb.host"), config.getInt("mongodb.port"), config.getString("mongodb.login.user"), config.getString("mongodb.login.pass").toCharArray(), config.getString("mongodb.login.db"));
+		if (config.contains("mongodb.hosts")) {
+			List<ServerAddress> hosts = new ArrayList<>();
+			for (String s : config.getStringList("mongodb.hosts")) {
+				String[] split = s.split(":");
+				hosts.add(new ServerAddress(split[0], split.length > 1 ? Integer.parseInt(split[1]) : ServerAddress.defaultPort()));
+			}
+			databaseClient = new DatabaseClient(config.getString("mongodb.database"), hosts, config.getString("mongodb.login.user"), config.getString("mongodb.login.pass").toCharArray(), config.getString("mongodb.login.db"));
+		} else {
+			databaseClient = new DatabaseClient(config.getString("mongodb.database"), config.getString("mongodb.host"), config.getInt("mongodb.port"), config.getString("mongodb.login.user"), config.getString("mongodb.login.pass").toCharArray(), config.getString("mongodb.login.db"));
+		}
 		try {
 			databaseClient.connect(10000);
 			databaseClient.collectionCount();
 		} catch (IOException e) {
-			throw new RuntimeException("Failed to connect to databse", e);
+			throw new RuntimeException("Failed to connect to database", e);
 		}
 
 		jongo = databaseClient.jongo();

src/main/resources/bungee.yml

@@ -1,4 +1,4 @@
 name: MCAuth-Server
 main: org.inventivetalent.mcauth.MCAuthServer
-version: 1.0.0
-author: inventivetalent
+version: 1.1.0
+author: inventivetalent