package com.aionemu.gameserver.world.container;
import java.util.Collection;
import java.util.Iterator;
import java.util.concurrent.ConcurrentHashMap;
import java.util.stream.Collectors;
import com.aionemu.gameserver.model.gameobjects.player.Player;
import com.aionemu.gameserver.world.exceptions.DuplicateAionObjectException;
/**
* Container for storing Players by objectId and name.
*
* @author -Nemesiss-, Neon
*/
public class PlayerContainer implements Iterable<Player> {
private final ConcurrentHashMap<Integer, Player> playersById = new ConcurrentHashMap<>();
private final ConcurrentHashMap<String, Player> playersByName = new ConcurrentHashMap<>();
public void add(Player player) {
if (playersById.put(player.getObjectId(), player) != null)
throw new DuplicateAionObjectException(player, playersById.get(player.getObjectId()));
if (playersByName.put(player.getName(), player) != null)
throw new DuplicateAionObjectException(player, playersByName.get(player.getName()));
}
public void remove(Player player) {
playersById.remove(player.getObjectId());
playersByName.remove(player.getName());
}
public Player get(int objectId) {
return playersById.get(objectId);
}
public Player get(String name) {
return playersByName.get(name);
}
@Override
public Iterator<Player> iterator() {
return playersById.values().iterator();
}
public Collection<Player> getAllPlayers() {
return playersById.values().stream().filter(p -> p != null).collect(Collectors.toList()); // ensure there are no null values (due to concurrent object removal)
}
public void updateCachedPlayerName(String oldName, Player player) {
playersByName.compute(oldName, (n, p)-> {
playersByName.put(player.getName(), player);
return player.equals(p) ? null : p;
});
}
}