mirror of
https://git.krews.org/morningstar/Arcturus-Community.git
synced 2025-02-18 19:32:36 +01:00
114 lines
3.6 KiB
Java
114 lines
3.6 KiB
Java
package com.eu.habbo.database;
|
|
|
|
import com.eu.habbo.Emulator;
|
|
import com.eu.habbo.core.ConfigurationManager;
|
|
import com.zaxxer.hikari.HikariDataSource;
|
|
import gnu.trove.map.hash.THashMap;
|
|
import gnu.trove.set.hash.THashSet;
|
|
import org.slf4j.Logger;
|
|
import org.slf4j.LoggerFactory;
|
|
|
|
import java.sql.Connection;
|
|
import java.sql.PreparedStatement;
|
|
import java.sql.SQLException;
|
|
import java.util.Map;
|
|
import java.util.regex.Matcher;
|
|
import java.util.regex.Pattern;
|
|
|
|
public class Database {
|
|
|
|
private static final Logger LOGGER = LoggerFactory.getLogger(Database.class);
|
|
|
|
private HikariDataSource dataSource;
|
|
private DatabasePool databasePool;
|
|
|
|
public Database(ConfigurationManager config) {
|
|
long millis = System.currentTimeMillis();
|
|
|
|
boolean SQLException = false;
|
|
|
|
try {
|
|
this.databasePool = new DatabasePool();
|
|
if (!this.databasePool.getStoragePooling(config)) {
|
|
LOGGER.info("Failed to connect to the database. Please check config.ini and make sure the MySQL process is running. Shutting down...");
|
|
SQLException = true;
|
|
return;
|
|
}
|
|
this.dataSource = this.databasePool.getDatabase();
|
|
} catch (Exception e) {
|
|
SQLException = true;
|
|
LOGGER.error("Failed to connect to your database.", e);
|
|
} finally {
|
|
if (SQLException) {
|
|
Emulator.prepareShutdown();
|
|
}
|
|
}
|
|
|
|
LOGGER.info("Database -> Connected! ({} MS)", System.currentTimeMillis() - millis);
|
|
}
|
|
|
|
public void dispose() {
|
|
if (this.databasePool != null) {
|
|
this.databasePool.getDatabase().close();
|
|
}
|
|
|
|
this.dataSource.close();
|
|
}
|
|
|
|
public HikariDataSource getDataSource() {
|
|
return this.dataSource;
|
|
}
|
|
|
|
public DatabasePool getDatabasePool() {
|
|
return this.databasePool;
|
|
}
|
|
|
|
public static PreparedStatement preparedStatementWithParams(Connection connection, String query, THashMap<String, Object> queryParams) throws SQLException {
|
|
THashMap<Integer, Object> params = new THashMap<Integer, Object>();
|
|
THashSet<String> quotedParams = new THashSet<>();
|
|
|
|
for(String key : queryParams.keySet()) {
|
|
quotedParams.add(Pattern.quote(key));
|
|
}
|
|
|
|
String regex = "(" + String.join("|", quotedParams) + ")";
|
|
|
|
Matcher m = Pattern.compile(regex).matcher(query);
|
|
|
|
int i = 1;
|
|
|
|
while (m.find()) {
|
|
try {
|
|
params.put(i, queryParams.get(m.group(1)));
|
|
i++;
|
|
}
|
|
catch (Exception ignored) { }
|
|
}
|
|
|
|
PreparedStatement statement = connection.prepareStatement(query.replaceAll(regex, "?"));
|
|
|
|
for(Map.Entry<Integer, Object> set : params.entrySet()) {
|
|
if(set.getValue().getClass() == String.class) {
|
|
statement.setString(set.getKey(), (String)set.getValue());
|
|
}
|
|
else if(set.getValue().getClass() == Integer.class) {
|
|
statement.setInt(set.getKey(), (Integer)set.getValue());
|
|
}
|
|
else if(set.getValue().getClass() == Double.class) {
|
|
statement.setDouble(set.getKey(), (Double)set.getValue());
|
|
}
|
|
else if(set.getValue().getClass() == Float.class) {
|
|
statement.setFloat(set.getKey(), (Float)set.getValue());
|
|
}
|
|
else if(set.getValue().getClass() == Long.class) {
|
|
statement.setLong(set.getKey(), (Long)set.getValue());
|
|
}
|
|
else {
|
|
statement.setObject(set.getKey(), set.getValue());
|
|
}
|
|
}
|
|
|
|
return statement;
|
|
}
|
|
}
|