package dev.plex.storage; import com.zaxxer.hikari.HikariConfig; import com.zaxxer.hikari.HikariDataSource; import dev.plex.Plex; import dev.plex.PlexBase; import javax.sql.DataSource; import java.io.File; import java.sql.Connection; import java.sql.SQLException; public class SQLConnection extends PlexBase { private HikariDataSource dataSource; public SQLConnection() { String host = plugin.config.getString("data.central.hostname"); int port = plugin.config.getInt("data.central.port"); String username = plugin.config.getString("data.central.user"); String password = plugin.config.getString("data.central.password"); String database = plugin.config.getString("data.central.db"); HikariConfig config = new HikariConfig(); config.addDataSourceProperty("cachePrepStmts", "true"); config.addDataSourceProperty("prepStmtCacheSize", "250"); config.addDataSourceProperty("prepStmtCacheSqlLimit", "2048"); this.dataSource = new HikariDataSource(); dataSource.setMaxLifetime(15000); dataSource.setIdleTimeout(15000 * 2); dataSource.setConnectionTimeout(15000 * 4); dataSource.setMinimumIdle(2); dataSource.setMaximumPoolSize(10); try { if (plugin.config.getString("data.central.storage").equalsIgnoreCase("sqlite")) { dataSource.setJdbcUrl("jdbc:sqlite:" + new File(plugin.getDataFolder(), "database.db").getAbsolutePath()); plugin.setStorageType(StorageType.SQLITE); } else if (plugin.config.getString("data.central.storage").equalsIgnoreCase("mariadb")) { Class.forName("org.mariadb.jdbc.Driver"); dataSource.setJdbcUrl("jdbc:mariadb://" + host + ":" + port + "/" + database); dataSource.setUsername(username); dataSource.setPassword(password); Plex.get().setStorageType(StorageType.MARIADB); } } catch (ClassNotFoundException throwables) { throwables.printStackTrace(); } try (Connection con = getCon()) { con.prepareStatement("CREATE TABLE IF NOT EXISTS `players` (" + "`uuid` VARCHAR(46) NOT NULL, " + "`name` VARCHAR(18), " + "`login_msg` VARCHAR(70), " + "`prefix` VARCHAR(45), " + "`rank` VARCHAR(20), " + "`ips` VARCHAR(2000), " + "`coins` BIGINT, " + "`vanished` BOOLEAN, " + "`commandspy` BOOLEAN, " + "PRIMARY KEY (`uuid`));").execute(); con.prepareStatement("CREATE TABLE IF NOT EXISTS `punishments` (" + "`punished` VARCHAR(46) NOT NULL, " + "`punisher` VARCHAR(46), " + "`punishedUsername` VARCHAR(16), " + "`ip` VARCHAR(2000), " + "`type` VARCHAR(30), " + "`reason` VARCHAR(2000), " + "`customTime` BOOLEAN, " + "`active` BOOLEAN, " + "`endDate` BIGINT" + ");").execute(); } catch (SQLException throwables) { throwables.printStackTrace(); } } public Connection getCon() { try { return dataSource.getConnection(); } catch (SQLException e) { e.printStackTrace(); } return null; } }