feat(数据库): 添加多数据库模式支持

引入 Database 枚举类定义支持的数据库类型
修改 Configuration 类添加数据库模式配置
更新 DatabaseController 根据配置选择不同数据库连接方式 #2

Author: ctkqiang

src/main/java/xin/ctkqiang/config/Configuration.java

@@ -1,6 +1,10 @@
 package xin.ctkqiang.config;
 
+import xin.ctkqiang.dto.Database;
+
 public class Configuration {
+    public String DatabaseMode;
+
     public static final String DB_NAME = "ExploitDB";
     public static final String DB_URL = "jdbc:mysql://localhost:3306/" + DB_NAME
             + "?serverTimezone=UTC&allowPublicKeyRetrieval=true&useSSL=false";
@@ -10,4 +14,12 @@ public class Configuration {
     public static final String DB_DRIVER = "com.mysql.cj.jdbc.Driver";
 
     public static final boolean DEBUG = false;
+
+    public String getDatabaseMode() {
+        return this.DatabaseMode;
+    }
+
+    public void setDatabaseMode(Database databaseMode) {
+        this.DatabaseMode = databaseMode.getValue();
+    }
 }

src/main/java/xin/ctkqiang/controller/DatabaseController.java

@@ -18,6 +18,7 @@
 import com.fasterxml.jackson.databind.ObjectMapper;
 
 import xin.ctkqiang.config.Configuration;
+import xin.ctkqiang.dto.Database;
 import xin.ctkqiang.dto.Exploit;
 
 /**
@@ -44,6 +45,8 @@ public class DatabaseController {
     /** 数据库驱动类名 */
     protected final static String DB_DRIVER = Configuration.DB_DRIVER;
 
+    private Configuration Config;
+
     /**
      * 静态初始化块
      * 在类加载时尝试加载数据库驱动
@@ -109,18 +112,41 @@ public void CreateTableIfNotExists() {
                 + ");";
 
         try {
-            // 确保数据库驱动已加载
-            Class.forName(DatabaseController.DB_DRIVER);
-
-            // 创建数据库连接并执行建表语句
-            try (Connection conn = DriverManager.getConnection(
-                    DatabaseController.DB_URL,
-                    DatabaseController.DB_USER,
-                    DatabaseController.DB_PASSWORD);
-                    Statement stmt = conn.createStatement()) {
-                stmt.execute(query);
-                System.out.println("💾 数据表【内容信息】初始化成功！");
+
+            String mode = this.Config.getDatabaseMode();
+            Database dbMode;
+
+            try {
+                dbMode = Database.fromValue(mode);
+            } catch (IllegalArgumentException e) {
+                dbMode = null;
             }
+
+            if (dbMode == null) {
+                System.out.println("⚠️ Unknown database mode: " + mode);
+                return;
+            }
+
+            switch (dbMode) {
+                case SQLITE:
+                    System.out.println("📦 Connect to SQLite database");
+                    break;
+                case MYSQL:
+                default:
+                    // 确保数据库驱动已加载
+                    Class.forName(DatabaseController.DB_DRIVER);
+
+                    // 创建数据库连接并执行建表语句
+                    try (Connection conn = DriverManager.getConnection(
+                            DatabaseController.DB_URL,
+                            DatabaseController.DB_USER,
+                            DatabaseController.DB_PASSWORD);
+                            Statement stmt = conn.createStatement()) {
+                        stmt.execute(query);
+                        System.out.println("💾 数据表【内容信息】初始化成功！");
+                    }
+            }
+
         } catch (Exception e) {
             System.err.println("❌ 数据表创建失败：" + e.getMessage());
             e.printStackTrace();

src/main/java/xin/ctkqiang/dto/Database.java

@@ -0,0 +1,26 @@
+package xin.ctkqiang.dto;
+
+public enum Database {
+    MYSQL("MySQL"),
+    SQLITE("SQLite");
+
+    private final String value;
+
+    Database(String value) {
+        this.value = value;
+    }
+
+    public String getValue() {
+        return value;
+    }
+
+    // 可选: 根据字符串反查 enum（比如传入 "MySQL" 返回 MYSQL 枚举值）
+    public static Database fromValue(String value) {
+        for (Database db : values()) {
+            if (db.value.equalsIgnoreCase(value)) {
+                return db;
+            }
+        }
+        throw new IllegalArgumentException("Unknown database type: " + value);
+    }
+}