Add SQLite support

Signed-off-by: applenick <[email protected]>
PGMDev · Jun 12, 2024 · af895c7 · af895c7
1 parent 384d715
commit af895c7
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 17 deletions.
diff --git a/src/main/java/dev/pgm/community/database/DatabaseConfig.java b/src/main/java/dev/pgm/community/database/DatabaseConfig.java
@@ -11,6 +11,7 @@ public class DatabaseConfig {
   private String databaseName;
   private String timezone;
   private int maxConnections;
+  private String sqliteFileName;
 
   public DatabaseConfig(Configuration config) {
     reload(config);
@@ -24,6 +25,7 @@ public void reload(Configuration config) {
     this.databaseName = config.getString("database.databaseName");
     this.timezone = config.getString("database.timezone");
     this.maxConnections = config.getInt("database.max-connections");
+    this.sqliteFileName = config.getString("database.sqlite-file");
   }
 
   public boolean isEnabled() {
@@ -51,6 +53,10 @@ public String getTimezone() {
   }
 
   public int getMaxDatabaseConnections() {
-    return maxConnections;
+    return isEnabled() ? maxConnections : 1;
+  }
+
+  public String getSQLiteFileName() {
+    return sqliteFileName;
   }
 }
diff --git a/src/main/java/dev/pgm/community/database/DatabaseConnection.java b/src/main/java/dev/pgm/community/database/DatabaseConnection.java
@@ -2,7 +2,9 @@
 
 import co.aikar.idb.BukkitDB;
 import co.aikar.idb.DatabaseOptions;
+import co.aikar.idb.DatabaseOptions.DatabaseOptionsBuilder;
 import co.aikar.idb.PooledDatabaseOptions;
+import co.aikar.idb.PooledDatabaseOptions.PooledDatabaseOptionsBuilder;
 import com.google.common.collect.Maps;
 import dev.pgm.community.Community;
 import java.util.Map;
@@ -17,25 +19,29 @@ public DatabaseConnection(Community plugin) {
     Map<String, Object> extraOptions = Maps.newHashMap();
     extraOptions.put("serverTimezone", config.getTimezone());
 
-    DatabaseOptions options =
+    DatabaseOptionsBuilder builder =
         DatabaseOptions.builder()
             .poolName(plugin.getDescription().getName() + " DB")
-            .logger(plugin.getLogger())
-            .mysql(
-                config.getUsername(),
-                config.getPassword(),
-                config.getDatabaseName(),
-                config.getHost())
-            .build();
-
-    PooledDatabaseOptions poolOptions =
+            .logger(plugin.getLogger());
+
+    if (config.isEnabled()) {
+      builder.mysql(
+          config.getUsername(), config.getPassword(), config.getDatabaseName(), config.getHost());
+    } else {
+      builder.sqlite(config.getSQLiteFileName());
+    }
+
+    PooledDatabaseOptionsBuilder poolBuilder =
         PooledDatabaseOptions.builder()
-            .options(options)
-            .maxConnections(config.getMaxDatabaseConnections())
-            .dataSourceProperties(extraOptions)
-            .build();
+            .options(builder.build())
+            .maxConnections(config.getMaxDatabaseConnections());
+
+    // Apply extra MySQL options
+    if (config.isEnabled()) {
+      poolBuilder.dataSourceProperties(extraOptions);
+    }
 
     // Setup the main global DB
-    BukkitDB.createHikariDatabase(plugin, poolOptions);
+    BukkitDB.createHikariDatabase(plugin, poolBuilder.build());
   }
 }
diff --git a/src/main/resources/config.yml b/src/main/resources/config.yml
@@ -318,7 +318,8 @@ network:
 
 # Database connection info 
 database:
-  enabled: true # True for mysql, false for sqlite
+  enabled: false # True for mysql, false for sqlite
+  sqlite-file: "community.sql"
   username: "username"
   password: "password"
   databaseName: "minecraft"