OpenSource/PaperMC
OpenSource/PaperMC
1
0
Fork 0
mirror of https://github.com/PaperMC/Paper.git synced 2025-01-03 17:52:28 +01:00
Code Issues Projects Releases Packages Wiki Activity

#1110: Make CraftMapColorCache more thread safe

Browse source 
By: DerFrZocker <derrieple@gmail.com>
This commit is contained in:
CraftBukkit/Spigot 2022-09-21 06:54:55 +10:00
parent c38953dda8
commit 61e6302177
1 changed files with 4 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
paper-server/src/main/java/org/bukkit/craftbukkit/map/CraftMapColorCache.java
View file

@ -11,6 +11,7 @@ import java.io.OutputStream;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.logging.Logger;
import java.util.zip.DeflaterOutputStream;
import java.util.zip.InflaterInputStream;
@ -24,7 +25,7 @@ public class CraftMapColorCache implements MapPalette.MapColorCache {
private byte[] cache;
private final Logger logger;
private boolean cached = false;
private boolean running = false;
private final AtomicBoolean running = new AtomicBoolean(false);
public CraftMapColorCache(Logger logger) {
this.logger = logger;
@ -57,7 +58,7 @@ public class CraftMapColorCache implements MapPalette.MapColorCache {
}
public CompletableFuture<Void> initCache() {
Preconditions.checkState(!cached && !running, "Cache is already build or is currently being build");
Preconditions.checkState(!cached && !running.getAndSet(true), "Cache is already build or is currently being build");
cache = new byte[256 * 256 * 256]; // Red, Green and Blue have each a range from 0 to 255 each mean we need space for 256 * 256 * 256 values
if (CACHE_FILE.exists()) {
@ -108,21 +109,18 @@ public class CraftMapColorCache implements MapPalette.MapColorCache {
}
private CompletableFuture<Void> buildAndSaveCache() {
running = true;
return CompletableFuture.runAsync(() -> {
buildCache();
if (!CACHE_FILE.exists()) {
try {
if (!CACHE_FILE.createNewFile()) {
running = false;
cached = true;
return;
}
} catch (IOException e) {
logger.warning("Error while building map color cache");
e.printStackTrace();
running = false;
cached = true;
return;
}
@ -133,12 +131,10 @@ public class CraftMapColorCache implements MapPalette.MapColorCache {
} catch (IOException e) {
logger.warning("Error while building map color cache");
e.printStackTrace();
running = false;
cached = true;
return;
}
running = false;
cached = true;
logger.info("Map color cache build successfully");
}, SystemUtils.backgroundExecutor());
@ -150,7 +146,7 @@ public class CraftMapColorCache implements MapPalette.MapColorCache {
@Override
public boolean isCached() {
return cached || (!running && initCache().isDone());
return cached || (!running.get() && initCache().isDone());
}
@Override