Skip to content

Commit

Permalink
Don't use the region chunk has table until a threshold.
Browse files Browse the repository at this point in the history
  • Loading branch information
sk89q committed Aug 22, 2014
1 parent 9505254 commit f57afb2
Showing 1 changed file with 60 additions and 44 deletions.
Expand Up @@ -51,8 +51,14 @@
*/
public class ChunkHashTable implements ConcurrentRegionIndex {

/**
* The number of regions required before this hash table kicks in.
*/
private static final int CACHE_THRESHOLD = 5;

private ListeningExecutorService executor = createExecutor();
private LongHashTable<ChunkState> states = new LongHashTable<ChunkState>();
@Nullable
private LongHashTable<ChunkState> states;
private final RegionIndex index;
private final Object lock = new Object();
@Nullable
Expand All @@ -79,35 +85,23 @@ private ListeningExecutorService createExecutor() {
}

/**
* Get a state object at the given position.
* Get a state object at the given position, returning an entry only
* if one exists.
*
* @param position the position
* @param create true to create an entry if one does not exist
* @return a chunk state object, or {@code null} (only if {@code create} is false)
* @return a chunk state object, or {@code null}
*/
@Nullable
private ChunkState get(Vector2D position, boolean create) {
ChunkState state;
synchronized (lock) {
state = states.get(position.getBlockX(), position.getBlockZ());
if (state == null && create) {
state = new ChunkState(position);
states.put(position.getBlockX(), position.getBlockZ(), state);
executor.submit(new EnumerateRegions(position));
private ChunkState get(Vector2D position) {
LongHashTable<ChunkState> states = this.states;

if (states != null) {
synchronized (lock) {
return states.get(position.getBlockX(), position.getBlockZ());
}
} else {
return null;
}
return state;
}

/**
* Get a state at the given position or create a new entry if one does
* not exist.
*
* @param position the position
* @return a state
*/
private ChunkState getOrCreate(Vector2D position) {
return get(position, true);
}

/**
Expand All @@ -118,22 +112,30 @@ private void rebuild() {
ListeningExecutorService previousExecutor = executor;
LongHashTable<ChunkState> previousStates = states;

previousExecutor.shutdownNow();
states = new LongHashTable<ChunkState>();
executor = createExecutor();
if (index.size() > CACHE_THRESHOLD) {
previousExecutor.shutdownNow();

List<Vector2D> positions = new ArrayList<Vector2D>();
for (ChunkState state : previousStates.values()) {
Vector2D position = state.getPosition();
positions.add(position);
states.put(position.getBlockX(), position.getBlockZ(), new ChunkState(position));
}
states = new LongHashTable<ChunkState>();
executor = createExecutor();

if (!positions.isEmpty()) {
executor.submit(new EnumerateRegions(positions));
}
if (previousStates != null) {
List<Vector2D> positions = new ArrayList<Vector2D>();

lastState = null;
for (ChunkState state : previousStates.values()) {
Vector2D position = state.getPosition();
positions.add(position);
states.put(position.getBlockX(), position.getBlockZ(), new ChunkState(position));
}

if (!positions.isEmpty()) {
executor.submit(new EnumerateRegions(positions));
}
}

lastState = null;
} else {
states = null;
}
}
}

Expand All @@ -159,7 +161,19 @@ public boolean awaitCompletion(long timeout, TimeUnit unit) throws InterruptedEx
@Override
public void bias(Vector2D chunkPosition) {
checkNotNull(chunkPosition);
getOrCreate(chunkPosition);

if (states != null) {
synchronized (lock) {
if (states != null) {
ChunkState state = states.get(chunkPosition.getBlockX(), chunkPosition.getBlockZ());
if (state == null) {
state = new ChunkState(chunkPosition);
states.put(chunkPosition.getBlockX(), chunkPosition.getBlockZ(), state);
executor.submit(new EnumerateRegions(chunkPosition));
}
}
}
}
}

@Override
Expand All @@ -175,10 +189,12 @@ public void biasAll(Collection<Vector2D> chunkPositions) {
public void forget(Vector2D chunkPosition) {
checkNotNull(chunkPosition);
synchronized (lock) {
states.remove(chunkPosition.getBlockX(), chunkPosition.getBlockZ());
ChunkState state = lastState;
if (state != null && state.getPosition().getBlockX() == chunkPosition.getBlockX() && state.getPosition().getBlockZ() == chunkPosition.getBlockZ()) {
lastState = null;
if (states != null) {
states.remove(chunkPosition.getBlockX(), chunkPosition.getBlockZ());
ChunkState state = lastState;
if (state != null && state.getPosition().getBlockX() == chunkPosition.getBlockX() && state.getPosition().getBlockZ() == chunkPosition.getBlockZ()) {
lastState = null;
}
}
}
}
Expand Down Expand Up @@ -238,7 +254,7 @@ public void applyContaining(Vector position, Predicate<ProtectedRegion> consumer
int chunkZ = position.getBlockZ() >> 4;

if (state == null || state.getPosition().getBlockX() != chunkX || state.getPosition().getBlockZ() != chunkZ) {
state = get(new Vector2D(chunkX, chunkZ), false);
state = get(new Vector2D(chunkX, chunkZ));
}

if (state != null && state.isLoaded()) {
Expand Down Expand Up @@ -301,7 +317,7 @@ private EnumerateRegions(List<Vector2D> positions) {
@Override
public void run() {
for (Vector2D position : positions) {
ChunkState state = get(position, false);
ChunkState state = get(position);

if (state != null) {
List<ProtectedRegion> regions = new ArrayList<ProtectedRegion>();
Expand Down

0 comments on commit f57afb2

Please sign in to comment.