Plan/Plan/common/src/main/java/com/djrapitops/plan/storage/database/transactions/webuser/StoreMissingWebPermissionsT...

64 lines
2.3 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.webuser;
import com.djrapitops.plan.storage.database.queries.objects.WebUserQueries;
import com.djrapitops.plan.storage.database.sql.tables.webuser.WebPermissionTable;
import com.djrapitops.plan.storage.database.transactions.ExecBatchStatement;
import com.djrapitops.plan.storage.database.transactions.Transaction;
import java.sql.PreparedStatement;
import java.sql.SQLException;
import java.util.Collection;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
/**
* Adds missing permissions to the permission table.
*
* @author AuroraLS3
*/
public class StoreMissingWebPermissionsTransaction extends Transaction {
private final Collection<String> permissions;
public StoreMissingWebPermissionsTransaction(Collection<String> permissions) {
this.permissions = permissions;
}
@Override
protected void performOperations() {
List<String> storedPermissions = query(WebUserQueries.fetchAvailablePermissions());
Set<String> missingPermissions = new HashSet<>();
for (String permission : permissions) {
if (!storedPermissions.contains(permission)) {
missingPermissions.add(permission);
}
}
execute(new ExecBatchStatement(WebPermissionTable.safeInsertSQL(dbType)) {
@Override
public void prepare(PreparedStatement statement) throws SQLException {
for (String permission : missingPermissions) {
statement.setString(1, permission);
statement.addBatch();
}
}
});
}
}