Upstream/Plan
1
0
Fork 0
mirror of https://github.com/plan-player-analytics/Plan.git synced 2025-02-11 01:41:55 +01:00
Code Issues Projects Releases Wiki Activity

Refactored UserInfoTable

Browse Source
This commit is contained in:
Rsl1122 2017-10-06 17:10:41 +03:00
parent c8b389643a
commit e1ec3532c5
1 changed files with 191 additions and 196 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

153
Plan/src/main/java/com/djrapitops/plan/database/tables/UserInfoTable.java
View File

@ -9,12 +9,14 @@ import main.java.com.djrapitops.plan.Plan;
import main.java.com.djrapitops.plan.api.exceptions.DBCreateTableException; import main.java.com.djrapitops.plan.api.exceptions.DBCreateTableException;
import main.java.com.djrapitops.plan.data.UserInfo; import main.java.com.djrapitops.plan.data.UserInfo;
import main.java.com.djrapitops.plan.database.databases.SQLDB; import main.java.com.djrapitops.plan.database.databases.SQLDB;
import main.java.com.djrapitops.plan.database.processing.ExecStatement;
import main.java.com.djrapitops.plan.database.processing.QueryAllStatement;
import main.java.com.djrapitops.plan.database.processing.QueryStatement;
import main.java.com.djrapitops.plan.database.sql.Select; import main.java.com.djrapitops.plan.database.sql.Select;
import main.java.com.djrapitops.plan.database.sql.Sql; import main.java.com.djrapitops.plan.database.sql.Sql;
import main.java.com.djrapitops.plan.database.sql.TableSqlParser; import main.java.com.djrapitops.plan.database.sql.TableSqlParser;
import main.java.com.djrapitops.plan.database.sql.Update; import main.java.com.djrapitops.plan.database.sql.Update;
import java.sql.Connection;
import java.sql.PreparedStatement; import java.sql.PreparedStatement;
import java.sql.ResultSet; import java.sql.ResultSet;
import java.sql.SQLException; import java.sql.SQLException;
@ -59,57 +61,56 @@ public class UserInfoTable extends UserIDTable {
usersTable.registerUser(uuid, registered, "Waiting for Update.."); usersTable.registerUser(uuid, registered, "Waiting for Update..");
} }
PreparedStatement statement = null; String sql = "INSERT INTO " + tableName + " (" +
try (Connection connection = getConnection()) {
statement = connection.prepareStatement("INSERT INTO " + tableName + " (" +
columnUserID + ", " + columnUserID + ", " +
columnRegistered + ", " + columnRegistered + ", " +
columnServerID + columnServerID +
") VALUES (" + ") VALUES (" +
usersTable.statementSelectID + ", " + usersTable.statementSelectID + ", " +
"?, " + "?, " +
serverTable.statementSelectServerID + ")"); serverTable.statementSelectServerID + ")";
execute(new ExecStatement(sql) {
@Override
public void prepare(PreparedStatement statement) throws SQLException {
statement.setString(1, uuid.toString()); statement.setString(1, uuid.toString());
statement.setLong(2, registered); statement.setLong(2, registered);
statement.setString(3, Plan.getServerUUID().toString()); statement.setString(3, Plan.getServerUUID().toString());
statement.execute();
commit(connection);
} finally {
close(statement);
} }
});
} }
public boolean isRegistered(UUID uuid) throws SQLException { public boolean isRegistered(UUID uuid) throws SQLException {
PreparedStatement statement = null; String sql = Select.from(tableName, "COUNT(" + columnUserID + ") as c")
ResultSet set = null;
try (Connection connection = getConnection()) {
statement = connection.prepareStatement(Select.from(tableName, "COUNT(" + columnUserID + ") as c")
.where(columnUserID + "=" + usersTable.statementSelectID) .where(columnUserID + "=" + usersTable.statementSelectID)
.toString()); .toString();
return query(new QueryStatement<Boolean>(sql) {
@Override
public void prepare(PreparedStatement statement) throws SQLException {
statement.setString(1, uuid.toString()); statement.setString(1, uuid.toString());
set = statement.executeQuery();
return set.next() && set.getInt("c") >= 1;
} finally {
close(set, statement);
} }
@Override
public Boolean processResults(ResultSet set) throws SQLException {
return set.next() && set.getInt("c") >= 1;
}
});
} }
public void updateOpAndBanStatus(UUID uuid, boolean opped, boolean banned) throws SQLException { public void updateOpAndBanStatus(UUID uuid, boolean opped, boolean banned) throws SQLException {
PreparedStatement statement = null; String sql = Update.values(tableName, columnOP, columnBanned)
try (Connection connection = getConnection()) {
statement = connection.prepareStatement(Update.values(tableName, columnOP, columnBanned)
.where(columnUserID + "=" + usersTable.statementSelectID) .where(columnUserID + "=" + usersTable.statementSelectID)
.toString()); .toString();
execute(new ExecStatement(sql) {
@Override
public void prepare(PreparedStatement statement) throws SQLException {
statement.setBoolean(1, opped); statement.setBoolean(1, opped);
statement.setBoolean(2, banned); statement.setBoolean(2, banned);
statement.setString(3, uuid.toString()); statement.setString(3, uuid.toString());
statement.execute();
commit(connection);
} finally {
close(statement);
} }
});
} }
public UserInfo getUserInfo(UUID uuid) throws SQLException { public UserInfo getUserInfo(UUID uuid) throws SQLException {
@ -117,12 +118,9 @@ public class UserInfoTable extends UserIDTable {
} }
public UserInfo getUserInfo(UUID uuid, UUID serverUUID) throws SQLException { public UserInfo getUserInfo(UUID uuid, UUID serverUUID) throws SQLException {
PreparedStatement statement = null;
ResultSet set = null;
try (Connection connection = getConnection()) {
String usersIDColumn = usersTable + "." + usersTable.getColumnID(); String usersIDColumn = usersTable + "." + usersTable.getColumnID();
String usersNameColumn = usersTable + "." + usersTable.getColumnName() + " as name"; String usersNameColumn = usersTable + "." + usersTable.getColumnName() + " as name";
statement = connection.prepareStatement("SELECT " + String sql = "SELECT " +
tableName + "." + columnRegistered + ", " + tableName + "." + columnRegistered + ", " +
columnOP + ", " + columnOP + ", " +
columnBanned + ", " + columnBanned + ", " +
@ -130,12 +128,17 @@ public class UserInfoTable extends UserIDTable {
" FROM " + tableName + " FROM " + tableName +
" JOIN " + usersTable + " on " + usersIDColumn + "=" + columnUserID + " JOIN " + usersTable + " on " + usersIDColumn + "=" + columnUserID +
" WHERE " + columnUserID + "=" + usersTable.statementSelectID + " WHERE " + columnUserID + "=" + usersTable.statementSelectID +
" AND " + columnServerID + "=" + serverTable.statementSelectServerID " AND " + columnServerID + "=" + serverTable.statementSelectServerID;
);
statement.setFetchSize(20000); return query(new QueryStatement<UserInfo>(sql) {
@Override
public void prepare(PreparedStatement statement) throws SQLException {
statement.setString(1, uuid.toString()); statement.setString(1, uuid.toString());
statement.setString(2, serverUUID.toString()); statement.setString(2, serverUUID.toString());
set = statement.executeQuery(); }
@Override
public UserInfo processResults(ResultSet set) throws SQLException {
if (set.next()) { if (set.next()) {
long registered = set.getLong(columnRegistered); long registered = set.getLong(columnRegistered);
boolean opped = set.getBoolean(columnOP); boolean opped = set.getBoolean(columnOP);
@ -144,9 +147,8 @@ public class UserInfoTable extends UserIDTable {
return new UserInfo(uuid, name, registered, opped, banned); return new UserInfo(uuid, name, registered, opped, banned);
} }
return null; return null;
} finally {
close(set, statement);
} }
});
} }
/** /**
@ -159,14 +161,10 @@ public class UserInfoTable extends UserIDTable {
} }
public List<UserInfo> getServerUserInfo(UUID serverUUID) throws SQLException { public List<UserInfo> getServerUserInfo(UUID serverUUID) throws SQLException {
PreparedStatement statement = null;
ResultSet set = null;
try (Connection connection = getConnection()) {
List<UserInfo> userInfo = new ArrayList<>();
String usersIDColumn = usersTable + "." + usersTable.getColumnID(); String usersIDColumn = usersTable + "." + usersTable.getColumnID();
String usersUUIDColumn = usersTable + "." + usersTable.getColumnUUID() + " as uuid"; String usersUUIDColumn = usersTable + "." + usersTable.getColumnUUID() + " as uuid";
String usersNameColumn = usersTable + "." + usersTable.getColumnName() + " as name"; String usersNameColumn = usersTable + "." + usersTable.getColumnName() + " as name";
statement = connection.prepareStatement("SELECT " + String sql = "SELECT " +
tableName + "." + columnRegistered + ", " + tableName + "." + columnRegistered + ", " +
columnOP + ", " + columnOP + ", " +
columnBanned + ", " + columnBanned + ", " +
@ -174,11 +172,17 @@ public class UserInfoTable extends UserIDTable {
usersUUIDColumn + usersUUIDColumn +
" FROM " + tableName + " FROM " + tableName +
" JOIN " + usersTable + " on " + usersIDColumn + "=" + columnUserID + " JOIN " + usersTable + " on " + usersIDColumn + "=" + columnUserID +
" WHERE " + columnServerID + "=" + serverTable.statementSelectServerID " WHERE " + columnServerID + "=" + serverTable.statementSelectServerID;
);
statement.setFetchSize(20000); return query(new QueryStatement<List<UserInfo>>(sql, 20000) {
@Override
public void prepare(PreparedStatement statement) throws SQLException {
statement.setString(1, serverUUID.toString()); statement.setString(1, serverUUID.toString());
set = statement.executeQuery(); }
@Override
public List<UserInfo> processResults(ResultSet set) throws SQLException {
List<UserInfo> userInfo = new ArrayList<>();
while (set.next()) { while (set.next()) {
long registered = set.getLong(columnRegistered); long registered = set.getLong(columnRegistered);
boolean opped = set.getBoolean(columnOP); boolean opped = set.getBoolean(columnOP);
@ -191,20 +195,16 @@ public class UserInfoTable extends UserIDTable {
} }
} }
return userInfo; return userInfo;
} finally {
close(set, statement);
} }
});
} }
public Map<UUID, List<UserInfo>> getAllUserInfo() throws SQLException { public Map<UUID, List<UserInfo>> getAllUserInfo() throws SQLException {
PreparedStatement statement = null;
ResultSet set = null;
try (Connection connection = getConnection()) {
String usersIDColumn = usersTable + "." + usersTable.getColumnID(); String usersIDColumn = usersTable + "." + usersTable.getColumnID();
String usersUUIDColumn = usersTable + "." + usersTable.getColumnUUID() + " as uuid"; String usersUUIDColumn = usersTable + "." + usersTable.getColumnUUID() + " as uuid";
String serverIDColumn = serverTable + "." + serverTable.getColumnID(); String serverIDColumn = serverTable + "." + serverTable.getColumnID();
String serverUUIDColumn = serverTable + "." + serverTable.getColumnUUID() + " as s_uuid"; String serverUUIDColumn = serverTable + "." + serverTable.getColumnUUID() + " as s_uuid";
statement = connection.prepareStatement("SELECT " + String sql = "SELECT " +
tableName + "." + columnRegistered + ", " + tableName + "." + columnRegistered + ", " +
columnBanned + ", " + columnBanned + ", " +
columnOP + ", " + columnOP + ", " +
@ -212,10 +212,11 @@ public class UserInfoTable extends UserIDTable {
serverUUIDColumn + serverUUIDColumn +
" FROM " + tableName + " FROM " + tableName +
" JOIN " + usersTable + " on " + usersIDColumn + "=" + columnUserID + " JOIN " + usersTable + " on " + usersIDColumn + "=" + columnUserID +
" JOIN " + serverTable + " on " + serverIDColumn + "=" + columnServerID " JOIN " + serverTable + " on " + serverIDColumn + "=" + columnServerID;
);
statement.setFetchSize(20000); return query(new QueryAllStatement<Map<UUID, List<UserInfo>>>(sql, 50000) {
set = statement.executeQuery(); @Override
public Map<UUID, List<UserInfo>> processResults(ResultSet set) throws SQLException {
Map<UUID, List<UserInfo>> serverMap = new HashMap<>(); Map<UUID, List<UserInfo>> serverMap = new HashMap<>();
while (set.next()) { while (set.next()) {
UUID serverUUID = UUID.fromString(set.getString("s_uuid")); UUID serverUUID = UUID.fromString(set.getString("s_uuid"));
@ -232,18 +233,16 @@ public class UserInfoTable extends UserIDTable {
serverMap.put(serverUUID, userInfos); serverMap.put(serverUUID, userInfos);
} }
return serverMap; return serverMap;
} finally {
close(set, statement);
} }
});
} }
public void insertUserInfo(Map<UUID, List<UserInfo>> allUserInfos) throws SQLException { public void insertUserInfo(Map<UUID, List<UserInfo>> allUserInfos) throws SQLException {
if (Verify.isEmpty(allUserInfos)) { if (Verify.isEmpty(allUserInfos)) {
return; return;
} }
PreparedStatement statement = null;
try (Connection connection = getConnection()) { String sql = "INSERT INTO " + tableName + " (" +
statement = connection.prepareStatement("INSERT INTO " + tableName + " (" +
columnUserID + ", " + columnUserID + ", " +
columnRegistered + ", " + columnRegistered + ", " +
columnServerID + ", " + columnServerID + ", " +
@ -252,8 +251,11 @@ public class UserInfoTable extends UserIDTable {
") VALUES (" + ") VALUES (" +
usersTable.statementSelectID + ", " + usersTable.statementSelectID + ", " +
"?, " + "?, " +
serverTable.statementSelectServerID + ", ?, ?)"); serverTable.statementSelectServerID + ", ?, ?)";
executeBatch(new ExecStatement(sql) {
@Override
public void prepare(PreparedStatement statement) throws SQLException {
// Every Server // Every Server
for (Map.Entry<UUID, List<UserInfo>> entry : allUserInfos.entrySet()) { for (Map.Entry<UUID, List<UserInfo>> entry : allUserInfos.entrySet()) {
UUID serverUUID = entry.getKey(); UUID serverUUID = entry.getKey();
@ -267,31 +269,25 @@ public class UserInfoTable extends UserIDTable {
statement.addBatch(); statement.addBatch();
} }
} }
statement.executeBatch();
commit(connection);
} finally {
close(statement);
} }
});
} }
public Map<UUID, Set<UUID>> getSavedUUIDs() throws SQLException { public Map<UUID, Set<UUID>> getSavedUUIDs() throws SQLException {
PreparedStatement statement = null;
ResultSet set = null;
try (Connection connection = getConnection()) {
String usersIDColumn = usersTable + "." + usersTable.getColumnID(); String usersIDColumn = usersTable + "." + usersTable.getColumnID();
String usersUUIDColumn = usersTable + "." + usersTable.getColumnUUID() + " as uuid"; String usersUUIDColumn = usersTable + "." + usersTable.getColumnUUID() + " as uuid";
String serverIDColumn = serverTable + "." + serverTable.getColumnID(); String serverIDColumn = serverTable + "." + serverTable.getColumnID();
String serverUUIDColumn = serverTable + "." + serverTable.getColumnUUID() + " as s_uuid"; String serverUUIDColumn = serverTable + "." + serverTable.getColumnUUID() + " as s_uuid";
statement = connection.prepareStatement("SELECT " + String sql = "SELECT " +
usersUUIDColumn + ", " + usersUUIDColumn + ", " +
serverUUIDColumn + serverUUIDColumn +
" FROM " + tableName + " FROM " + tableName +
" JOIN " + usersTable + " on " + usersIDColumn + "=" + columnUserID + " JOIN " + usersTable + " on " + usersIDColumn + "=" + columnUserID +
" JOIN " + serverTable + " on " + serverIDColumn + "=" + columnServerID " JOIN " + serverTable + " on " + serverIDColumn + "=" + columnServerID;
);
statement.setFetchSize(5000); return query(new QueryAllStatement<Map<UUID, Set<UUID>>>(sql, 50000) {
set = statement.executeQuery(); @Override
public Map<UUID, Set<UUID>> processResults(ResultSet set) throws SQLException {
Map<UUID, Set<UUID>> serverMap = new HashMap<>(); Map<UUID, Set<UUID>> serverMap = new HashMap<>();
while (set.next()) { while (set.next()) {
UUID serverUUID = UUID.fromString(set.getString("s_uuid")); UUID serverUUID = UUID.fromString(set.getString("s_uuid"));
@ -303,8 +299,7 @@ public class UserInfoTable extends UserIDTable {
serverMap.put(serverUUID, uuids); serverMap.put(serverUUID, uuids);
} }
return serverMap; return serverMap;
} finally { }
close(set, statement); });
}
} }
} }