Upstream/Plan
1
0
Fork 0
mirror of https://github.com/plan-player-analytics/Plan.git synced 2024-11-04 17:59:56 +01:00
Code Issues Projects Releases Wiki Activity

Added connection validation to MySQL

Browse Source
This commit is contained in:
Rsl1122 2018-08-06 11:54:12 +03:00
parent cccdf14e9b
commit 8641bd5f0d
1 changed files with 44 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
Plan/src/main/java/com/djrapitops/plan/system/database/databases/sql/MySQLDB.java
View File

@ -1,6 +1,7 @@
package com.djrapitops.plan.system.database.databases.sql;
import com.djrapitops.plan.api.exceptions.database.DBInitException;
import com.djrapitops.plan.api.exceptions.database.DBOpException;
import com.djrapitops.plan.system.locale.Locale;
import com.djrapitops.plan.system.locale.lang.PluginLang;
import com.djrapitops.plan.system.settings.Settings;
@ -33,6 +34,7 @@ public class MySQLDB extends SQLDB {
*/
@Override
public void setupDataSource() throws DBInitException {
try {
HikariConfig config = new HikariConfig();
String host = Settings.DB_HOST.toString();
@ -56,11 +58,15 @@ public class MySQLDB extends SQLDB {
increment++;
config.setAutoCommit(true);
config.setReadOnly(false);
config.setMaximumPoolSize(8);
config.setLeakDetectionThreshold(TimeAmount.MINUTE.ms() * 10L);
this.dataSource = new HikariDataSource(config);
getConnection();
} catch (SQLException e) {
throw new DBInitException("Failed to set-up HikariCP Datasource: " + e.getMessage(), e);
}
}
/**
@ -73,7 +79,18 @@ public class MySQLDB extends SQLDB {
@Override
public Connection getConnection() throws SQLException {
return dataSource.getConnection();
Connection connection = dataSource.getConnection();
if (!connection.isValid(5)) {
if (dataSource instanceof HikariDataSource) {
((HikariDataSource) dataSource).close();
}
try {
setupDataSource();
} catch (DBInitException e) {
throw new DBOpException("Failed to restart DataSource after a connection was invalid: " + e.getMessage(), e);
}
}
return connection;
}
@Override