feat: ✨ load serverinfo from database

Author: AnzhiZhang

src/main/java/org/bsrserver/EventListener.java

@@ -5,6 +5,7 @@
 import java.net.URL;
 import java.net.HttpURLConnection;
 import java.util.ArrayList;
+import java.util.HashMap;
 import javax.xml.parsers.DocumentBuilderFactory;
 
 import com.velocitypowered.api.event.Subscribe;
@@ -20,10 +21,19 @@
 import org.w3c.dom.NodeList;
 
 public class EventListener {
-    ProxyServer proxyServer;
+    private final ProxyServer proxyServer;
+    private final HashMap<String, ServerInfo> serverInfoHashMap;
 
-    private int getOpenDays() {
-        return (int) ChronoUnit.DAYS.between(LocalDate.of(2021, 1, 10), LocalDate.now());
+    private String getOpenDays(RegisteredServer server) {
+        String serverName = server.getServerInfo().getName();
+        ServerInfo serverInfo = serverInfoHashMap.get(serverName);
+        String namedName = serverName;
+        int daysBetween = 0;
+        if (serverInfo != null) {
+            namedName = serverInfo.namedName() != null ? serverInfo.namedName() : namedName;
+            daysBetween = (int) ChronoUnit.DAYS.between(serverInfo.foundationTime().toLocalDate(), LocalDate.now());
+        }
+        return "这是 " + namedName + " 开服的第 " + daysBetween + " 天\n\n";
     }
 
     private String getSentence() {
@@ -81,8 +91,9 @@ private Component getServerList(RegisteredServer server) {
         return Component.join(JoinConfiguration.separator(Component.text(" ")), components);
     }
 
-    public EventListener(ProxyServer proxyServer) {
-        this.proxyServer = proxyServer;
+    public EventListener(Main main) {
+        this.proxyServer = main.getProxyServer();
+        this.serverInfoHashMap = main.getServerInfoHashMap();
     }
 
     @Subscribe
@@ -91,7 +102,7 @@ public void onServerConnectedEvent(ServerConnectedEvent event) {
             Component message = Component.text("-".repeat(40) + "\n")
                     .append(Component.text("§e§l" + event.getPlayer().getUsername()))
                     .append(Component.text("§r, 欢迎回到 §bBSR 服务器§r！\n"))
-                    .append(Component.text("这是 BSR 服务器开服的第 " + getOpenDays() + " 天\n\n"))
+                    .append(Component.text(getOpenDays(event.getServer())))
                     .append(Component.text("[§a一言§r] " + getSentence() + "\n\n"))
                     .append(getServerList(event.getServer()))
                     .append(Component.text("\n"))

src/main/java/org/bsrserver/Main.java

@@ -1,10 +1,16 @@
 package org.bsrserver;
 
+import java.sql.*;
+import java.nio.file.Path;
+import java.util.HashMap;
+
+import org.slf4j.Logger;
 import com.google.inject.Inject;
 import com.velocitypowered.api.event.proxy.ProxyInitializeEvent;
 import com.velocitypowered.api.plugin.Plugin;
 import com.velocitypowered.api.event.Subscribe;
 import com.velocitypowered.api.proxy.ProxyServer;
+import com.velocitypowered.api.plugin.annotation.DataDirectory;
 
 @Plugin(
         id = "bsrgreeter",
@@ -16,14 +22,64 @@
 )
 public class Main {
     private final ProxyServer proxyServer;
+    private final Logger logger;
+    private final Path dataDirectory;
+    private final HashMap<String, ServerInfo> serverInfoHashMap = new HashMap<>();
 
     @Inject
-    public Main(ProxyServer proxyServer) {
+    public Main(ProxyServer proxyServer, Logger logger, @DataDirectory Path dataDirectory) {
         this.proxyServer = proxyServer;
+        this.logger = logger;
+        this.dataDirectory = dataDirectory;
     }
 
     @Subscribe
     public void onInitialize(ProxyInitializeEvent event) {
-        proxyServer.getEventManager().register(this, new EventListener(proxyServer));
+        // load config
+        Config.getInstance().loadConfig(dataDirectory);
+
+        // read database
+        try {
+            // connect
+            Class.forName("org.postgresql.Driver");
+            Connection connection = DriverManager.getConnection(
+                    Config.getInstance().getDatabaseUrl(),
+                    Config.getInstance().getDatabaseUser(),
+                    Config.getInstance().getDatabasePassword()
+            );
+            logger.info("Successfully connected to database");
+
+            // select
+            String tableName = Config.getInstance().getDatabaseTable();
+            Statement statement = connection.createStatement();
+            ResultSet resultSet = statement.executeQuery("SELECT * FROM " + tableName);
+            while (resultSet.next()) {
+                String serverName = resultSet.getString("server_name");
+                ServerInfo serverInfo = new ServerInfo(
+                        serverName,
+                        resultSet.getString("named_name"),
+                        resultSet.getDate("foundation_time")
+                );
+                serverInfoHashMap.put(serverName, serverInfo);
+            }
+            resultSet.close();
+            statement.close();
+            connection.close();
+            logger.info("Loaded servers: " + serverInfoHashMap.keySet());
+        } catch (Exception e) {
+            e.printStackTrace();
+            logger.error("Fail connect to database");
+        }
+
+        // register command
+        proxyServer.getEventManager().register(this, new EventListener(this));
+    }
+
+    public ProxyServer getProxyServer() {
+        return proxyServer;
+    }
+
+    public HashMap<String, ServerInfo> getServerInfoHashMap() {
+        return serverInfoHashMap;
     }
 }

src/main/java/org/bsrserver/ServerInfo.java

@@ -0,0 +1,18 @@
+package org.bsrserver;
+
+import java.sql.Date;
+
+public record ServerInfo(
+        String serverName,
+        String namedName,
+        Date foundationTime
+) {
+    @Override
+    public String toString() {
+        return "ServerInfo{" +
+                "serverName='" + serverName + '\'' +
+                ", namedName='" + namedName + '\'' +
+                ", foundationTime=" + foundationTime +
+                '}';
+    }
+}