MinecraftMods/BCLib
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

[Change] Better Thread-Syncing abilities for RegistrySupplier

Browse source
This commit is contained in:
Frank 2023-05-23 21:46:45 +02:00
parent 508dfc105a
commit cc9c4e3c93
2 changed files with 17 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

18
src/main/java/org/betterx/bclib/api/v3/datagen/RegistrySupplier.java
View file

@ -18,7 +18,7 @@ import java.util.concurrent.Semaphore;
import org.jetbrains.annotations.Nullable;
public abstract class RegistrySupplier {
private static final int MAX_PERMITS = 1000;
private static final int MAX_PERMITS = 2000;
private final Semaphore BOOTSTRAP_LOCK = new Semaphore(MAX_PERMITS);
final List<RegistrySupplier.RegistryInfo<?>> allRegistries;
@ -55,7 +55,23 @@ public abstract class RegistrySupplier {
void releaseLock() {
BOOTSTRAP_LOCK.release(MAX_PERMITS);
}
public <T> void addWithLock(
RegistrySetBuilder registryBuilder,
ResourceKey<? extends Registry<T>> resourceKey,
RegistrySetBuilder.RegistryBootstrap<T> registryBootstrap
) {
try {
BOOTSTRAP_LOCK.acquire();
} catch (InterruptedException e) {
throw new RuntimeException(e);
}
registryBuilder.add(resourceKey, (ctx) -> {
registryBootstrap.run(ctx);
BOOTSTRAP_LOCK.release();
});
}
public class InfoList extends LinkedList<RegistrySupplier.RegistryInfo<?>> {

1
src/main/java/org/betterx/datagen/bclib/BCLibDatagen.java
View file

@ -36,7 +36,6 @@ public class BCLibDatagen implements DataGeneratorEntrypoint {
pack.addProvider(WorldPresetDataProvider::new);
pack.addProvider(BCLibRegistriesDataProvider::new);
pack.addProvider(BCLAdvancementDataProvider::new);
}