45 lines
1.6 KiB
Java
45 lines
1.6 KiB
Java
|
package net.minecraft.util;
|
||
|
|
|
||
|
|
import com.mojang.logging.LogUtils;
|
||
|
|
import java.util.concurrent.CancellationException;
|
||
|
|
import java.util.concurrent.CompletableFuture;
|
||
|
|
import java.util.concurrent.CompletionException;
|
||
|
|
import java.util.concurrent.Executor;
|
||
|
|
import java.util.function.Consumer;
|
||
|
|
import org.slf4j.Logger;
|
||
|
|
|
||
|
|
public class FutureChain implements TaskChainer, AutoCloseable {
|
||
|
|
private static final Logger LOGGER = LogUtils.getLogger();
|
||
|
|
private CompletableFuture<?> head = CompletableFuture.completedFuture(null);
|
||
|
|
private final Executor executor;
|
||
|
|
private volatile boolean closed;
|
||
|
|
|
||
|
|
public FutureChain(Executor p_242395_) {
|
||
|
|
this.executor = p_242395_;
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override
|
||
|
|
public <T> void append(CompletableFuture<T> p_310575_, Consumer<T> p_310607_) {
|
||
|
|
this.head = this.head.<T, Object>thenCombine(p_310575_, (p_308966_, p_308967_) -> p_308967_).thenAcceptAsync(p_308969_ -> {
|
||
|
|
if (!this.closed) {
|
||
|
|
p_310607_.accept((T)p_308969_);
|
||
|
|
}
|
||
|
|
}, this.executor).exceptionally(p_242215_ -> {
|
||
|
|
if (p_242215_ instanceof CompletionException completionexception) {
|
||
|
|
p_242215_ = completionexception.getCause();
|
||
|
|
}
|
||
|
|
|
||
|
|
if (p_242215_ instanceof CancellationException cancellationexception) {
|
||
|
|
throw cancellationexception;
|
||
|
|
} else {
|
||
|
|
LOGGER.error("Chain link failed, continuing to next one", p_242215_);
|
||
|
|
return null;
|
||
|
|
}
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override
|
||
|
|
public void close() {
|
||
|
|
this.closed = true;
|
||
|
|
}
|
||
|
|
}
|