Data source

Author: IzzelAliz

src/main/java/io/izzel/netzach/Netzach.java

@@ -1,11 +1,50 @@
 package io.izzel.netzach;
 
+import io.izzel.netzach.config.ConfigSpec;
+import io.izzel.netzach.data.DataManager;
 import io.izzel.taboolib.loader.Plugin;
+import io.izzel.taboolib.module.locale.TLocale;
+import org.yaml.snakeyaml.Yaml;
+import org.yaml.snakeyaml.constructor.CustomClassLoaderConstructor;
 
 public final class Netzach extends Plugin {
 
+    private ConfigSpec configSpec;
+    private DataManager dataManager;
+
+    @SuppressWarnings("deprecation")
+    @Override
+    public void onLoading() {
+        this.saveDefaultConfig();
+        Yaml yaml = new Yaml(new CustomClassLoaderConstructor(ConfigSpec.class, this.getClassLoader()));
+        this.configSpec = yaml.load(this.getConfig().saveToString());
+        this.dataManager = new DataManager();
+    }
+
     @Override
     public void onStarting() {
+        this.dataManager.open();
+        TLocale.sendToConsole("load.complete");
+    }
+
+    @Override
+    public void onStopping() {
+        this.dataManager.close();
+    }
+
+    public DataManager getDataManager() {
+        return dataManager;
+    }
+
+    public ConfigSpec getConfigSpec() {
+        return configSpec;
+    }
+
+    public static Netzach instance() {
+        return (Netzach) plugin;
+    }
 
+    public static ConfigSpec config() {
+        return instance().getConfigSpec();
     }
 }

src/main/java/io/izzel/netzach/config/ConfigSpec.java

@@ -0,0 +1,14 @@
+package io.izzel.netzach.config;
+
+public class ConfigSpec {
+
+    private DatabaseSpec database = new DatabaseSpec();
+
+    public DatabaseSpec database() {
+        return database;
+    }
+
+    public void setDatabase(DatabaseSpec database) {
+        this.database = database;
+    }
+}

src/main/java/io/izzel/netzach/config/DatabaseSpec.java

@@ -0,0 +1,44 @@
+package io.izzel.netzach.config;
+
+import java.util.HashMap;
+import java.util.Map;
+
+public class DatabaseSpec {
+
+    private String tablePrefix = "netzach_";
+    private String jdbcUrl;
+    private Map<String, String> properties = new HashMap<>();
+    private Map<String, String> dataSourceProperties = new HashMap<>();
+
+    public String tablePrefix() {
+        return tablePrefix;
+    }
+
+    public void setTablePrefix(String tablePrefix) {
+        this.tablePrefix = tablePrefix;
+    }
+
+    public String jdbcUrl() {
+        return jdbcUrl;
+    }
+
+    public void setJdbcUrl(String jdbcUrl) {
+        this.jdbcUrl = jdbcUrl;
+    }
+
+    public Map<String, String> properties() {
+        return properties;
+    }
+
+    public void setProperties(Map<String, String> properties) {
+        this.properties = properties;
+    }
+
+    public Map<String, String> dataSourceProperties() {
+        return dataSourceProperties;
+    }
+
+    public void setDataSourceProperties(Map<String, String> dataSourceProperties) {
+        this.dataSourceProperties = dataSourceProperties;
+    }
+}

src/main/java/io/izzel/netzach/data/DataManager.java

@@ -0,0 +1,50 @@
+package io.izzel.netzach.data;
+
+import com.zaxxer.hikari.HikariConfig;
+import com.zaxxer.hikari.HikariDataSource;
+import com.zaxxer.hikari.util.PropertyElf;
+import io.izzel.netzach.Netzach;
+import io.izzel.taboolib.module.locale.TLocale;
+
+import java.sql.Connection;
+import java.sql.SQLException;
+import java.util.Properties;
+
+public class DataManager {
+
+    private HikariDataSource dataSource;
+
+    public void open() {
+        String jdbcUrl = Netzach.config().database().jdbcUrl();
+        HikariConfig hikariConfig = new HikariConfig();
+        hikariConfig.setJdbcUrl(jdbcUrl);
+        hikariConfig.setPoolName("Netzach");
+
+        if (jdbcUrl.startsWith("jdbc:sqlite")) {
+            hikariConfig.setDriverClassName("org.sqlite.JDBC");
+        } else if (jdbcUrl.startsWith("jdbc:mysql")) {
+            hikariConfig.setDriverClassName("com.mysql.jdbc.Driver");
+        }
+
+        Properties properties = new Properties();
+        properties.putAll(Netzach.config().database().properties());
+        PropertyElf.setTargetFromProperties(hikariConfig, properties);
+
+        Properties dataSourceProp = new Properties();
+        dataSourceProp.putAll(Netzach.config().database().dataSourceProperties());
+        hikariConfig.setDataSourceProperties(dataSourceProp);
+        this.dataSource = new HikariDataSource(hikariConfig);
+        TLocale.sendToConsole("load.database");
+    }
+
+    public void close() {
+        if (this.dataSource != null) {
+            this.dataSource.close();
+            this.dataSource = null;
+        }
+    }
+
+    public Connection connection() throws SQLException {
+        return this.dataSource.getConnection();
+    }
+}

src/main/resources/config.yml

@@ -0,0 +1,19 @@
+
+database:
+  tablePrefix: 'netzach_'
+  jdbcUrl: 'jdbc:sqlite:./plugins/Netzach/data.db'
+  #jdbcUrl: 'jdbc:mysql://localhost:3306/database'
+  properties:
+    #username: root
+    #password: pass
+    connectionTimeout: 5000
+    maximumPoolSize: 10
+    maxLifetime: 1800000
+  dataSourceProperties:
+    useUnicode: true
+    characterEncoding: utf8
+    cachePrepStmts: true
+    prepStmtCacheSize: 64
+    prepStmtCacheSqlLimit: 256
+    #useSSL: false
+    #verifyServerCertificate: false

src/main/resources/lang/zh_CN.yml

@@ -0,0 +1,6 @@
+
+load:
+  database: '数据库初始化完成'
+  complete: 'Netzach 加载完成'
+database:
+  unknown_protocol: '&未知的 jdbc 协议 {0}'