spdis/voxy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added compression and decompression contexts

Browse Source
This commit is contained in:
mcrcortex
2024-12-05 12:15:05 +10:00
parent 1b82b3020f
commit 8c36021746
1 changed files with 22 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
src/main/java/me/cortex/voxy/common/storage/compressors/ZSTDCompressor.java
View File

@@ -6,9 +6,29 @@ import me.cortex.voxy.common.storage.config.ConfigBuildCtx;
import me.cortex.voxy.common.util.MemoryBuffer; import me.cortex.voxy.common.util.MemoryBuffer;
import me.cortex.voxy.common.world.SaveLoadSystem; import me.cortex.voxy.common.world.SaveLoadSystem;
import java.lang.ref.Cleaner;
import static org.lwjgl.util.zstd.Zstd.*; import static org.lwjgl.util.zstd.Zstd.*;
public class ZSTDCompressor implements StorageCompressor { public class ZSTDCompressor implements StorageCompressor {
private static final Cleaner CLEANER = Cleaner.create();
private record Ref(long ptr) {}
private static Ref createCleanableCompressionContext() {
long ctx = ZSTD_createCCtx();
var ref = new Ref(ctx);
CLEANER.register(ref, ()->ZSTD_freeCCtx(ctx));
return ref;
}
private static Ref createCleanableDecompressionContext() {
long ctx = ZSTD_createDCtx();
var ref = new Ref(ctx);
CLEANER.register(ref, ()->ZSTD_freeDCtx(ctx));
return ref;
}
private static final ThreadLocal<Ref> COMPRESSION_CTX = ThreadLocal.withInitial(ZSTDCompressor::createCleanableCompressionContext);
private static final ThreadLocal<Ref> DECOMPRESSION_CTX = ThreadLocal.withInitial(ZSTDCompressor::createCleanableDecompressionContext);
private final int level; private final int level;
public ZSTDCompressor(int level) { public ZSTDCompressor(int level) {
@@ -18,14 +38,14 @@ public class ZSTDCompressor implements StorageCompressor {
@Override @Override
public MemoryBuffer compress(MemoryBuffer saveData) { public MemoryBuffer compress(MemoryBuffer saveData) {
MemoryBuffer compressedData = new MemoryBuffer((int)ZSTD_COMPRESSBOUND(saveData.size)); MemoryBuffer compressedData = new MemoryBuffer((int)ZSTD_COMPRESSBOUND(saveData.size));
long compressedSize = nZSTD_compress(compressedData.address, compressedData.size, saveData.address, saveData.size, this.level); long compressedSize = nZSTD_compressCCtx(COMPRESSION_CTX.get().ptr, compressedData.address, compressedData.size, saveData.address, saveData.size, this.level);
return compressedData.subSize(compressedSize); return compressedData.subSize(compressedSize);
} }
@Override @Override
public MemoryBuffer decompress(MemoryBuffer saveData) { public MemoryBuffer decompress(MemoryBuffer saveData) {
var decompressed = new MemoryBuffer(SaveLoadSystem.BIGGEST_SERIALIZED_SECTION_SIZE); var decompressed = new MemoryBuffer(SaveLoadSystem.BIGGEST_SERIALIZED_SECTION_SIZE);
long size = nZSTD_decompress(decompressed.address, decompressed.size, saveData.address, saveData.size); long size = nZSTD_decompressDCtx(DECOMPRESSION_CTX.get().ptr, decompressed.address, decompressed.size, saveData.address, saveData.size);
return decompressed.subSize(size); return decompressed.subSize(size);
} }