mirror/Plan
2
0
Fork 0
mirror of https://github.com/plan-player-analytics/Plan.git synced 2025-01-06 15:44:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch '4.0.0-BungeeCord-Support' of https://github.com/Rsl1122/Plan-PlayerAnalytics

Browse Source
This commit is contained in:
Fuzzlemann 2017-09-03 11:48:02 +02:00
commit 4dad6a25e1
2 changed files with 123 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

117
Plan/src/main/java/com/djrapitops/plan/database/tables/UsersTable.java
View File

@ -2,6 +2,7 @@ package main.java.com.djrapitops.plan.database.tables;
import com.djrapitops.plugin.utilities.Verify;
import main.java.com.djrapitops.plan.api.exceptions.DBCreateTableException;
import main.java.com.djrapitops.plan.data.UserInfo;
import main.java.com.djrapitops.plan.database.databases.SQLDB;
import main.java.com.djrapitops.plan.database.sql.*;
@ -21,6 +22,7 @@ public class UsersTable extends UserIDTable {
private final String columnRegistered = "registered";
private final String columnName = "name";
private final String columnTimesKicked = "times_kicked";
private String insertStatement;
/**
* @param db
@ -29,6 +31,10 @@ public class UsersTable extends UserIDTable {
public UsersTable(SQLDB db, boolean usingMySQL) {
super("plan_users", db, usingMySQL);
statementSelectID = "(" + Select.from(tableName, tableName + "." + columnID).where(columnUUID + "=?").toString() + ")";
insertStatement = Insert.values(tableName,
columnUUID,
columnRegistered,
columnName);
}
/**
@ -58,7 +64,7 @@ public class UsersTable extends UserIDTable {
ResultSet set = null;
try {
statement = prepareStatement(Select.from(tableName, columnUUID).toString());
statement.setFetchSize(2000);
statement.setFetchSize(5000);
set = statement.executeQuery();
while (set.next()) {
@ -160,10 +166,7 @@ public class UsersTable extends UserIDTable {
PreparedStatement statement = null;
try {
statement = prepareStatement(Insert.values(tableName,
columnUUID,
columnRegistered,
columnName));
statement = prepareStatement(insertStatement);
statement.setString(1, uuid.toString());
statement.setLong(2, registered);
statement.setString(3, name);
@ -300,4 +303,108 @@ public class UsersTable extends UserIDTable {
public String getColumnName() {
return columnName;
}
/**
* Inserts UUIDs, Register dates and Names to the table.
* <p>
* This method is for batch operations, and should not be used to add information of users.
* Use UserInfoTable instead.
*
* @param users
* @throws SQLException
*/
public void insertUsers(Map<UUID, UserInfo> users) throws SQLException {
if (Verify.isEmpty(users)) {
return;
}
PreparedStatement statement = null;
try {
statement = prepareStatement(insertStatement);
for (Map.Entry<UUID, UserInfo> entry : users.entrySet()) {
UUID uuid = entry.getKey();
UserInfo info = entry.getValue();
long registered = info.getRegistered();
String name = info.getName();
statement.setString(1, uuid.toString());
statement.setLong(2, registered);
statement.setString(3, name);
statement.addBatch();
}
statement.executeBatch();
commit(statement.getConnection());
} finally {
close(statement);
}
}
public Map<UUID, UserInfo> getUsers() throws SQLException {
PreparedStatement statement = null;
ResultSet set = null;
try {
statement = prepareStatement(Select.all(tableName)
.toString());
statement.setFetchSize(5000);
set = statement.executeQuery();
Map<UUID, UserInfo> users = new HashMap<>();
if (set.next()) {
UUID uuid = UUID.fromString(set.getString(columnUUID));
String name = set.getString(columnName);
long registered = set.getLong(columnRegistered);
users.put(uuid, new UserInfo(uuid, name, registered, false, false));
}
return users;
} finally {
endTransaction(statement);
close(set, statement);
}
}
public void updateKicked(Map<UUID, Integer> timesKicked) throws SQLException {
if (Verify.isEmpty(timesKicked)) {
return;
}
PreparedStatement statement = null;
try {
statement = prepareStatement("UPDATE " + tableName + " SET "
+ columnTimesKicked + "=?" +
" WHERE " + columnUUID + "=?");
for (Map.Entry<UUID, Integer> entry : timesKicked.entrySet()) {
UUID uuid = entry.getKey();
int kickCount = entry.getValue();
statement.setInt(1, kickCount);
statement.setString(2, uuid.toString());
statement.addBatch();
}
statement.executeBatch();
commit(statement.getConnection());
} finally {
close(statement);
}
}
public Map<UUID, Integer> getAllTimesKicked() throws SQLException {
PreparedStatement statement = null;
ResultSet set = null;
try {
statement = prepareStatement(Select.from(tableName, columnUUID, columnTimesKicked)
.toString());
statement.setFetchSize(5000);
set = statement.executeQuery();
Map<UUID, Integer> timesKicked = new HashMap<>();
while (set.next()) {
UUID uuid = UUID.fromString(set.getString(columnUUID));
int kickCount = set.getInt(columnTimesKicked);
timesKicked.put(uuid, kickCount);
}
return timesKicked;
} finally {
endTransaction(statement);
close(set, statement);
}
}
}

11
Plan/src/main/java/com/djrapitops/plan/database/tables/move/BatchOperationTable.java
View File

@ -8,6 +8,7 @@ import main.java.com.djrapitops.plan.api.exceptions.DBCreateTableException;
import main.java.com.djrapitops.plan.database.databases.SQLDB;
import main.java.com.djrapitops.plan.database.tables.ServerTable;
import main.java.com.djrapitops.plan.database.tables.Table;
import main.java.com.djrapitops.plan.database.tables.UsersTable;
import main.java.com.djrapitops.plan.systems.info.server.ServerInfo;
import java.sql.SQLException;
@ -116,4 +117,14 @@ public class BatchOperationTable extends Table {
}
toDB.db.getWorldTable().saveWorlds(db.getWorldTable().getWorlds());
}
public void copyUsers(BatchOperationTable toDB) throws SQLException {
if (toDB.equals(this)) {
return;
}
UsersTable fromTable = db.getUsersTable();
UsersTable toTable = toDB.db.getUsersTable();
toTable.insertUsers(fromTable.getUsers());
toTable.updateKicked(fromTable.getAllTimesKicked());
}
}