Upstream Paper

src/main/java/net/minecraft/server/SchedulerBatch.java

@@ -0,0 +1,70 @@
+package net.minecraft.server;
+
+import java.util.concurrent.CompletableFuture;
+import org.apache.logging.log4j.LogManager;
+import org.apache.logging.log4j.Logger;
+
+public class SchedulerBatch<K, T extends SchedulerTask<K, T>, R> {
+
+    private static final Logger a = LogManager.getLogger();
+    private final Scheduler<K, T, R> 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<K, T, R> scheduler) {
+        this.b = scheduler;
+    }
+
+    public void a() throws InterruptedException {
+        this.b.b();
+    }
+
+    public void startBatch() { b(); } // Paper - OBFHELPER
+    public void b() {
+        lock.lock(); // Paper
+        if (false && this.c) { // Paper
+            throw new RuntimeException("Batch already started.");
+        } else {
+            this.d = 1000;
+            this.c = true;
+        }
+    }
+
+    public CompletableFuture<R> add(K k0) { return a(k0); } // Paper - OBFHELPER
+    public CompletableFuture<R> a(K k0) {
+        if (!this.c) {
+            throw new RuntimeException("Batch not properly started. Please use startBatch to create a new batch.");
+        } else {
+            CompletableFuture<R> completablefuture = this.b.a(k0);
+
+            --this.d;
+            if (this.d == 0) {
+                completablefuture = this.b.a();
+                this.d = 1000;
+            }
+
+            return completablefuture;
+        }
+    }
+
+    public CompletableFuture<R> executeBatch() { return c(); } // Paper - OBFHELPER
+    public CompletableFuture<R> 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) {
+                this.b.a();
+            }
+
+            this.c = false;
+            return this.b.c();
+        }
+        } finally { lock.unlock(); } // Paper
+    }
+}