Plan/Plan/common/src/main/java/com/djrapitops/plan/storage/database/transactions/patches/LitebansTableHeaderPatch.java

90 lines
3.8 KiB
Java

/*
* This file is part of Player Analytics (Plan).
*
* Plan is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License v3 as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* Plan is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with Plan. If not, see <https://www.gnu.org/licenses/>.
*/
package com.djrapitops.plan.storage.database.transactions.patches;
import com.djrapitops.plan.storage.database.queries.QueryStatement;
import com.djrapitops.plan.storage.database.sql.tables.extension.ExtensionPluginTable;
import com.djrapitops.plan.storage.database.sql.tables.extension.ExtensionServerTableValueTable;
import com.djrapitops.plan.storage.database.sql.tables.extension.ExtensionTableProviderTable;
import com.djrapitops.plan.storage.database.transactions.ExecBatchStatement;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.HashSet;
import java.util.Set;
import static com.djrapitops.plan.storage.database.sql.building.Sql.*;
/**
* Removes invalid data caused by <a href="https://github.com/plan-player-analytics/Plan/issues/1355">issue #1355</a>.
*
* @author AuroraLS3
*/
public class LitebansTableHeaderPatch extends Patch {
private Set<Integer> found;
@Override
public boolean hasBeenApplied() {
String sql = SELECT + DISTINCT + "pr." + ExtensionTableProviderTable.ID + " as id" +
FROM + ExtensionServerTableValueTable.TABLE_NAME + " v" +
INNER_JOIN + ExtensionTableProviderTable.TABLE_NAME + " pr on pr." + ExtensionTableProviderTable.ID + "=v." + ExtensionServerTableValueTable.TABLE_ID +
INNER_JOIN + ExtensionPluginTable.TABLE_NAME + " p on p." + ExtensionPluginTable.ID + "=pr." + ExtensionTableProviderTable.PLUGIN_ID +
WHERE + "p." + ExtensionPluginTable.PLUGIN_NAME + "=?" +
AND + "(pr." + ExtensionTableProviderTable.PROVIDER_NAME + "=?" +
OR + "pr." + ExtensionTableProviderTable.PROVIDER_NAME + "=?" +
OR + "pr." + ExtensionTableProviderTable.PROVIDER_NAME + "=?" +
OR + "pr." + ExtensionTableProviderTable.PROVIDER_NAME + "=?)";
found = query(new QueryStatement<>(sql) {
@Override
public void prepare(PreparedStatement statement) throws SQLException {
statement.setString(1, "Litebans");
statement.setString(2, "bans");
statement.setString(3, "warns");
statement.setString(4, "kicks");
statement.setString(5, "mutes");
}
@Override
public Set<Integer> processResults(ResultSet set) throws SQLException {
Set<Integer> ids = new HashSet<>();
while (set.next()) {
ids.add(set.getInt("id"));
}
return ids;
}
});
return found.isEmpty();
}
@Override
protected void applyPatch() {
String sql = DELETE_FROM + ExtensionServerTableValueTable.TABLE_NAME +
WHERE + ExtensionServerTableValueTable.TABLE_ID + "=?";
execute(new ExecBatchStatement(sql) {
@Override
public void prepare(PreparedStatement statement) throws SQLException {
for (Integer id : found) {
statement.setInt(1, id);
statement.addBatch();
}
}
});
}
}