package com.aionemu.gameserver.dao;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.aionemu.commons.database.DatabaseFactory;
/**
* @author ViAl
*/
public class CommandsAccessDAO {
private static final Logger log = LoggerFactory.getLogger(CommandsAccessDAO.class);
private static final String LOAD_QUERY = "SELECT * FROM commands_access";
private static final String INSERT_QUERY = "INSERT INTO commands_access(player_id, command) VALUES (?,?)";
private static final String DELETE_QUERY = "DELETE FROM commands_access WHERE player_id = ? AND command = ?";
private static final String DELETE_ALL_QUERY = "DELETE FROM commands_access WHERE player_id = ?";
public static Map<Integer, Set<String>> loadAccesses() {
Map<Integer, Set<String>> accesses = new HashMap<>();
try (Connection conn = DatabaseFactory.getConnection();
PreparedStatement stmt = conn.prepareStatement(LOAD_QUERY);
ResultSet rset = stmt.executeQuery()) {
while (rset.next()) {
int playerId = rset.getInt("player_id");
String command = rset.getString("command");
accesses.compute(playerId, (key, commands) -> {
if (commands == null)
commands = new HashSet<>();
commands.add(command);
return commands;
});
}
} catch (Exception e) {
log.error("Error while loading commands accesses.", e);
}
return accesses;
}
public static void addAccess(int playerId, String commandName) {
try (Connection conn = DatabaseFactory.getConnection(); PreparedStatement stmt = conn.prepareStatement(INSERT_QUERY)) {
stmt.setInt(1, playerId);
stmt.setString(2, commandName);
stmt.executeUpdate();
} catch (Exception e) {
log.error("Error while adding access on command " + commandName + " to player " + playerId, e);
}
}
public static void removeAccess(int playerId, String commandName) {
try (Connection conn = DatabaseFactory.getConnection(); PreparedStatement stmt = conn.prepareStatement(DELETE_QUERY)) {
stmt.setInt(1, playerId);
stmt.setString(2, commandName);
stmt.executeUpdate();
} catch (Exception e) {
log.error("Error while removing access on command " + commandName + " from player " + playerId, e);
}
}
public static void removeAllAccesses(int playerId) {
try (Connection conn = DatabaseFactory.getConnection(); PreparedStatement stmt = conn.prepareStatement(DELETE_ALL_QUERY)) {
stmt.setInt(1, playerId);
stmt.executeUpdate();
} catch (Exception e) {
log.error("Error while removing all accesses from player " + playerId, e);
}
}
}