From d36d1367dd5f8207585ac03adcb92d936482125c Mon Sep 17 00:00:00 2001 From: Aikar Date: Sat, 21 Jul 2018 16:55:04 -0400 Subject: [PATCH] Async Chunk Loading and Generation This brings back parity to 1.12 and older versions in that any chunk requested as part of the PlayerChunkMap can be loaded asynchronously, since the chunk isn't needed "immediately". The previous system used by CraftBukkit has been completely abandoned, as mojang has put more concurrency checks into the process. The new process is no longer lock free, but tries to maintain locks as short as possible. But with 1.13, we now have Chunk Conversions too. A main issue about keeping just loading parity to 1.12 is that standard loads now are treated as generation level events, to run the converter on another thread. However mojangs code was pretty bad here and doesn't actually provide any concurrency... Mojangs code is still not thread safe, and can only operate on one world per thread safely, but this is still a major improvement to get world generation off of the main thread for exploration. This change brings Chunk Requests triggered by the Chunk Map to be lazily loaded asynchronously. Standard chunk loads can load in parallel across a shared executor. However, chunk conversions and generations must only run one per world at a time, so we have a single thread executor for those operations per world, that all of those requests get scheduled to. getChunkAt method is now thread safe, but has not been tested in use by other threads for generations, but should be safe to do. However, we are not encouraging plugins to go getting chunks async, as while looking the chunk up may be safe, absolutely nothing about reading or writing to the chunk will be safe, so plugins still should not be touching chunks asynchronously! diff --git a/src/main/java/com/destroystokyo/paper/PaperConfig.java b/src/main/java/com/destroystokyo/paper/PaperConfig.java index 6f76351b8..e8b03795f 100644 --- a/src/main/java/com/destroystokyo/paper/PaperConfig.java +++ b/src/main/java/com/destroystokyo/paper/PaperConfig.java @@ -376,4 +376,57 @@ public class PaperConfig { } } } + + public static boolean asyncChunks = false; + public static boolean asyncChunkGeneration = true; + public static boolean asyncChunkGenThreadPerWorld = true; + public static int asyncChunkLoadThreads = -1; + private static void asyncChunks() { + if (version < 15) { + boolean enabled = config.getBoolean("settings.async-chunks", true); + ConfigurationSection section = config.createSection("settings.async-chunks"); + section.set("enable", enabled); + section.set("load-threads", -1); + section.set("generation", true); + section.set("thread-per-world-generation", true); + } + + asyncChunks = getBoolean("settings.async-chunks.enable", true); + asyncChunkGeneration = getBoolean("settings.async-chunks.generation", true); + asyncChunkGenThreadPerWorld = getBoolean("settings.async-chunks.thread-per-world-generation", true); + asyncChunkLoadThreads = getInt("settings.async-chunks.load-threads", -1); + if (asyncChunkLoadThreads <= 0) { + asyncChunkLoadThreads = (int) Math.min(Integer.getInteger("paper.maxChunkThreads", 8), Runtime.getRuntime().availableProcessors() * 1.5); + } + + // Let Shared Host set some limits + String sharedHostEnvGen = System.getenv("PAPER_ASYNC_CHUNKS_SHARED_HOST_GEN"); + String sharedHostEnvLoad = System.getenv("PAPER_ASYNC_CHUNKS_SHARED_HOST_LOAD"); + if ("1".equals(sharedHostEnvGen)) { + log("Async Chunks - Generation: Your host has requested to use a single thread world generation"); + asyncChunkGenThreadPerWorld = false; + } else if ("2".equals(sharedHostEnvGen)) { + log("Async Chunks - Generation: Your host has disabled async world generation - You will experience lag from world generation"); + asyncChunkGeneration = false; + } + + if (sharedHostEnvLoad != null) { + try { + asyncChunkLoadThreads = Math.max(1, Math.min(asyncChunkLoadThreads, Integer.parseInt(sharedHostEnvLoad))); + } catch (NumberFormatException ignored) {} + } + + if (!asyncChunks) { + log("Async Chunks: Disabled - Chunks will be managed synchronosuly, and will cause tremendous lag."); + } else { + log("Async Chunks: Enabled - Chunks will be loaded much faster, without lag."); + if (!asyncChunkGeneration) { + log("Async Chunks - Generation: Disabled - Chunks will be generated synchronosuly, and will cause tremendous lag."); + } else if (asyncChunkGenThreadPerWorld) { + log("Async Chunks - Generation: Enabled - Chunks will be generated much faster, without lag."); + } else { + log("Async Chunks - Generation: Enabled (Single Thread) - Chunks will be generated much faster, without lag."); + } + } + } } diff --git a/src/main/java/com/destroystokyo/paper/util/PriorityQueuedExecutor.java b/src/main/java/com/destroystokyo/paper/util/PriorityQueuedExecutor.java new file mode 100644 index 000000000..8f18c2869 --- /dev/null +++ b/src/main/java/com/destroystokyo/paper/util/PriorityQueuedExecutor.java @@ -0,0 +1,347 @@ +package com.destroystokyo.paper.util; + +import javax.annotation.Nonnull; +import java.util.ArrayList; +import java.util.List; +import java.util.concurrent.AbstractExecutorService; +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.ConcurrentLinkedQueue; +import java.util.concurrent.RejectedExecutionException; +import java.util.concurrent.TimeUnit; +import java.util.concurrent.atomic.AtomicBoolean; +import java.util.concurrent.atomic.AtomicInteger; +import java.util.function.Supplier; + +/** + * Implements an Executor Service that allows specifying Task Priority + * and bumping of task priority. + * + * This is a non blocking executor with 3 priority levels. + * + * URGENT: Rarely used, something that is critical to take action now. + * HIGH: Something with more importance than the base tasks + * + * @author Daniel Ennis <aikar@aikar.co> + */ +@SuppressWarnings({"WeakerAccess", "UnusedReturnValue", "unused"}) +public class PriorityQueuedExecutor extends AbstractExecutorService { + + private final ConcurrentLinkedQueue urgent = new ConcurrentLinkedQueue<>(); + private final ConcurrentLinkedQueue high = new ConcurrentLinkedQueue<>(); + private final ConcurrentLinkedQueue normal = new ConcurrentLinkedQueue<>(); + private final List threads = new ArrayList<>(); + private final RejectionHandler handler; + + private volatile boolean shuttingDown = false; + private volatile boolean shuttingDownNow = false; + + public PriorityQueuedExecutor(String name) { + this(name, Math.max(1, Runtime.getRuntime().availableProcessors() - 1)); + } + + public PriorityQueuedExecutor(String name, int threads) { + this(name, threads, Thread.NORM_PRIORITY, null); + } + + public PriorityQueuedExecutor(String name, int threads, int threadPriority) { + this(name, threads, threadPriority, null); + } + + public PriorityQueuedExecutor(String name, int threads, RejectionHandler handler) { + this(name, threads, Thread.NORM_PRIORITY, handler); + } + + public PriorityQueuedExecutor(String name, int threads, int threadPriority, RejectionHandler handler) { + for (int i = 0; i < threads; i++) { + ExecutorThread thread = new ExecutorThread(this::processQueues); + thread.setDaemon(true); + thread.setName(threads == 1 ? name : name + "-" + (i + 1)); + thread.setPriority(threadPriority); + thread.start(); + this.threads.add(thread); + } + if (handler == null) { + handler = ABORT_POLICY; + } + this.handler = handler; + } + + /** + * If the Current thread belongs to a PriorityQueuedExecutor, return that Executro + * @return The executor that controls this thread + */ + public static PriorityQueuedExecutor getExecutor() { + if (!(Thread.currentThread() instanceof ExecutorThread)) { + return null; + } + return ((ExecutorThread) Thread.currentThread()).getExecutor(); + } + + public void shutdown() { + shuttingDown = true; + synchronized (this) { + this.notifyAll(); + } + } + + @Nonnull + @Override + public List shutdownNow() { + shuttingDown = true; + shuttingDownNow = true; + List tasks = new ArrayList<>(high.size() + normal.size()); + Runnable run; + while ((run = getTask()) != null) { + tasks.add(run); + } + + return tasks; + } + + @Override + public boolean isShutdown() { + return shuttingDown; + } + + @Override + public boolean isTerminated() { + if (!shuttingDown) { + return false; + } + return high.isEmpty() && normal.isEmpty(); + } + + @Override + public boolean awaitTermination(long timeout, @Nonnull TimeUnit unit) { + synchronized (this) { + this.notifyAll(); + } + final long wait = unit.toNanos(timeout); + final long max = System.nanoTime() + wait; + for (;!threads.isEmpty() && System.nanoTime() < max;) { + threads.removeIf(thread -> !thread.isAlive()); + } + return isTerminated(); + } + + + public PendingTask createPendingTask(Runnable task) { + return createPendingTask(task, Priority.NORMAL); + } + public PendingTask createPendingTask(Runnable task, Priority priority) { + return createPendingTask(() -> { + task.run(); + return null; + }, priority); + } + + public PendingTask createPendingTask(Supplier task) { + return createPendingTask(task, Priority.NORMAL); + } + + public PendingTask createPendingTask(Supplier task, Priority priority) { + return new PendingTask<>(task, priority); + } + + public PendingTask submitTask(Runnable run) { + return createPendingTask(run).submit(); + } + + public PendingTask submitTask(Runnable run, Priority priority) { + return createPendingTask(run, priority).submit(); + } + + public PendingTask submitTask(Supplier run) { + return createPendingTask(run).submit(); + } + + public PendingTask submitTask(Supplier run, Priority priority) { + PendingTask task = createPendingTask(run, priority); + return task.submit(); + } + + @Override + public void execute(@Nonnull Runnable command) { + submitTask(command); + } + + public boolean isCurrentThread() { + final Thread thread = Thread.currentThread(); + if (!(thread instanceof ExecutorThread)) { + return false; + } + return ((ExecutorThread) thread).getExecutor() == this; + } + + public Runnable getUrgentTask() { + return urgent.poll(); + } + + public Runnable getTask() { + Runnable run = urgent.poll(); + if (run != null) { + return run; + } + run = high.poll(); + if (run != null) { + return run; + } + return normal.poll(); + } + + private void processQueues() { + Runnable run = null; + while (true) { + if (run != null) { + run.run(); + } + if (shuttingDownNow) { + return; + } + if ((run = getTask()) != null) { + continue; + } + synchronized (PriorityQueuedExecutor.this) { + if ((run = getTask()) != null) { + continue; + } + + if (shuttingDown || shuttingDownNow) { + return; + } + try { + PriorityQueuedExecutor.this.wait(); + } catch (InterruptedException ignored) { + } + } + } + } + + public boolean processUrgentTasks() { + Runnable run; + boolean hadTask = false; + while ((run = getUrgentTask()) != null) { + run.run(); + hadTask = true; + } + return hadTask; + } + + public enum Priority { + NORMAL, HIGH, URGENT + } + + public class ExecutorThread extends Thread { + public ExecutorThread(Runnable runnable) { + super(runnable); + } + + public PriorityQueuedExecutor getExecutor() { + return PriorityQueuedExecutor.this; + } + } + + public class PendingTask implements Runnable { + + private final AtomicBoolean hasRan = new AtomicBoolean(); + private final AtomicInteger submitted = new AtomicInteger(-1); + private final AtomicInteger priority; + private final Supplier run; + private final CompletableFuture future = new CompletableFuture<>(); + private volatile PriorityQueuedExecutor executor; + + public PendingTask(Supplier run) { + this(run, Priority.NORMAL); + } + + public PendingTask(Supplier run, Priority priority) { + this.priority = new AtomicInteger(priority.ordinal()); + this.run = run; + } + + public boolean cancel() { + return hasRan.compareAndSet(false, true); + } + + @Override + public void run() { + if (!hasRan.compareAndSet(false, true)) { + return; + } + + try { + future.complete(run.get()); + } catch (Throwable e) { + future.completeExceptionally(e); + } + } + + public void bumpPriority() { + bumpPriority(Priority.HIGH); + } + + public void bumpPriority(Priority newPriority) { + for (;;) { + int current = this.priority.get(); + int ordinal = newPriority.ordinal(); + if (current >= ordinal || priority.compareAndSet(current, ordinal)) { + break; + } + } + + + if (this.submitted.get() == -1 || this.hasRan.get()) { + return; + } + + // Only resubmit if it hasnt ran yet and has been submitted + submit(); + } + + public CompletableFuture onDone() { + return future; + } + + public PendingTask submit() { + if (shuttingDown) { + handler.onRejection(this, PriorityQueuedExecutor.this); + return this; + } + for (;;) { + final int submitted = this.submitted.get(); + final int priority = this.priority.get(); + if (submitted == priority) { + return this; + } + if (this.submitted.compareAndSet(submitted, priority)) { + if (priority == Priority.URGENT.ordinal()) { + urgent.add(this); + } else if (priority == Priority.HIGH.ordinal()) { + high.add(this); + } else { + normal.add(this); + } + + break; + } + } + + synchronized (PriorityQueuedExecutor.this) { + // Wake up a thread to take this work + PriorityQueuedExecutor.this.notify(); + } + return this; + } + } + public interface RejectionHandler { + void onRejection(Runnable run, PriorityQueuedExecutor executor); + } + + public static final RejectionHandler ABORT_POLICY = (run, executor) -> { + throw new RejectedExecutionException("Executor has been shutdown"); + }; + public static final RejectionHandler CALLER_RUNS_POLICY = (run, executor) -> { + run.run(); + }; + +} diff --git a/src/main/java/net/minecraft/server/Chunk.java b/src/main/java/net/minecraft/server/Chunk.java index 44d59d253..0bd975b6d 100644 --- a/src/main/java/net/minecraft/server/Chunk.java +++ b/src/main/java/net/minecraft/server/Chunk.java @@ -184,6 +184,7 @@ public class Chunk implements IChunkAccess { for (k = 0; k < this.sections.length; ++k) { this.sections[k] = protochunk.getSections()[k]; + if (this.sections[k] != null) this.sections[k].disableLocks(); // Paper - Async Chunks - disable locks used during world gen } Iterator iterator = protochunk.s().iterator(); diff --git a/src/main/java/net/minecraft/server/ChunkMap.java b/src/main/java/net/minecraft/server/ChunkMap.java index e14ae2b42..1662e4eba 100644 --- a/src/main/java/net/minecraft/server/ChunkMap.java +++ b/src/main/java/net/minecraft/server/ChunkMap.java @@ -14,9 +14,17 @@ public class ChunkMap extends Long2ObjectOpenHashMap { } public Chunk a(long i, Chunk chunk) { + org.spigotmc.AsyncCatcher.catchOp("Async Chunk put"); // Paper chunk.world.timings.syncChunkLoadPostTimer.startTiming(); // Paper lastChunkByPos = chunk; // Paper - Chunk chunk1 = (Chunk) super.put(i, chunk); + // Paper start + Chunk chunk1; + synchronized (this) { + // synchronize so any async gets are safe + chunk1 = (Chunk) super.put(i, chunk); + } + if (chunk1 == null) { // Paper - we should never be overwriting chunks + // Paper end ChunkCoordIntPair chunkcoordintpair = new ChunkCoordIntPair(i); for (int j = chunkcoordintpair.x - 1; j <= chunkcoordintpair.x + 1; ++j) { @@ -47,7 +55,11 @@ public class ChunkMap extends Long2ObjectOpenHashMap { chunk.setNeighborLoaded(x, z); } } + // Paper start + } } else { + a.error("Overwrote existing chunk! (" + chunk.world.getWorld().getName() + ":" + chunk.locX+"," + chunk.locZ + ")", new IllegalStateException()); } + // Paper end // Paper start - if this is a spare chunk (not part of any players view distance), go ahead and queue it for unload. if (!((WorldServer)chunk.world).getPlayerChunkMap().isChunkInUse(chunk.locX, chunk.locZ)) { if (chunk.world.paperConfig.delayChunkUnloadsBy > 0) { @@ -64,11 +76,19 @@ public class ChunkMap extends Long2ObjectOpenHashMap { } public Chunk a(Long olong, Chunk chunk) { - return this.a(olong, chunk); + return MCUtil.ensureMain("Chunk Put", () -> this.a(olong.longValue(), chunk)); // Paper } public Chunk a(long i) { - Chunk chunk = (Chunk) super.remove(i); + // Paper start + org.spigotmc.AsyncCatcher.catchOp("Async Chunk remove"); + Chunk chunk; + synchronized (this) { + // synchronize so any async gets are safe + chunk = super.remove(i); + } + if (chunk != null) { // Paper - don't decrement if we didn't remove anything + // Paper end ChunkCoordIntPair chunkcoordintpair = new ChunkCoordIntPair(i); for (int j = chunkcoordintpair.x - 1; j <= chunkcoordintpair.x + 1; ++j) { @@ -84,6 +104,7 @@ public class ChunkMap extends Long2ObjectOpenHashMap { } // Paper start + } // close if (chunk != null) if (lastChunkByPos != null && i == lastChunkByPos.chunkKey) { lastChunkByPos = null; } @@ -93,15 +114,22 @@ public class ChunkMap extends Long2ObjectOpenHashMap { @Override public Chunk get(long l) { - if (lastChunkByPos != null && l == lastChunkByPos.chunkKey) { - return lastChunkByPos; + if (MCUtil.isMainThread()) { + if (lastChunkByPos != null && l == lastChunkByPos.chunkKey) { + return lastChunkByPos; + } + final Chunk chunk = super.get(l); + return chunk != null ? (lastChunkByPos = chunk) : null; + } else { + synchronized (this) { + return super.get(l); + } } - return lastChunkByPos = super.get(l); } // Paper end public Chunk a(Object object) { - return this.a((Long) object); + return MCUtil.ensureMain("Chunk Remove", () -> this.a(((Long) object).longValue())); // Paper } public void putAll(Map map) { @@ -114,11 +142,11 @@ public class ChunkMap extends Long2ObjectOpenHashMap { // CraftBukkit start - decompile errors public Chunk remove(long i) { - return this.a(i); + return MCUtil.ensureMain("Chunk Remove", () -> this.a(i)); // Paper } public Chunk put(long i, Chunk object) { - return this.a(i, (Chunk) object); + return MCUtil.ensureMain("Chunk Put", () -> this.a(i, (Chunk) object)); // Paper } public Chunk remove(Object object) { diff --git a/src/main/java/net/minecraft/server/ChunkProviderServer.java b/src/main/java/net/minecraft/server/ChunkProviderServer.java index dcc6e9762..87a1bddd0 100644 --- a/src/main/java/net/minecraft/server/ChunkProviderServer.java +++ b/src/main/java/net/minecraft/server/ChunkProviderServer.java @@ -35,12 +35,12 @@ public class ChunkProviderServer implements IChunkProvider { private long lastProcessedSaves = 0L; // Paper private long lastSaveStatPrinted = System.currentTimeMillis(); // Paper end - public final Long2ObjectMap chunks = Long2ObjectMaps.synchronize(new ChunkMap(8192)); + public final Long2ObjectMap chunks = new ChunkMap(8192); // Paper - remove synchronize - we keep everything on main for manip private Chunk lastChunk; private final ChunkTaskScheduler chunkScheduler; - private final SchedulerBatch batchScheduler; + final SchedulerBatch batchScheduler; // Paper public final WorldServer world; - private final IAsyncTaskHandler asyncTaskHandler; + final IAsyncTaskHandler asyncTaskHandler; // Paper public ChunkProviderServer(WorldServer worldserver, IChunkLoader ichunkloader, ChunkGenerator chunkgenerator, IAsyncTaskHandler iasynctaskhandler) { this.world = worldserver; @@ -77,10 +77,76 @@ public class ChunkProviderServer implements IChunkProvider { this.unloadQueue.remove(ChunkCoordIntPair.a(i, j)); } + // Paper start - defaults if Async Chunks is not enabled + boolean chunkGoingToExists(int x, int z) { + final long k = ChunkCoordIntPair.asLong(x, z); + return chunkScheduler.progressCache.containsKey(k); + } + public void bumpPriority(ChunkCoordIntPair coords) { + // do nothing, override in async + } + + public List getSpiralOutChunks(BlockPosition blockposition, int radius) { + List list = com.google.common.collect.Lists.newArrayList(); + + for (int r = 1; r <= radius; r++) { + int x = -r; + int z = r; + list.add(new ChunkCoordIntPair(blockposition.getX(), blockposition.getZ())); + // Iterates the edge of half of the box; then negates for other half. + while (x <= r && z > -r) { + list.add(new ChunkCoordIntPair(blockposition.getX() + x, blockposition.getZ() + z)); + list.add(new ChunkCoordIntPair(blockposition.getX() - x, blockposition.getZ() - z)); + + if (x < r) { + x++; + } else { + z--; + } + } + } + return list; + } + + public Chunk getChunkAt(int x, int z, boolean load, boolean gen, Consumer consumer) { + return getChunkAt(x, z, load, gen, false, consumer); + } + public Chunk getChunkAt(int x, int z, boolean load, boolean gen, boolean priority, Consumer consumer) { + Chunk chunk = getChunkAt(x, z, load, gen); + if (consumer != null) { + consumer.accept(chunk); + } + return chunk; + } + + PaperAsyncChunkProvider.CancellableChunkRequest requestChunk(int x, int z, boolean gen, boolean priority, Consumer consumer) { + Chunk chunk = getChunkAt(x, z, gen, priority, consumer); + return new PaperAsyncChunkProvider.CancellableChunkRequest() { + @Override + public void cancel() { + + } + + @Override + public Chunk getChunk() { + return chunk; + } + }; + } + // Paper end + @Nullable public Chunk getChunkAt(int i, int j, boolean flag, boolean flag1) { IChunkLoader ichunkloader = this.chunkLoader; Chunk chunk; + // Paper start - do already loaded checks before synchronize + long k = ChunkCoordIntPair.a(i, j); + chunk = (Chunk) this.chunks.get(k); + if (chunk != null) { + //this.lastChunk = chunk; // Paper remove vanilla lastChunk + return chunk; + } + // Paper end synchronized (this.chunkLoader) { // Paper start - remove vanilla lastChunk, we do it more accurately @@ -88,13 +154,15 @@ public class ChunkProviderServer implements IChunkProvider { return this.lastChunk; }*/ // Paper end - long k = ChunkCoordIntPair.a(i, j); + // Paper start - move up + //long k = ChunkCoordIntPair.a(i, j); - chunk = (Chunk) this.chunks.get(k); + /*chunk = (Chunk) this.chunks.get(k); if (chunk != null) { //this.lastChunk = chunk; // Paper remove vanilla lastChunk return chunk; - } + }*/ + // Paper end if (flag) { try (co.aikar.timings.Timing timing = world.timings.syncChunkLoadTimer.startTiming()) { // Paper @@ -150,7 +218,8 @@ public class ChunkProviderServer implements IChunkProvider { return (IChunkAccess) (chunk != null ? chunk : (IChunkAccess) this.chunkScheduler.b(new ChunkCoordIntPair(i, j), flag)); } - public CompletableFuture a(Iterable iterable, Consumer consumer) { + public CompletableFuture loadAllChunks(Iterable iterable, Consumer consumer) { return a(iterable, consumer).thenCompose(protoChunk -> null); } // Paper - overriden in async chunk provider + private CompletableFuture a(Iterable iterable, Consumer consumer) { // Paper - mark private, use above method this.batchScheduler.b(); Iterator iterator = iterable.iterator(); @@ -168,6 +237,7 @@ public class ChunkProviderServer implements IChunkProvider { return this.batchScheduler.c(); } + ReportedException generateChunkError(int i, int j, Throwable throwable) { return a(i, j, throwable); } // Paper - OBFHELPER private ReportedException a(int i, int j, Throwable throwable) { CrashReport crashreport = CrashReport.a(throwable, "Exception generating new chunk"); CrashReportSystemDetails crashreportsystemdetails = crashreport.a("Chunk to be generated"); @@ -289,11 +359,13 @@ public class ChunkProviderServer implements IChunkProvider { } public void close() { - try { + // Paper start - we do not need to wait for chunk generations to finish on close + /*try { this.batchScheduler.a(); } catch (InterruptedException interruptedexception) { ChunkProviderServer.a.error("Couldn't stop taskManager", interruptedexception); - } + }*/ + // Paper end } diff --git a/src/main/java/net/minecraft/server/ChunkRegionLoader.java b/src/main/java/net/minecraft/server/ChunkRegionLoader.java index 26a3f28cb..5ce57a6d4 100644 --- a/src/main/java/net/minecraft/server/ChunkRegionLoader.java +++ b/src/main/java/net/minecraft/server/ChunkRegionLoader.java @@ -120,7 +120,7 @@ public class ChunkRegionLoader implements IChunkLoader, IAsyncChunkSaver { // CraftBukkit start private boolean check(ChunkProviderServer cps, int x, int z) throws IOException { if (cps != null) { - com.google.common.base.Preconditions.checkState(org.bukkit.Bukkit.isPrimaryThread(), "primary thread"); + //com.google.common.base.Preconditions.checkState(org.bukkit.Bukkit.isPrimaryThread(), "primary thread"); // Paper - this is safe if (cps.isLoaded(x, z)) { return true; } @@ -380,11 +380,12 @@ public class ChunkRegionLoader implements IChunkLoader, IAsyncChunkSaver { } }; } else { + /* // Paper start - we will never invoke this in an unsafe way NBTTagCompound nbttagcompound2 = this.a(world, chunkcoordintpair.x, chunkcoordintpair.z); if (nbttagcompound2 != null && this.a(nbttagcompound2) == ChunkStatus.Type.LEVELCHUNK) { return; - } + }*/ // Paper end completion = new Supplier() { public NBTTagCompound get() { diff --git a/src/main/java/net/minecraft/server/ChunkSection.java b/src/main/java/net/minecraft/server/ChunkSection.java index bdfc7d81f..a5c4564d6 100644 --- a/src/main/java/net/minecraft/server/ChunkSection.java +++ b/src/main/java/net/minecraft/server/ChunkSection.java @@ -24,7 +24,17 @@ public class ChunkSection { this.skyLight = new NibbleArray(); } + // Paper start - Async Chunks - Lock during world gen + if (chunk instanceof ProtoChunk) { + this.blockIds.enableLocks(); + } else { + this.blockIds.disableLocks(); + } + } + void disableLocks() { + this.blockIds.disableLocks(); } + // Paper end public IBlockData getType(int i, int j, int k) { return this.blockIds.a(i, j, k); diff --git a/src/main/java/net/minecraft/server/ChunkTaskScheduler.java b/src/main/java/net/minecraft/server/ChunkTaskScheduler.java index ad7e4a036..f10f235ed 100644 --- a/src/main/java/net/minecraft/server/ChunkTaskScheduler.java +++ b/src/main/java/net/minecraft/server/ChunkTaskScheduler.java @@ -20,13 +20,14 @@ public class ChunkTaskScheduler extends Scheduler d; private final IChunkLoader e; private final IAsyncTaskHandler f; - private final Long2ObjectMap progressCache = new ExpiringMap(8192, 5000) { // CraftBukkit - decompile error + final Long2ObjectMap progressCache = new ExpiringMap(8192, 5000) { // CraftBukkit - decompile error // Paper - protected protected boolean a(Scheduler.a scheduler_a) { ProtoChunk protochunk = (ProtoChunk) scheduler_a.a(); return !protochunk.ab_() /*&& !protochunk.h()*/; // Paper } }; + private final Long2ObjectMap> pendingSchedulers = new it.unimi.dsi.fastutil.longs.Long2ObjectOpenHashMap<>(); // Paper public ChunkTaskScheduler(int i, World world, ChunkGenerator chunkgenerator, IChunkLoader ichunkloader, IAsyncTaskHandler iasynctaskhandler) { super("WorldGen", i, ChunkStatus.FINALIZED, () -> { @@ -50,8 +51,28 @@ public class ChunkTaskScheduler extends Scheduler { + // Paper start - refactor a lot of this - avoid generating a chunk while holding lock on expiring map + java.util.concurrent.CompletableFuture pending = null; + boolean created = false; + long key = chunkcoordintpair.a(); + synchronized (pendingSchedulers) { + Scheduler.a existing = this.progressCache.get(key); + if (existing != null) { + return existing; + } + pending = this.pendingSchedulers.get(key); + if (pending == null) { + if (!flag) { + return null; + } + created = true; + pending = new java.util.concurrent.CompletableFuture<>(); + pendingSchedulers.put(key, pending); + } + } + if (created) { + java.util.function.Function get = (i) -> { + // Paper end ProtoChunk protochunk; try { @@ -70,8 +91,18 @@ public class ChunkTaskScheduler extends Scheduler map) { diff --git a/src/main/java/net/minecraft/server/DataPaletteBlock.java b/src/main/java/net/minecraft/server/DataPaletteBlock.java index 71a3636be..ff0fe2541 100644 --- a/src/main/java/net/minecraft/server/DataPaletteBlock.java +++ b/src/main/java/net/minecraft/server/DataPaletteBlock.java @@ -3,7 +3,7 @@ package net.minecraft.server; import com.destroystokyo.paper.antixray.ChunkPacketInfo; // Paper - Anti-Xray import java.util.Arrays; import java.util.Objects; -import java.util.concurrent.locks.ReentrantLock; +import java.util.concurrent.locks.ReentrantReadWriteLock; import java.util.function.Function; import java.util.stream.Collectors; @@ -20,25 +20,42 @@ public class DataPaletteBlock implements DataPaletteExpandable { protected DataBits a; protected DataBits getDataBits() { return this.a; } // Paper - OBFHELPER private DataPalette h; private DataPalette getDataPalette() { return this.h; } // Paper - OBFHELPER private int i; private int getBitsPerObject() { return this.i; } // Paper - OBFHELPER - private final ReentrantLock j = new ReentrantLock(); + // Paper start - use read write locks only during generation, disable once back on main thread + private static final NoopLock NOOP_LOCK = new NoopLock(); + private java.util.concurrent.locks.Lock readLock = NOOP_LOCK; + private java.util.concurrent.locks.Lock writeLock = NOOP_LOCK; + + private static class NoopLock extends ReentrantReadWriteLock.WriteLock { + private NoopLock() { + super(new ReentrantReadWriteLock()); + } + + @Override + public final void lock() { + } + + @Override + public final void unlock() { - private void b() { - if (this.j.isLocked() && !this.j.isHeldByCurrentThread()) { - String s = (String)Thread.getAllStackTraces().keySet().stream().filter(Objects::nonNull).map((thread) -> { - return thread.getName() + ": \n\tat " + (String)Arrays.stream(thread.getStackTrace()).map(Object::toString).collect(Collectors.joining("\n\tat ")); - }).collect(Collectors.joining("\n")); - CrashReport crashreport = new CrashReport("Writing into PalettedContainer from multiple threads", new IllegalStateException()); - CrashReportSystemDetails crashreportsystemdetails = crashreport.a("Thread dumps"); - crashreportsystemdetails.a("Thread dumps", s); - throw new ReportedException(crashreport); - } else { - this.j.lock(); } } + synchronized void enableLocks() { + ReentrantReadWriteLock lock = new ReentrantReadWriteLock(); + readLock = lock.readLock(); + writeLock = lock.writeLock(); + } + synchronized void disableLocks() { + readLock = NOOP_LOCK; + writeLock = NOOP_LOCK; + } + private void b() { + writeLock.lock(); + } private void c() { - this.j.unlock(); + writeLock.unlock(); } + // Paper end public DataPaletteBlock(DataPalette datapalette, RegistryBlockID registryblockid, Function function, Function function1, T object) { // Paper start - Anti-Xray - Support default constructor @@ -147,8 +164,13 @@ public class DataPaletteBlock implements DataPaletteExpandable { } protected T a(int ix) { - T object = this.h.a(this.a.a(ix)); // Paper - decompile fix - return (T)(object == null ? this.g : object); + try { // Paper start - read lock + readLock.lock(); + T object = this.h.a(this.a.a(ix)); // Paper - decompile fix + return (T)(object == null ? this.g : object); + } finally { + readLock.unlock(); + } // Paper end } // Paper start - Anti-Xray - Support default methods diff --git a/src/main/java/net/minecraft/server/DefinedStructureManager.java b/src/main/java/net/minecraft/server/DefinedStructureManager.java index 271dc41d4..bd15534c2 100644 --- a/src/main/java/net/minecraft/server/DefinedStructureManager.java +++ b/src/main/java/net/minecraft/server/DefinedStructureManager.java @@ -19,7 +19,7 @@ import org.apache.logging.log4j.Logger; public class DefinedStructureManager implements IResourcePackListener { private static final Logger a = LogManager.getLogger(); - private final Map b = Maps.newHashMap(); + private final Map b = Maps.newConcurrentMap(); // Paper private final DataFixer c; private final MinecraftServer d; private final java.nio.file.Path e; diff --git a/src/main/java/net/minecraft/server/Entity.java b/src/main/java/net/minecraft/server/Entity.java index 522af3d53..ee878191a 100644 --- a/src/main/java/net/minecraft/server/Entity.java +++ b/src/main/java/net/minecraft/server/Entity.java @@ -209,7 +209,7 @@ public abstract class Entity implements INamableTileEntity, ICommandListener, Ke this.random = SHARED_RANDOM; // Paper this.fireTicks = -this.getMaxFireTicks(); this.justCreated = true; - this.uniqueID = MathHelper.a(this.random); + this.uniqueID = MathHelper.a(java.util.concurrent.ThreadLocalRandom.current()); // Paper this.au = this.uniqueID.toString(); this.aJ = Sets.newHashSet(); this.aL = new double[] { 0.0D, 0.0D, 0.0D}; diff --git a/src/main/java/net/minecraft/server/IChunkLoader.java b/src/main/java/net/minecraft/server/IChunkLoader.java index 4698ee99f..dfb45cc4e 100644 --- a/src/main/java/net/minecraft/server/IChunkLoader.java +++ b/src/main/java/net/minecraft/server/IChunkLoader.java @@ -6,6 +6,8 @@ import javax.annotation.Nullable; public interface IChunkLoader { + void loadEntities(NBTTagCompound nbttagcompound, Chunk chunk); // Paper - Async Chunks + Object[] loadChunk(GeneratorAccess generatoraccess, int i, int j, Consumer consumer) throws IOException; // Paper - Async Chunks @Nullable Chunk a(GeneratorAccess generatoraccess, int i, int j, Consumer consumer) throws IOException; diff --git a/src/main/java/net/minecraft/server/MathHelper.java b/src/main/java/net/minecraft/server/MathHelper.java index 49fba0979..9ad646f8d 100644 --- a/src/main/java/net/minecraft/server/MathHelper.java +++ b/src/main/java/net/minecraft/server/MathHelper.java @@ -142,6 +142,7 @@ public class MathHelper { return Math.floorMod(i, j); } + public static float normalizeYaw(float fx) { return g(fx); } // Paper - OBFHELPER public static float g(float fx) { fx = fx % 360.0F; if (fx >= 180.0F) { diff --git a/src/main/java/net/minecraft/server/MinecraftServer.java b/src/main/java/net/minecraft/server/MinecraftServer.java index c73a61d94..17ae26cb8 100644 --- a/src/main/java/net/minecraft/server/MinecraftServer.java +++ b/src/main/java/net/minecraft/server/MinecraftServer.java @@ -503,6 +503,7 @@ public abstract class MinecraftServer implements IAsyncTaskHandler, IMojangStati // CraftBukkit start - fire WorldLoadEvent and handle whether or not to keep the spawn in memory Stopwatch stopwatch = Stopwatch.createStarted(); + boolean waitForChunks = Boolean.getBoolean("paper.waitforchunks"); // Paper for (WorldServer worldserver : this.getWorlds()) { MinecraftServer.LOGGER.info("Preparing start region for level " + worldserver.dimension + " (Seed: " + worldserver.getSeed() + ")"); if (!worldserver.getWorld().getKeepSpawnInMemory()) { @@ -510,29 +511,25 @@ public abstract class MinecraftServer implements IAsyncTaskHandler, IMojangStati } BlockPosition blockposition = worldserver.getSpawn(); - ArrayList arraylist = Lists.newArrayList(); + List arraylist = worldserver.getChunkProviderServer().getSpiralOutChunks(blockposition, worldserver.paperConfig.keepLoadedRange >> 4); // Paper Set set = Sets.newConcurrentHashSet(); - // Paper start - short radius = worldserver.paperConfig.keepLoadedRange; - for (int i = -radius; i <= radius && this.isRunning(); i += 16) { - for (int j = -radius; j <= radius && this.isRunning(); j += 16) { - // Paper end - arraylist.add(new ChunkCoordIntPair(blockposition.getX() + i >> 4, blockposition.getZ() + j >> 4)); - } - } // Paper + // Paper - remove arraylist creation, call spiral above if (this.isRunning()) { // Paper int expected = arraylist.size(); // Paper - CompletableFuture completablefuture = worldserver.getChunkProviderServer().a((Iterable) arraylist, (chunk) -> { + CompletableFuture completablefuture = worldserver.getChunkProviderServer().loadAllChunks(arraylist, (chunk) -> { // Paper set.add(chunk.getPos()); - if (set.size() < expected && set.size() % 25 == 0) this.a(new ChatMessage("menu.preparingSpawn", new Object[0]), set.size() * 100 / expected); // Paper + if (waitForChunks && (set.size() == expected || (set.size() < expected && set.size() % (set.size() / 10) == 0))) { + this.a(new ChatMessage("menu.preparingSpawn", new Object[0]), set.size() * 100 / expected); // Paper + } }); - while (!completablefuture.isDone()) { + while (waitForChunks && !completablefuture.isDone() && isRunning()) { // Paper try { - completablefuture.get(1L, TimeUnit.SECONDS); + PaperAsyncChunkProvider.processMainThreadQueue(this); // Paper + completablefuture.get(50L, TimeUnit.MILLISECONDS); // Paper } catch (InterruptedException interruptedexception) { throw new RuntimeException(interruptedexception); } catch (ExecutionException executionexception) { @@ -542,11 +539,11 @@ public abstract class MinecraftServer implements IAsyncTaskHandler, IMojangStati throw new RuntimeException(executionexception.getCause()); } catch (TimeoutException timeoutexception) { - this.a(new ChatMessage("menu.preparingSpawn", new Object[0]), set.size() * 100 / expected); // Paper + //this.a(new ChatMessage("menu.preparingSpawn", new Object[0]), set.size() * 100 / expected); // Paper } } - this.a(new ChatMessage("menu.preparingSpawn", new Object[0]), set.size() * 100 / expected); // Paper + if (waitForChunks) this.a(new ChatMessage("menu.preparingSpawn", new Object[0]), set.size() * 100 / expected); // Paper } } @@ -650,6 +647,7 @@ public abstract class MinecraftServer implements IAsyncTaskHandler, IMojangStati if (hasStopped) return; hasStopped = true; } + PaperAsyncChunkProvider.stop(this); // Paper // CraftBukkit end MinecraftServer.LOGGER.info("Stopping server"); MinecraftTimings.stopServer(); // Paper @@ -1017,6 +1015,7 @@ public abstract class MinecraftServer implements IAsyncTaskHandler, IMojangStati while ((futuretask = (FutureTask) this.f.poll()) != null) { SystemUtils.a(futuretask, MinecraftServer.LOGGER); } + PaperAsyncChunkProvider.processMainThreadQueue(this); // Paper MinecraftTimings.minecraftSchedulerTimer.stopTiming(); // Paper this.methodProfiler.c("commandFunctions"); @@ -1053,6 +1052,7 @@ public abstract class MinecraftServer implements IAsyncTaskHandler, IMojangStati // CraftBukkit - dropTickTime for (Iterator iterator = this.getWorlds().iterator(); iterator.hasNext();) { WorldServer worldserver = (WorldServer) iterator.next(); + PaperAsyncChunkProvider.processMainThreadQueue(worldserver); // Paper TileEntityHopper.skipHopperEvents = worldserver.paperConfig.disableHopperMoveEvents || org.bukkit.event.inventory.InventoryMoveItemEvent.getHandlerList().getRegisteredListeners().length == 0; // Paper i = SystemUtils.c(); if (true || worldserver.worldProvider.getDimensionManager() == DimensionManager.OVERWORLD || this.getAllowNether()) { // CraftBukkit diff --git a/src/main/java/net/minecraft/server/PaperAsyncChunkProvider.java b/src/main/java/net/minecraft/server/PaperAsyncChunkProvider.java new file mode 100644 index 000000000..e9a38f9d9 --- /dev/null +++ b/src/main/java/net/minecraft/server/PaperAsyncChunkProvider.java @@ -0,0 +1,655 @@ +/* + * This file is licensed under the MIT License (MIT). + * + * Copyright (c) 2018 Daniel Ennis + * + * Permission is hereby granted, free of charge, to any person obtaining a copy + * of this software and associated documentation files (the "Software"), to deal + * in the Software without restriction, including without limitation the rights + * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell + * copies of the Software, and to permit persons to whom the Software is + * furnished to do so, subject to the following conditions: + * + * The above copyright notice and this permission notice shall be included in + * all copies or substantial portions of the Software. + * + * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR + * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, + * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE + * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER + * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, + * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN + * THE SOFTWARE. + */ +package net.minecraft.server; + +import com.destroystokyo.paper.PaperConfig; +import com.destroystokyo.paper.util.PriorityQueuedExecutor; +import com.destroystokyo.paper.util.PriorityQueuedExecutor.Priority; +import it.unimi.dsi.fastutil.longs.Long2ObjectMap; +import it.unimi.dsi.fastutil.longs.Long2ObjectMaps; +import it.unimi.dsi.fastutil.longs.Long2ObjectOpenHashMap; +import org.bukkit.Bukkit; +import org.bukkit.craftbukkit.generator.CustomChunkGenerator; +import org.bukkit.craftbukkit.generator.InternalChunkGenerator; + +import javax.annotation.Nullable; +import java.io.IOException; +import java.util.ArrayList; +import java.util.Iterator; +import java.util.List; +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.ConcurrentLinkedDeque; +import java.util.concurrent.atomic.AtomicBoolean; +import java.util.concurrent.atomic.AtomicInteger; +import java.util.function.Consumer; + +@SuppressWarnings("unused") +public class PaperAsyncChunkProvider extends ChunkProviderServer { + + private static final int GEN_THREAD_PRIORITY = Integer.getInteger("paper.genThreadPriority", 3); + private static final int LOAD_THREAD_PRIORITY = Integer.getInteger("paper.loadThreadPriority", 4); + private static final PriorityQueuedExecutor EXECUTOR = new PriorityQueuedExecutor("PaperChunkLoader", PaperConfig.asyncChunks ? PaperConfig.asyncChunkLoadThreads : 0, LOAD_THREAD_PRIORITY); + private static final PriorityQueuedExecutor SINGLE_GEN_EXECUTOR = new PriorityQueuedExecutor("PaperChunkGenerator", PaperConfig.asyncChunks && PaperConfig.asyncChunkGeneration && !PaperConfig.asyncChunkGenThreadPerWorld ? 1 : 0, GEN_THREAD_PRIORITY); + private static final ConcurrentLinkedDeque MAIN_THREAD_QUEUE = new ConcurrentLinkedDeque<>(); + + private final PriorityQueuedExecutor generationExecutor; + //private static final PriorityQueuedExecutor generationExecutor = new PriorityQueuedExecutor("PaperChunkGen", 1); + private final Long2ObjectMap pendingChunks = Long2ObjectMaps.synchronize(new Long2ObjectOpenHashMap<>()); + private final IAsyncTaskHandler asyncHandler; + + private final WorldServer world; + private final IChunkLoader chunkLoader; + private final MinecraftServer server; + private final boolean shouldGenSync; + + public PaperAsyncChunkProvider(WorldServer world, IChunkLoader chunkLoader, InternalChunkGenerator generator, MinecraftServer server) { + super(world, chunkLoader, generator, server); + + this.server = world.getMinecraftServer(); + this.world = world; + this.asyncHandler = server; + this.chunkLoader = chunkLoader; + String worldName = this.world.getWorld().getName(); + this.shouldGenSync = generator instanceof CustomChunkGenerator && !(((CustomChunkGenerator) generator).asyncSupported) || !PaperConfig.asyncChunkGeneration; + this.generationExecutor = PaperConfig.asyncChunkGenThreadPerWorld ? new PriorityQueuedExecutor("PaperChunkGen-" + worldName, shouldGenSync ? 0 : 1, GEN_THREAD_PRIORITY) : SINGLE_GEN_EXECUTOR; + } + + private static Priority calculatePriority(boolean isBlockingMain, boolean priority) { + if (isBlockingMain) { + return Priority.URGENT; + } + + if (priority) { + return Priority.HIGH; + } + + return Priority.NORMAL; + } + + static void stop(MinecraftServer server) { + for (WorldServer world : server.getWorlds()) { + world.getPlayerChunkMap().shutdown(); + } + } + + static void processMainThreadQueue(MinecraftServer server) { + for (WorldServer world : server.getWorlds()) { + processMainThreadQueue(world); + } + } + + static void processMainThreadQueue(World world) { + IChunkProvider chunkProvider = world.getChunkProvider(); + if (chunkProvider instanceof PaperAsyncChunkProvider) { + ((PaperAsyncChunkProvider) chunkProvider).processMainThreadQueue(); + } + } + + private void processMainThreadQueue() { + processMainThreadQueue((PendingChunk) null); + } + private boolean processMainThreadQueue(PendingChunk pending) { + Runnable run; + boolean hadLoad = false; + while ((run = MAIN_THREAD_QUEUE.poll()) != null) { + run.run(); + hadLoad = true; + if (pending != null && pending.hasPosted) { + break; + } + } + return hadLoad; + } + + @Override + public void bumpPriority(ChunkCoordIntPair coords) { + final PendingChunk pending = pendingChunks.get(coords.asLong()); + if (pending != null) { + pending.bumpPriority(Priority.HIGH); + } + } + + @Nullable + @Override + public Chunk getChunkAt(int x, int z, boolean load, boolean gen) { + return getChunkAt(x, z, load, gen, null); + } + + @Nullable + @Override + public Chunk getChunkAt(int x, int z, boolean load, boolean gen, boolean priority, Consumer consumer) { + final long key = ChunkCoordIntPair.asLong(x, z); + final Chunk chunk = this.chunks.get(key); + if (chunk != null || !load) { // return null if we aren't loading + if (consumer != null) { + consumer.accept(chunk); + } + return chunk; + } + return loadOrGenerateChunk(x, z, gen, priority, consumer); // Async overrides this method + } + + private Chunk loadOrGenerateChunk(int x, int z, boolean gen, boolean priority, Consumer consumer) { + return requestChunk(x, z, gen, priority, consumer).getChunk(); + } + + final PendingChunkRequest requestChunk(int x, int z, boolean gen, boolean priority, Consumer consumer) { + try (co.aikar.timings.Timing timing = world.timings.syncChunkLoadTimer.startTiming()) { + final long key = ChunkCoordIntPair.asLong(x, z); + final boolean isChunkThread = isChunkThread(); + final boolean isBlockingMain = consumer == null && server.isMainThread(); + final boolean loadOnThisThread = isChunkThread || isBlockingMain; + final Priority taskPriority = calculatePriority(isBlockingMain, priority); + + // Obtain a PendingChunk + final PendingChunk pending; + synchronized (pendingChunks) { + PendingChunk pendingChunk = pendingChunks.get(key); + if (pendingChunk == null) { + pending = new PendingChunk(x, z, key, gen, taskPriority); + pendingChunks.put(key, pending); + } else if (pendingChunk.hasFinished && gen && !pendingChunk.canGenerate && pendingChunk.chunk == null) { + // need to overwrite the old + pending = new PendingChunk(x, z, key, true, taskPriority); + pendingChunks.put(key, pending); + } else { + pending = pendingChunk; + if (pending.taskPriority != taskPriority) { + pending.bumpPriority(taskPriority); + } + } + } + + // Listen for when result is ready + final CompletableFuture future = new CompletableFuture<>(); + final PendingChunkRequest request = pending.addListener(future, gen, !loadOnThisThread); + + // Chunk Generation can trigger Chunk Loading, those loads may need to convert, and could be slow + // Give an opportunity for urgent tasks to jump in at these times + if (isChunkThread) { + processUrgentTasks(); + } + + if (loadOnThisThread) { + // do loads on main if blocking, or on current if we are a load/gen thread + // gen threads do trigger chunk loads + pending.loadTask.run(); + } + + if (isBlockingMain) { + while (!future.isDone()) { + // We aren't done, obtain lock on queue + synchronized (MAIN_THREAD_QUEUE) { + // We may of received our request now, check it + if (processMainThreadQueue(pending)) { + // If we processed SOMETHING, don't wait + continue; + } + try { + // We got nothing from the queue, wait until something has been added + MAIN_THREAD_QUEUE.wait(1); + } catch (InterruptedException ignored) { + } + } + // Queue has been notified or timed out, process it + processMainThreadQueue(pending); + } + // We should be done AND posted into chunk map now, return it + request.initialReturnChunk = pending.postChunk(); + } else if (consumer == null) { + // This is on another thread + request.initialReturnChunk = future.join(); + } else { + future.thenAccept((c) -> this.asyncHandler.postToMainThread(() -> consumer.accept(c))); + } + + return request; + } + } + + private void processUrgentTasks() { + final PriorityQueuedExecutor executor = PriorityQueuedExecutor.getExecutor(); + if (executor != null) { + executor.processUrgentTasks(); + } + } + + @Override + public CompletableFuture loadAllChunks(Iterable iterable, Consumer consumer) { + final Iterator iterator = iterable.iterator(); + + final List> all = new ArrayList<>(); + while (iterator.hasNext()) { + final ChunkCoordIntPair chunkcoordintpair = iterator.next(); + final CompletableFuture future = new CompletableFuture<>(); + all.add(future); + this.getChunkAt(chunkcoordintpair.x, chunkcoordintpair.z, true, true, chunk -> { + future.complete(chunk); + if (consumer != null) { + consumer.accept(chunk); + } + }); + } + return CompletableFuture.allOf(all.toArray(new CompletableFuture[0])); + } + + boolean chunkGoingToExists(int x, int z) { + synchronized (pendingChunks) { + PendingChunk pendingChunk = pendingChunks.get(ChunkCoordIntPair.asLong(x, z)); + return pendingChunk != null && pendingChunk.canGenerate; + } + } + + private enum PendingStatus { + /** + * Request has just started + */ + STARTED, + /** + * Chunk is attempting to be loaded from disk + */ + LOADING, + /** + * Chunk must generate on main and is pending main + */ + GENERATION_PENDING, + /** + * Chunk is generating + */ + GENERATING, + /** + * Chunk is ready and is pending post to main + */ + PENDING_MAIN, + /** + * Could not load chunk, and did not need to generat + */ + FAIL, + /** + * Fully done with this request (may or may not of loaded) + */ + DONE, + /** + * Chunk load was cancelled (no longer needed) + */ + CANCELLED + } + + public interface CancellableChunkRequest { + void cancel(); + Chunk getChunk(); + } + + public static class PendingChunkRequest implements CancellableChunkRequest { + private final PendingChunk pending; + private final AtomicBoolean cancelled = new AtomicBoolean(false); + private volatile boolean generating; + private volatile Chunk initialReturnChunk; + + private PendingChunkRequest(PendingChunk pending) { + this.pending = pending; + this.cancelled.set(true); + } + + private PendingChunkRequest(PendingChunk pending, boolean gen) { + this.pending = pending; + this.generating = gen; + } + + public void cancel() { + this.pending.cancel(this); + } + + /** + * Will be null on asynchronous loads + */ + @Override @Nullable + public Chunk getChunk() { + return initialReturnChunk; + } + } + + private boolean isLoadThread() { + return EXECUTOR.isCurrentThread(); + } + + private boolean isGenThread() { + return generationExecutor.isCurrentThread(); + } + private boolean isChunkThread() { + return isLoadThread() || isGenThread(); + } + + private class PendingChunk implements Runnable { + private final int x; + private final int z; + private final long key; + private final long started = System.currentTimeMillis(); + private final CompletableFuture loadOnly = new CompletableFuture<>(); + private final CompletableFuture generate = new CompletableFuture<>(); + private final AtomicInteger requests = new AtomicInteger(0); + + private volatile PendingStatus status = PendingStatus.STARTED; + private volatile PriorityQueuedExecutor.PendingTask loadTask; + private volatile PriorityQueuedExecutor.PendingTask genTask; + private volatile Priority taskPriority; + private volatile boolean generating; + private volatile boolean canGenerate; + private volatile boolean isHighPriority; + private volatile boolean hasPosted; + private volatile boolean hasFinished; + private volatile Chunk chunk; + private volatile NBTTagCompound pendingLevel; + + PendingChunk(int x, int z, long key, boolean canGenerate, boolean priority) { + this.x = x; + this.z = z; + this.key = key; + this.canGenerate = canGenerate; + taskPriority = priority ? Priority.HIGH : Priority.NORMAL; + } + + PendingChunk(int x, int z, long key, boolean canGenerate, Priority taskPriority) { + this.x = x; + this.z = z; + this.key = key; + this.canGenerate = canGenerate; + this.taskPriority = taskPriority; + } + + private synchronized void setStatus(PendingStatus status) { + this.status = status; + } + + private Chunk loadChunk(int x, int z) throws IOException { + setStatus(PendingStatus.LOADING); + Object[] data = chunkLoader.loadChunk(world, x, z, null); + if (data != null) { + // Level must be loaded on main + this.pendingLevel = ((NBTTagCompound) data[1]).getCompound("Level"); + return (Chunk) data[0]; + } else { + return null; + } + } + + private Chunk generateChunk() { + synchronized (this) { + if (requests.get() <= 0) { + return null; + } + } + CompletableFuture pending = new CompletableFuture<>(); + batchScheduler.startBatch(); + batchScheduler.add(new ChunkCoordIntPair(x, z)); + try { + ProtoChunk protoChunk = batchScheduler.executeBatch().join(); + boolean saved = false; + if (!Bukkit.isPrimaryThread()) { + // If we are async, dispatch later + try { + chunkLoader.saveChunk(world, protoChunk, true); + saved = true; + } catch (IOException | ExceptionWorldConflict e) { + e.printStackTrace(); + } + } + Chunk chunk = new Chunk(world, protoChunk, x, z); + if (saved) { + chunk.setLastSaved(world.getTime()); + } + generateFinished(chunk); + + return chunk; + } catch (Exception e) { + MinecraftServer.LOGGER.error("Couldn't generate chunk (" +world.getWorld().getName() + ":" + x + "," + z + ")", e); + generateFinished(null); + return null; + } + } + + boolean loadFinished(Chunk chunk) { + if (chunk != null) { + postChunkToMain(chunk); + return false; + } + loadOnly.complete(null); + + synchronized (this) { + boolean cancelled = requests.get() <= 0; + if (!canGenerate || cancelled) { + if (!cancelled) { + setStatus(PendingStatus.FAIL); + } + this.chunk = null; + this.hasFinished = true; + pendingChunks.remove(key); + return false; + } else { + setStatus(PendingStatus.GENERATING); + generating = true; + return true; + } + } + } + + void generateFinished(Chunk chunk) { + synchronized (this) { + this.chunk = chunk; + this.hasFinished = true; + } + if (chunk != null) { + postChunkToMain(chunk); + } else { + synchronized (this) { + pendingChunks.remove(key); + completeFutures(null); + } + } + } + + synchronized private void completeFutures(Chunk chunk) { + loadOnly.complete(chunk); + generate.complete(chunk); + } + + private void postChunkToMain(Chunk chunk) { + synchronized (this) { + setStatus(PendingStatus.PENDING_MAIN); + this.chunk = chunk; + this.hasFinished = true; + } + + if (server.isMainThread()) { + postChunk(); + return; + } + + // Don't post here, even if on main, it must enter the queue so we can exit any open batch + // schedulers, as post stage may trigger a new generation and cause errors + synchronized (MAIN_THREAD_QUEUE) { + if (this.taskPriority == Priority.URGENT) { + MAIN_THREAD_QUEUE.addFirst(this::postChunk); + } else { + MAIN_THREAD_QUEUE.addLast(this::postChunk); + } + MAIN_THREAD_QUEUE.notify(); + } + } + + Chunk postChunk() { + if (!server.isMainThread()) { + throw new IllegalStateException("Must post from main"); + } + synchronized (this) { + if (hasPosted || requests.get() <= 0) { // if pending is 0, all were cancelled + return chunk; + } + hasPosted = true; + } + try { + if (chunk == null) { + chunk = chunks.get(key); + completeFutures(chunk); + return chunk; + } + if (pendingLevel != null) { + chunkLoader.loadEntities(pendingLevel, chunk); + pendingLevel = null; + } + synchronized (chunks) { + final Chunk other = chunks.get(key); + if (other != null) { + this.chunk = other; + completeFutures(other); + return other; + } + if (chunk != null) { + chunks.put(key, chunk); + } + } + + chunk.addEntities(); + + completeFutures(chunk); + return chunk; + } finally { + pendingChunks.remove(key); + setStatus(PendingStatus.DONE); + } + } + + synchronized PendingChunkRequest addListener(CompletableFuture future, boolean gen, boolean autoSubmit) { + if (hasFinished) { + future.complete(chunk); + return new PendingChunkRequest(this); + } else if (gen) { + canGenerate = true; + generate.thenAccept(future::complete); + } else { + if (generating) { + future.complete(null); + return new PendingChunkRequest(this); + } else { + loadOnly.thenAccept(future::complete); + } + } + + requests.incrementAndGet(); + if (loadTask == null) { + // Take care of a race condition in that a request could be cancelled after the synchronize + // on pendingChunks, but before a listener is added, which would erase these pending tasks. + genTask = generationExecutor.createPendingTask(this::generateChunk, taskPriority); + loadTask = EXECUTOR.createPendingTask(this, taskPriority); + if (autoSubmit) { + // We will execute it outside of the synchronized context immediately after + loadTask.submit(); + } + } + return new PendingChunkRequest(this, gen); + } + + @Override + public void run() { + try { + if (!loadFinished(loadChunk(x, z))) { + return; + } + } catch (Exception ex) { + MinecraftServer.LOGGER.error("Couldn't load chunk (" +world.getWorld().getName() + ":" + x + "," + z + ")", ex); + if (ex instanceof IOException) { + generateFinished(null); + return; + } + } + + if (shouldGenSync) { + synchronized (this) { + setStatus(PendingStatus.GENERATION_PENDING); + if (this.taskPriority == Priority.URGENT) { + MAIN_THREAD_QUEUE.addFirst(() -> generateFinished(this.generateChunk())); + } else { + MAIN_THREAD_QUEUE.addLast(() -> generateFinished(this.generateChunk())); + } + + } + synchronized (MAIN_THREAD_QUEUE) { + MAIN_THREAD_QUEUE.notify(); + } + } else { + if (isGenThread()) { + // ideally we should never run into 1 chunk generating another chunk... + // but if we do, let's apply same solution + genTask.run(); + } else { + genTask.submit(); + } + } + } + + void bumpPriority() { + bumpPriority(Priority.HIGH); + } + + void bumpPriority(Priority newPriority) { + if (taskPriority.ordinal() >= newPriority.ordinal()) { + return; + } + + this.taskPriority = newPriority; + PriorityQueuedExecutor.PendingTask loadTask = this.loadTask; + PriorityQueuedExecutor.PendingTask genTask = this.genTask; + if (loadTask != null) { + loadTask.bumpPriority(newPriority); + } + if (genTask != null) { + genTask.bumpPriority(newPriority); + } + } + + public synchronized boolean isCancelled() { + return requests.get() <= 0; + } + + public synchronized void cancel(PendingChunkRequest request) { + synchronized (pendingChunks) { + if (!request.cancelled.compareAndSet(false, true)) { + return; + } + + if (requests.decrementAndGet() > 0) { + return; + } + + boolean c1 = genTask.cancel(); + boolean c2 = loadTask.cancel(); + loadTask = null; + genTask = null; + pendingChunks.remove(key); + setStatus(PendingStatus.CANCELLED); + } + } + } + +} diff --git a/src/main/java/net/minecraft/server/PlayerChunk.java b/src/main/java/net/minecraft/server/PlayerChunk.java index 068d203c8..cbff0d946 100644 --- a/src/main/java/net/minecraft/server/PlayerChunk.java +++ b/src/main/java/net/minecraft/server/PlayerChunk.java @@ -29,16 +29,59 @@ public class PlayerChunk { // You know the drill, https://hub.spigotmc.org/stash/projects/SPIGOT/repos/craftbukkit/browse // All may seem good at first, but there's deeper issues if you play for a bit boolean chunkExists; // Paper - private boolean loadInProgress = false; - private Runnable loadedRunnable = new Runnable() { - public void run() { - loadInProgress = false; - PlayerChunk.this.chunk = PlayerChunk.this.playerChunkMap.getWorld().getChunkProviderServer().getChunkAt(location.x, location.z, true, true); + PaperAsyncChunkProvider.CancellableChunkRequest chunkRequest; + // Paper start + private java.util.function.Consumer chunkLoadedConsumer = chunk -> { + chunkRequest = null; + PlayerChunk pChunk = PlayerChunk.this; + pChunk.chunk = chunk; + markChunkUsed(); // Paper - delay chunk unloads + }; + private boolean markedHigh = false; + void checkHighPriority(EntityPlayer player) { + if (done || markedHigh || chunk != null) { + return; + } + final double dist = getDistance(player.locX, player.locZ); + if (dist > 8) { + return; + } + if (dist >= 3 && getDistance(player, 5) > 3.5) { + return; + } + + markedHigh = true; + playerChunkMap.getWorld().getChunkProviderServer().bumpPriority(location); + if (chunkRequest == null) { + requestChunkIfNeeded(PlayerChunkMap.CAN_GEN_CHUNKS.test(player)); + } + } + private void requestChunkIfNeeded(boolean flag) { + if (chunkRequest == null) { + chunkRequest = this.playerChunkMap.getWorld().getChunkProviderServer().requestChunk(this.location.x, this.location.z, flag, markedHigh, chunkLoadedConsumer); + this.chunk = chunkRequest.getChunk(); // Paper) markChunkUsed(); // Paper - delay chunk unloads } - }; + } + private double getDistance(EntityPlayer player, int inFront) { + final float yaw = MathHelper.normalizeYaw(player.yaw); + final double x = player.locX + (inFront * Math.cos(Math.toRadians(yaw))); + final double z = player.locZ + (inFront * Math.sin(Math.toRadians(yaw))); + return getDistance(x, z); + } + + private double getDistance(double blockX, double blockZ) { + final double x = location.x - ((int)Math.floor(blockX) >> 4); + final double z = location.z - ((int)Math.floor(blockZ) >> 4); + return Math.sqrt((x * x) + (z * z)); + } + // Paper end // Paper start - delay chunk unloads private void markChunkUsed() { + if (!chunkHasPlayers && chunkRequest != null) { + chunkRequest.cancel(); + chunkRequest = null; + } if (chunk == null) { return; } @@ -58,8 +101,8 @@ public class PlayerChunk { ChunkProviderServer chunkproviderserver = playerchunkmap.getWorld().getChunkProviderServer(); chunkproviderserver.a(i, j); - this.chunk = chunkproviderserver.getChunkAt(i, j, true, false); - this.chunkExists = this.chunk != null || ChunkIOExecutor.hasQueuedChunkLoad(playerChunkMap.getWorld(), i, j); // Paper + this.chunk = chunkproviderserver.getChunkAt(i, j, false, false); // Paper + this.chunkExists = this.chunk != null || chunkproviderserver.chunkGoingToExists(i, j); // Paper markChunkUsed(); // Paper - delay chunk unloads } @@ -80,7 +123,7 @@ public class PlayerChunk { this.c.add(entityplayer); if (this.done) { this.sendChunk(entityplayer); - } + } else checkHighPriority(entityplayer); // Paper } } @@ -105,8 +148,9 @@ public class PlayerChunk { if (this.chunk != null) { return true; } else { - this.chunk = this.playerChunkMap.getWorld().getChunkProviderServer().getChunkAt(this.location.x, this.location.z, true, flag); - markChunkUsed(); // Paper - delay chunk unloads + // Paper start - async chunks + requestChunkIfNeeded(flag); + // Paper end return this.chunk != null; } } diff --git a/src/main/java/net/minecraft/server/PlayerChunkMap.java b/src/main/java/net/minecraft/server/PlayerChunkMap.java index 39e5b2484..a92557fd2 100644 --- a/src/main/java/net/minecraft/server/PlayerChunkMap.java +++ b/src/main/java/net/minecraft/server/PlayerChunkMap.java @@ -27,10 +27,10 @@ public class PlayerChunkMap { }; private static final Predicate b = (entityplayer) -> { return entityplayer != null && (!entityplayer.isSpectator() || entityplayer.getWorldServer().getGameRules().getBoolean("spectatorsGenerateChunks")); - }; + }; static final Predicate CAN_GEN_CHUNKS = b; // Paper - OBFHELPER private final WorldServer world; private final List managedPlayers = Lists.newArrayList(); - private final Long2ObjectMap e = new Long2ObjectOpenHashMap(4096); + private final Long2ObjectMap e = new Long2ObjectOpenHashMap(4096); Long2ObjectMap getChunks() { return e; } // Paper - OBFHELPER private final Set f = Sets.newHashSet(); private final List g = Lists.newLinkedList(); private final List h = Lists.newLinkedList(); @@ -349,7 +349,13 @@ public class PlayerChunkMap { if (playerchunk != null) { playerchunk.b(entityplayer); } + } else { // Paper start + PlayerChunk playerchunk = this.getChunk(l1 - j1, i2 - k1); + if (playerchunk != null) { + playerchunk.checkHighPriority(entityplayer); // Paper + } } + // Paper end } } @@ -360,7 +366,11 @@ public class PlayerChunkMap { // CraftBukkit start - send nearest chunks first Collections.sort(chunksToLoad, new ChunkCoordComparator(entityplayer)); for (ChunkCoordIntPair pair : chunksToLoad) { - this.c(pair.x, pair.z).a(entityplayer); + // Paper start + PlayerChunk c = this.c(pair.x, pair.z); + c.checkHighPriority(entityplayer); + c.a(entityplayer); + // Paper end } // CraftBukkit end } @@ -432,6 +442,15 @@ public class PlayerChunkMap { } } } + + void shutdown() { + getChunks().values().forEach(pchunk -> { + PaperAsyncChunkProvider.CancellableChunkRequest chunkRequest = pchunk.chunkRequest; + if (chunkRequest != null) { + chunkRequest.cancel(); + } + }); + } // Paper end private void e() { diff --git a/src/main/java/net/minecraft/server/RegionLimitedWorldAccess.java b/src/main/java/net/minecraft/server/RegionLimitedWorldAccess.java index 38f3afb48..ddd7b91a9 100644 --- a/src/main/java/net/minecraft/server/RegionLimitedWorldAccess.java +++ b/src/main/java/net/minecraft/server/RegionLimitedWorldAccess.java @@ -35,7 +35,7 @@ public class RegionLimitedWorldAccess implements GeneratorAccess { this.d = l; this.e = i; this.f = j; - this.g = world; + this.g = world.regionLimited(this); // Paper this.h = world.getSeed(); this.m = world.getChunkProvider().getChunkGenerator().getSettings(); this.i = world.getSeaLevel(); diff --git a/src/main/java/net/minecraft/server/SchedulerBatch.java b/src/main/java/net/minecraft/server/SchedulerBatch.java index d868149d1..0d45d933e 100644 --- a/src/main/java/net/minecraft/server/SchedulerBatch.java +++ b/src/main/java/net/minecraft/server/SchedulerBatch.java @@ -9,6 +9,7 @@ public class SchedulerBatch, R> { private final Scheduler b; private boolean c; private int d = 1000; + private final java.util.concurrent.locks.ReentrantLock lock = new java.util.concurrent.locks.ReentrantLock(true); // Paper public SchedulerBatch(Scheduler scheduler) { this.b = scheduler; @@ -18,8 +19,10 @@ public class SchedulerBatch, R> { this.b.b(); } + public void startBatch() { b(); } // Paper - OBFHELPER public void b() { - if (this.c) { + lock.lock(); // Paper + if (false && this.c) { // Paper throw new RuntimeException("Batch already started."); } else { this.d = 1000; @@ -27,6 +30,7 @@ public class SchedulerBatch, R> { } } + public CompletableFuture add(K k0) { return a(k0); } // Paper - OBFHELPER public CompletableFuture a(K object) { if (!this.c) { throw new RuntimeException("Batch not properly started. Please use startBatch to create a new batch."); @@ -42,8 +46,14 @@ public class SchedulerBatch, R> { } } + public CompletableFuture executeBatch() { return c(); } // Paper - OBFHELPER public CompletableFuture c() { - if (!this.c) { + // Paper start + if (!lock.isHeldByCurrentThread()) { + throw new IllegalStateException("Current thread does not hold the write lock"); + } + try {// Paper end + if (false && !this.c) { // Paper throw new RuntimeException("Batch not properly started. Please use startBatch to create a new batch."); } else { if (this.d != 1000) { @@ -53,5 +63,6 @@ public class SchedulerBatch, R> { this.c = false; return this.b.c(); } + } finally { lock.unlock(); } // Paper } } diff --git a/src/main/java/net/minecraft/server/StructurePiece.java b/src/main/java/net/minecraft/server/StructurePiece.java index d444eb30f..d8c96b5c3 100644 --- a/src/main/java/net/minecraft/server/StructurePiece.java +++ b/src/main/java/net/minecraft/server/StructurePiece.java @@ -14,7 +14,7 @@ public abstract class StructurePiece { private EnumBlockMirror b; private EnumBlockRotation c; protected int o; - private static final Set d = ImmutableSet.builder().add(Blocks.NETHER_BRICK_FENCE).add(Blocks.TORCH).add(Blocks.WALL_TORCH).add(Blocks.OAK_FENCE).add(Blocks.SPRUCE_FENCE).add(Blocks.DARK_OAK_FENCE).add(Blocks.ACACIA_FENCE).add(Blocks.BIRCH_FENCE).add(Blocks.JUNGLE_FENCE).add(Blocks.LADDER).add(Blocks.IRON_BARS).build(); + private static final Set d = ImmutableSet.builder().add(Blocks.NETHER_BRICK_FENCE).add(Blocks.TORCH).add(Blocks.WALL_TORCH).add(Blocks.OAK_FENCE).add(Blocks.SPRUCE_FENCE).add(Blocks.DARK_OAK_FENCE).add(Blocks.ACACIA_FENCE).add(Blocks.BIRCH_FENCE).add(Blocks.JUNGLE_FENCE).add(Blocks.LADDER).add(Blocks.IRON_BARS).build(); // Paper - decompile error public StructurePiece() { } @@ -63,11 +63,11 @@ public abstract class StructurePiece { } public static StructurePiece a(List list, StructureBoundingBox structureboundingbox) { - for(StructurePiece structurepiece : list) { + synchronized (list) { for(StructurePiece structurepiece : list) { // Paper - synchronize main structure list if (structurepiece.d() != null && structurepiece.d().a(structureboundingbox)) { return structurepiece; } - } + }} // Paper return null; } diff --git a/src/main/java/net/minecraft/server/StructureStart.java b/src/main/java/net/minecraft/server/StructureStart.java index 1926c902a..1117e4ae2 100644 --- a/src/main/java/net/minecraft/server/StructureStart.java +++ b/src/main/java/net/minecraft/server/StructureStart.java @@ -6,7 +6,7 @@ import java.util.List; import java.util.Random; public abstract class StructureStart { - protected final List a = Lists.newArrayList(); + protected final List a = java.util.Collections.synchronizedList(Lists.newArrayList()); // Paper protected StructureBoundingBox b; protected int c; protected int d; @@ -49,9 +49,9 @@ public abstract class StructureStart { protected void a(IBlockAccess var1) { this.b = StructureBoundingBox.a(); - for(StructurePiece structurepiece : this.a) { + synchronized (this.a) {for(StructurePiece structurepiece : this.a) { // Paper - synchronize this.b.b(structurepiece.d()); - } + }} // Paper } @@ -114,9 +114,9 @@ public abstract class StructureStart { int l = k - this.b.e; this.b.a(0, l, 0); - for(StructurePiece structurepiece : this.a) { + synchronized (this.a) {for(StructurePiece structurepiece : this.a) { // Paper - synchronize structurepiece.a(0, l, 0); - } + }} // Paper } @@ -132,9 +132,9 @@ public abstract class StructureStart { int i1 = l - this.b.b; this.b.a(0, i1, 0); - for(StructurePiece structurepiece : this.a) { + synchronized (this.a) {for(StructurePiece structurepiece : this.a) { // Paper - synchronize structurepiece.a(0, i1, 0); - } + }} // Paper } diff --git a/src/main/java/net/minecraft/server/World.java b/src/main/java/net/minecraft/server/World.java index eabf50e24..f8afa6f72 100644 --- a/src/main/java/net/minecraft/server/World.java +++ b/src/main/java/net/minecraft/server/World.java @@ -46,7 +46,7 @@ import org.bukkit.event.entity.CreatureSpawnEvent.SpawnReason; import org.bukkit.generator.ChunkGenerator; // CraftBukkit end -public abstract class World implements IEntityAccess, GeneratorAccess, IIBlockAccess, AutoCloseable { +public abstract class World implements IEntityAccess, GeneratorAccess, IIBlockAccess, AutoCloseable, Cloneable { // Paper protected static final Logger e = LogManager.getLogger(); private static final EnumDirection[] a = EnumDirection.values(); @@ -109,6 +109,24 @@ public abstract class World implements IEntityAccess, GeneratorAccess, IIBlockAc protected PersistentVillage villages; public final MethodProfiler methodProfiler; public final boolean isClientSide; + // Paper start - yes this is hacky as shit + RegionLimitedWorldAccess regionLimited; + World originalWorld; + public World regionLimited(RegionLimitedWorldAccess limitedWorldAccess) { + try { + World clone = (World) super.clone(); + clone.regionLimited = limitedWorldAccess; + clone.originalWorld = this; + return clone; + } catch (CloneNotSupportedException e1) { + } + return null; + } + ChunkCoordIntPair[] strongholdCoords; + final java.util.concurrent.atomic.AtomicBoolean + strongholdInit = new java.util.concurrent.atomic.AtomicBoolean + (false); + // Paper end public boolean allowMonsters; public boolean allowAnimals; private boolean J; @@ -742,17 +760,42 @@ public abstract class World implements IEntityAccess, GeneratorAccess, IIBlockAc } - public IBlockData getType(BlockPosition blockposition) { - // CraftBukkit start - tree generation + // Paper - async variant + public java.util.concurrent.CompletableFuture getTypeAsync(BlockPosition blockposition) { + int x = blockposition.getX(); + int z = blockposition.getZ(); if (captureTreeGeneration) { Iterator it = capturedBlockStates.iterator(); while (it.hasNext()) { CraftBlockState previous = it.next(); - if (previous.getX() == blockposition.getX() && previous.getY() == blockposition.getY() && previous.getZ() == blockposition.getZ()) { - return previous.getHandle(); + if (previous.getX() == x && previous.getY() == blockposition.getY() && previous.getZ() == z) { + return java.util.concurrent.CompletableFuture.completedFuture(previous.getHandle()); } } } + if (blockposition.isInvalidYLocation()) { + return java.util.concurrent.CompletableFuture.completedFuture(Blocks.VOID_AIR.getBlockData()); + } else { + java.util.concurrent.CompletableFuture future = new java.util.concurrent.CompletableFuture<>(); + ((ChunkProviderServer) chunkProvider).getChunkAt(x << 4, z << 4, true, true, (chunk) -> { + future.complete(chunk.getType(blockposition)); + }); + return future; + } + } + // Paper end + + public IBlockData getType(BlockPosition blockposition) { + // CraftBukkit start - tree generation + if (captureTreeGeneration) { // If any of this logic updates, update async variant above + Iterator it = capturedBlockStates.iterator(); + while (it.hasNext()) { // If any of this logic updates, update async variant above + CraftBlockState previous = it.next(); + if (previous.getX() == blockposition.getX() && previous.getY() == blockposition.getY() && previous.getZ() == blockposition.getZ()) { + return previous.getHandle(); // If any of this logic updates, update async variant above + } + } // If any of this logic updates, update async variant above + } // CraftBukkit end if (blockposition.isInvalidYLocation()) { // Paper return Blocks.VOID_AIR.getBlockData(); @@ -1021,6 +1064,11 @@ public abstract class World implements IEntityAccess, GeneratorAccess, IIBlockAc } public boolean addEntity(Entity entity, SpawnReason spawnReason) { // Changed signature, added SpawnReason + // Paper start + if (regionLimited != null) { + return regionLimited.addEntity(entity, spawnReason); + } + // Paper end org.spigotmc.AsyncCatcher.catchOp( "entity add"); // Spigot if (entity == null) return false; if (entity.valid) { MinecraftServer.LOGGER.error("Attempted Double World add on " + entity, new Throwable()); return true; } // Paper diff --git a/src/main/java/net/minecraft/server/WorldGenStronghold.java b/src/main/java/net/minecraft/server/WorldGenStronghold.java index fa99fe014..4f49786aa 100644 --- a/src/main/java/net/minecraft/server/WorldGenStronghold.java +++ b/src/main/java/net/minecraft/server/WorldGenStronghold.java @@ -9,24 +9,29 @@ import java.util.Random; import javax.annotation.Nullable; public class WorldGenStronghold extends StructureGenerator { - private boolean b; - private ChunkCoordIntPair[] c; - private long d; + // Paper start - no shared state + //private boolean b; + //private ChunkCoordIntPair[] c; + //private long d; + // Paper end public WorldGenStronghold() { } protected boolean a(ChunkGenerator chunkgenerator, Random var2, int i, int j) { - if (this.d != chunkgenerator.getSeed()) { + // Paper start + /*if (this.d != chunkgenerator.getSeed()) { this.c(); - } + }*/ - if (!this.b) { + synchronized (chunkgenerator.getWorld().strongholdInit) { + if (chunkgenerator.getWorld().strongholdInit.compareAndSet(false, true)) { // Paper this.a(chunkgenerator); - this.b = true; - } + //this.b = true; + }} // Paper + // Paper end - for(ChunkCoordIntPair chunkcoordintpair : this.c) { + for(ChunkCoordIntPair chunkcoordintpair : chunkgenerator.getWorld().strongholdCoords) { // Paper if (i == chunkcoordintpair.x && j == chunkcoordintpair.z) { return true; } @@ -36,8 +41,8 @@ public class WorldGenStronghold extends StructureGenerator chunkgenerator) { - this.d = chunkgenerator.getSeed(); + //this.d = chunkgenerator.getSeed(); // Paper ArrayList arraylist = Lists.newArrayList(); for(BiomeBase biomebase : IRegistry.BIOME) { @@ -111,7 +119,7 @@ public class WorldGenStronghold extends StructureGenerator= k) { - this.c[j1] = new ChunkCoordIntPair(k1, l1); + strongholdCoords[j1] = new ChunkCoordIntPair(k1, l1); // Paper } d1 += (Math.PI * 2D) / (double)i; @@ -153,7 +161,7 @@ public class WorldGenStronghold extends StructureGenerator> 4)) {{ + int j = coords.x; + int k = coords.z; + // Paper end + long l = System.currentTimeMillis(); if (l < i) { @@ -1036,7 +1040,7 @@ public final class CraftServer implements Server { } BlockPosition chunkcoordinates = internal.getSpawn(); - internal.getChunkProviderServer().getChunkAt(chunkcoordinates.getX() + j >> 4, chunkcoordinates.getZ() + k >> 4, true, true); + internal.getChunkProviderServer().getChunkAt(chunkcoordinates.getX() + j >> 4, chunkcoordinates.getZ() + k >> 4, true, true, c -> {}); // Paper } } } diff --git a/src/main/java/org/bukkit/craftbukkit/CraftWorld.java b/src/main/java/org/bukkit/craftbukkit/CraftWorld.java index 7e6a7b0e9..ef5a7bc38 100644 --- a/src/main/java/org/bukkit/craftbukkit/CraftWorld.java +++ b/src/main/java/org/bukkit/craftbukkit/CraftWorld.java @@ -162,6 +162,16 @@ public class CraftWorld implements World { } } + // Paper start - Async chunk load API + @Override + public java.util.concurrent.CompletableFuture getChunkAtAsync(final int x, final int z, final boolean gen) { + final ChunkProviderServer cps = this.world.getChunkProviderServer(); + java.util.concurrent.CompletableFuture future = new java.util.concurrent.CompletableFuture<>(); + cps.getChunkAt(x, z, true, gen, chunk -> future.complete(chunk != null ? chunk.bukkitChunk : null)); + return future; + } + // Paper end + public Chunk getChunkAt(int x, int z) { return this.world.getChunkProviderServer().getChunkAt(x, z, true, true).bukkitChunk; } @@ -1457,10 +1467,13 @@ public class CraftWorld implements World { int chunkCoordZ = chunkcoordinates.getZ() >> 4; // Cycle through the 25x25 Chunks around it to load/unload the chunks. int radius = world.paperConfig.keepLoadedRange / 16; // Paper - for (int x = -radius; x <= radius; x++) { // Paper - for (int z = -radius; z <= radius; z++) { // Paper + // Paper start + for (ChunkCoordIntPair coords : world.getChunkProviderServer().getSpiralOutChunks(world.getSpawn(), radius)) {{ + int x = coords.x; + int z = coords.z; + // Paper end if (keepLoaded) { - loadChunk(chunkCoordX + x, chunkCoordZ + z); + getChunkAtAsync(chunkCoordX + x, chunkCoordZ + z, chunk -> {}); // Paper - Async Chunks } else { if (isChunkLoaded(chunkCoordX + x, chunkCoordZ + z)) { unloadChunk(chunkCoordX + x, chunkCoordZ + z); diff --git a/src/main/java/org/bukkit/craftbukkit/event/CraftEventFactory.java b/src/main/java/org/bukkit/craftbukkit/event/CraftEventFactory.java index 4a0a456a0..4068b8072 100644 --- a/src/main/java/org/bukkit/craftbukkit/event/CraftEventFactory.java +++ b/src/main/java/org/bukkit/craftbukkit/event/CraftEventFactory.java @@ -81,6 +81,7 @@ public class CraftEventFactory { public static final DamageSource POISON = CraftDamageSource.copyOf(DamageSource.MAGIC); public static org.bukkit.block.Block blockDamage; // For use in EntityDamageByBlockEvent public static Entity entityDamage; // For use in EntityDamageByEntityEvent + public static boolean isWorldGen(GeneratorAccess world) { return world instanceof net.minecraft.server.RegionLimitedWorldAccess; } // Paper // helper methods private static boolean canBuild(CraftWorld world, Player player, int x, int z) { @@ -382,6 +383,7 @@ public class CraftEventFactory { CraftServer craftServer = (CraftServer) entity.getServer(); CreatureSpawnEvent event = new CreatureSpawnEvent(entity, spawnReason); + if (isWorldGen(entityliving.world)) return event; // Paper - do not call during world gen craftServer.getPluginManager().callEvent(event); return event; } @@ -1029,6 +1031,7 @@ public class CraftEventFactory { } BlockIgniteEvent event = new BlockIgniteEvent(bukkitWorld.getBlockAt(block.getX(), block.getY(), block.getZ()), cause, igniter); + if (isWorldGen(world)) return event; // Paper - do not call during world gen world.getServer().getPluginManager().callEvent(event); return event; } @@ -1053,6 +1056,7 @@ public class CraftEventFactory { } BlockIgniteEvent event = new BlockIgniteEvent(bukkitWorld.getBlockAt(pos.getX(), pos.getY(), pos.getZ()), cause, bukkitIgniter); + if (isWorldGen(world)) return event; // Paper - do not call during world gen world.getServer().getPluginManager().callEvent(event); return event; } @@ -1260,7 +1264,8 @@ public class CraftEventFactory { public static BlockPhysicsEvent callBlockPhysicsEvent(GeneratorAccess world, BlockPosition blockposition) { org.bukkit.block.Block block = CraftBlock.at(world, blockposition); BlockPhysicsEvent event = new BlockPhysicsEvent(block, block.getBlockData()); - world.getMinecraftWorld().getMinecraftServer().server.getPluginManager().callEvent(event); + if (isWorldGen(world)) return event; // Paper - do not call during world gen + Bukkit.getPluginManager().callEvent(event); // Paper return event; } @@ -1296,6 +1301,7 @@ public class CraftEventFactory { } EntityPotionEffectEvent event = new EntityPotionEffectEvent((LivingEntity) entity.getBukkitEntity(), bukkitOldEffect, bukkitNewEffect, cause, action, willOverride); + if (isWorldGen(entity.world)) return event; // Paper - do not call during world gen Bukkit.getPluginManager().callEvent(event); return event; @@ -1314,6 +1320,7 @@ public class CraftEventFactory { blockState.setData(block); BlockFormEvent event = (entity == null) ? new BlockFormEvent(blockState.getBlock(), blockState) : new EntityBlockFormEvent(entity.getBukkitEntity(), blockState.getBlock(), blockState); + if (isWorldGen(world)) return true; // Paper - do not call during world gen world.getServer().getPluginManager().callEvent(event); if (!event.isCancelled()) { diff --git a/src/main/java/org/bukkit/craftbukkit/generator/CustomChunkGenerator.java b/src/main/java/org/bukkit/craftbukkit/generator/CustomChunkGenerator.java index 04e29f58c..5fae0c6ad 100644 --- a/src/main/java/org/bukkit/craftbukkit/generator/CustomChunkGenerator.java +++ b/src/main/java/org/bukkit/craftbukkit/generator/CustomChunkGenerator.java @@ -21,6 +21,7 @@ public class CustomChunkGenerator extends InternalChunkGenerator