MiniDay/HamsterCurrency-Parent
1
0
Fork 0
mirror of https://github.com/MiniDay/HamsterCurrency-Parent.git synced 2025-08-24 13:15:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

使用 HikariCP 作为数据库连接池

Browse Source
This commit is contained in:
叁只仓鼠
2022-01-15 12:54:51 +08:00
parent 870a3789f2
commit eac536a971
3 changed files with 63 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
currency-plugin/build.gradle
View File

@@ -3,7 +3,7 @@ plugins {
} }
group 'cn.hamster3' group 'cn.hamster3'
version '2.0.8-SNAPSHOT' version '2.0.9-SNAPSHOT'
repositories { repositories {
maven { maven {

91
currency-plugin/src/main/java/cn/hamster3/currency/core/SQLDataManager.java
View File

@@ -1,11 +1,12 @@
package cn.hamster3.currency.core; package cn.hamster3.currency.core;
import cn.hamster3.api.HamsterAPI;
import cn.hamster3.currency.HamsterCurrency; import cn.hamster3.currency.HamsterCurrency;
import cn.hamster3.currency.data.CurrencyType; import cn.hamster3.currency.data.CurrencyType;
import cn.hamster3.currency.data.PlayerData; import cn.hamster3.currency.data.PlayerData;
import cn.hamster3.service.bukkit.api.ServiceMessageAPI; import cn.hamster3.service.bukkit.api.ServiceMessageAPI;
import com.google.gson.JsonParser; import com.google.gson.JsonParser;
import com.zaxxer.hikari.HikariConfig;
import com.zaxxer.hikari.HikariDataSource;
import org.bukkit.Bukkit; import org.bukkit.Bukkit;
import org.bukkit.configuration.ConfigurationSection; import org.bukkit.configuration.ConfigurationSection;
import org.bukkit.configuration.InvalidConfigurationException; import org.bukkit.configuration.InvalidConfigurationException;
@@ -22,9 +23,12 @@ import java.util.*;
import static cn.hamster3.currency.HamsterCurrency.getLogUtils; import static cn.hamster3.currency.HamsterCurrency.getLogUtils;
public class SQLDataManager implements IDataManager { public class SQLDataManager implements IDataManager {
private final JsonParser parser;
private final Connection connection;
private final HamsterCurrency plugin; private final HamsterCurrency plugin;
private final JsonParser parser;
private final String database;
private final HikariDataSource dataSource;
private final HashSet<PlayerData> playerData; private final HashSet<PlayerData> playerData;
private final HashSet<CurrencyType> currencyTypes; private final HashSet<CurrencyType> currencyTypes;
@@ -34,18 +38,32 @@ public class SQLDataManager implements IDataManager {
playerData = new HashSet<>(); playerData = new HashSet<>();
currencyTypes = new HashSet<>(); currencyTypes = new HashSet<>();
connection = HamsterAPI.getSQLConnection(FileManager.getPluginConfig().getConfigurationSection("datasource")); ConfigurationSection datasourceConfig = FileManager.getPluginConfig().getConfigurationSection("datasource");
database = datasourceConfig.getString("database");
HikariConfig hikariConfig = new HikariConfig();
hikariConfig.setDriverClassName(datasourceConfig.getString("driver"));
hikariConfig.setJdbcUrl(datasourceConfig.getString("url"));
hikariConfig.setUsername(datasourceConfig.getString("user"));
hikariConfig.setPassword(datasourceConfig.getString("password"));
hikariConfig.setMaximumPoolSize(10);
hikariConfig.setMinimumIdle(1);
dataSource = new HikariDataSource(hikariConfig);
Connection connection = dataSource.getConnection();
Statement statement = connection.createStatement(); Statement statement = connection.createStatement();
statement.execute("CREATE TABLE IF NOT EXISTS hamster_currency_player_data(" + statement.execute("CREATE TABLE IF NOT EXISTS " + database + ".hamster_currency_player_data(" +
"uuid VARCHAR(36) PRIMARY KEY," + "uuid VARCHAR(36) PRIMARY KEY," +
"data TEXT" + "data TEXT" +
") CHARACTER SET = utf8mb4;"); ") CHARACTER SET = utf8mb4;");
statement.execute("CREATE TABLE IF NOT EXISTS hamster_currency_settings(" + statement.execute("CREATE TABLE IF NOT EXISTS " + database + ".hamster_currency_settings(" +
"title VARCHAR(64) PRIMARY KEY," + "title VARCHAR(64) PRIMARY KEY," +
"data TEXT" + "data TEXT" +
") CHARACTER SET = utf8mb4;"); ") CHARACTER SET = utf8mb4;");
statement.close(); statement.close();
connection.close();
} }
public void uploadConfigToSQL() { public void uploadConfigToSQL() {
@@ -55,14 +73,16 @@ public class SQLDataManager implements IDataManager {
getLogUtils().info("配置文件重载完成!"); getLogUtils().info("配置文件重载完成!");
try { try {
getLogUtils().info("将配置文件上传至数据库..."); getLogUtils().info("将配置文件上传至数据库...");
Connection connection = dataSource.getConnection();
Statement statement = connection.createStatement(); Statement statement = connection.createStatement();
String data = Base64.getEncoder().encodeToString(config.saveToString().getBytes(StandardCharsets.UTF_8)); String data = Base64.getEncoder().encodeToString(config.saveToString().getBytes(StandardCharsets.UTF_8));
statement.executeUpdate(String.format( statement.executeUpdate(String.format(
"REPLACE INTO hamster_currency_settings VALUES('%s', '%s');", "REPLACE INTO " + database + ".hamster_currency_settings VALUES('%s', '%s');",
"pluginConfig", "pluginConfig",
data data
)); ));
statement.close(); statement.close();
connection.close();
getLogUtils().info("配置文件上传完成!"); getLogUtils().info("配置文件上传完成!");
} catch (SQLException e) { } catch (SQLException e) {
getLogUtils().error(e, "插件上传 pluginConfig 至数据库时遇到了一个异常: "); getLogUtils().error(e, "插件上传 pluginConfig 至数据库时遇到了一个异常: ");
@@ -75,8 +95,9 @@ public class SQLDataManager implements IDataManager {
public void loadConfigFromSQL() { public void loadConfigFromSQL() {
try { try {
getLogUtils().info("从数据库中下载配置文件..."); getLogUtils().info("从数据库中下载配置文件...");
Connection connection = dataSource.getConnection();
Statement statement = connection.createStatement(); Statement statement = connection.createStatement();
ResultSet set = statement.executeQuery("SELECT * FROM hamster_currency_settings;"); ResultSet set = statement.executeQuery("SELECT * FROM " + database + ".hamster_currency_settings;");
while (set.next()) { while (set.next()) {
String title = set.getString("title"); String title = set.getString("title");
String data = new String(Base64.getDecoder().decode(set.getString("data")), StandardCharsets.UTF_8); String data = new String(Base64.getDecoder().decode(set.getString("data")), StandardCharsets.UTF_8);
@@ -93,6 +114,7 @@ public class SQLDataManager implements IDataManager {
} }
} }
statement.close(); statement.close();
connection.close();
getLogUtils().info("配置文件下载完成!"); getLogUtils().info("配置文件下载完成!");
} catch (SQLException e) { } catch (SQLException e) {
getLogUtils().error(e, "插件从数据库中下载 pluginConfig 时遇到了一个异常: "); getLogUtils().error(e, "插件从数据库中下载 pluginConfig 时遇到了一个异常: ");
@@ -125,6 +147,7 @@ public class SQLDataManager implements IDataManager {
getLogUtils().info("玩家经济列名: %s", moneyCol); getLogUtils().info("玩家经济列名: %s", moneyCol);
getLogUtils().info("导入至经济类型: %s", currencyType); getLogUtils().info("导入至经济类型: %s", currencyType);
try { try {
Connection connection = dataSource.getConnection();
Statement statement = connection.createStatement(); Statement statement = connection.createStatement();
ResultSet set = statement.executeQuery(String.format("SELECT * FROM %s.%s;", database, table)); ResultSet set = statement.executeQuery(String.format("SELECT * FROM %s.%s;", database, table));
synchronized (playerData) { synchronized (playerData) {
@@ -147,7 +170,7 @@ public class SQLDataManager implements IDataManager {
} }
for (PlayerData data : playerData) { for (PlayerData data : playerData) {
statement.executeUpdate(String.format( statement.executeUpdate(String.format(
"REPLACE INTO hamster_currency_player_data VALUES('%s', '%s');", "REPLACE INTO " + database + ".hamster_currency_player_data VALUES('%s', '%s');",
data.getUuid().toString(), data.getUuid().toString(),
data.saveToJson().toString() data.saveToJson().toString()
)); ));
@@ -155,6 +178,7 @@ public class SQLDataManager implements IDataManager {
ServiceMessageAPI.sendServiceMessage("HamsterCurrency", "savedPlayerData", data.getUuid().toString()); ServiceMessageAPI.sendServiceMessage("HamsterCurrency", "savedPlayerData", data.getUuid().toString());
} }
statement.close(); statement.close();
connection.close();
} catch (SQLException e) { } catch (SQLException e) {
getLogUtils().error(e, "从其他插件中导入数据时发生了一个异常:"); getLogUtils().error(e, "从其他插件中导入数据时发生了一个异常:");
} }
@@ -165,8 +189,9 @@ public class SQLDataManager implements IDataManager {
public void onEnable() { public void onEnable() {
getLogUtils().info("从数据库中读取玩家数据..."); getLogUtils().info("从数据库中读取玩家数据...");
try { try {
Connection connection = dataSource.getConnection();
Statement statement = connection.createStatement(); Statement statement = connection.createStatement();
ResultSet set = statement.executeQuery("SELECT * FROM hamster_currency_player_data;"); ResultSet set = statement.executeQuery("SELECT * FROM " + database + ".hamster_currency_player_data;");
synchronized (playerData) { synchronized (playerData) {
while (set.next()) { while (set.next()) {
String uuid = set.getString("uuid"); String uuid = set.getString("uuid");
@@ -181,6 +206,7 @@ public class SQLDataManager implements IDataManager {
} }
set.close(); set.close();
statement.close(); statement.close();
connection.close();
} catch (SQLException e) { } catch (SQLException e) {
getLogUtils().error(e, "从数据库中读取玩家数据时出现了一个异常:"); getLogUtils().error(e, "从数据库中读取玩家数据时出现了一个异常:");
} }
@@ -211,9 +237,10 @@ public class SQLDataManager implements IDataManager {
@Override @Override
public void loadPlayerData(UUID uuid) { public void loadPlayerData(UUID uuid) {
try { try {
Connection connection = dataSource.getConnection();
Statement statement = connection.createStatement(); Statement statement = connection.createStatement();
ResultSet set = statement.executeQuery(String.format( ResultSet set = statement.executeQuery(String.format(
"SELECT * FROM hamster_currency_player_data WHERE uuid='%s';", "SELECT * FROM " + database + ".hamster_currency_player_data WHERE uuid='%s';",
uuid uuid
)); ));
PlayerData data; PlayerData data;
@@ -236,6 +263,7 @@ public class SQLDataManager implements IDataManager {
} }
set.close(); set.close();
statement.close(); statement.close();
connection.close();
getLogUtils().info("已加载玩家 %s 的存档数据.", data.getUuid()); getLogUtils().info("已加载玩家 %s 的存档数据.", data.getUuid());
} catch (SQLException e) { } catch (SQLException e) {
getLogUtils().error(e, "加载玩家 %s 的存档数据时出错!", uuid); getLogUtils().error(e, "加载玩家 %s 的存档数据时出错!", uuid);
@@ -244,26 +272,27 @@ public class SQLDataManager implements IDataManager {
@Override @Override
public void savePlayerData(PlayerData data) { public void savePlayerData(PlayerData data) {
Bukkit.getScheduler().runTaskAsynchronously(plugin, Bukkit.getScheduler().runTaskAsynchronously(plugin, () -> {
() -> { try {
try { Connection connection = dataSource.getConnection();
Statement statement = connection.createStatement(); Statement statement = connection.createStatement();
statement.executeUpdate(String.format( statement.executeUpdate(String.format(
"REPLACE INTO hamster_currency_player_data VALUES('%s', '%s');", "REPLACE INTO " + database + ".hamster_currency_player_data VALUES('%s', '%s');",
data.getUuid().toString(), data.getUuid().toString(),
data.saveToJson().toString() data.saveToJson().toString()
)); ));
statement.close(); statement.close();
} catch (SQLException e) { connection.close();
getLogUtils().error(e, "保存玩家 %s 的存档数据时出错!", data.getUuid()); } catch (SQLException e) {
} getLogUtils().error(e, "保存玩家 %s 的存档数据时出错!", data.getUuid());
getLogUtils().info("已保存玩家 %s 的存档数据.", data.getUuid()); }
ServiceMessageAPI.sendServiceMessage( getLogUtils().info("已保存玩家 %s 的存档数据.", data.getUuid());
"HamsterCurrency", ServiceMessageAPI.sendServiceMessage(
"savedPlayerData", "HamsterCurrency",
data.getUuid().toString() "savedPlayerData",
); data.getUuid().toString()
}); );
});
} }
@Override @Override

4
currency-plugin/src/main/resources/config.yml
View File

@@ -5,11 +5,11 @@ useBC: true
# 若开启跨服模式，则需要配置datasource # 若开启跨服模式，则需要配置datasource
datasource: datasource:
driver: "com.mysql.jdbc.Driver" driver: "com.mysql.cj.jdbc.Driver"
url: "jdbc:mysql://test3.hamster3.cn:3306?serverTimezone=UTC&useUnicode=true&characterEncoding=utf8&autoReconnect=true&useSSL=false" url: "jdbc:mysql://test3.hamster3.cn:3306?serverTimezone=UTC&useUnicode=true&characterEncoding=utf8&autoReconnect=true&useSSL=false"
user: Test user: Test
password: Test123.. password: Test123..
database: Test database: Test1
# 是否将这个服务器的配置文件设为模板 # 是否将这个服务器的配置文件设为模板
# 若设为true，config将会在服务器启动时自动上传至数据库 # 若设为true，config将会在服务器启动时自动上传至数据库
# 其他template为false的服务器将会在启动和重载时自动从数据库上下载config # 其他template为false的服务器将会在启动和重载时自动从数据库上下载config