build: update FlowSched

RelativityMC · Dec 11, 2023 · a9b1df2 · a9b1df2
1 parent a641cb3
commit a9b1df2
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/FlowSched b/FlowSched
diff --git a/c2me-base/src/main/java/com/ishland/c2me/base/common/scheduler/ScheduledTask.java b/c2me-base/src/main/java/com/ishland/c2me/base/common/scheduler/ScheduledTask.java
@@ -18,8 +18,9 @@ public ScheduledTask(long pos, Supplier<CompletableFuture<T>> action, LockToken[
  }
 
  @Override
- public void run() {
+ public void run(Runnable releaseLocks) {
  action.get().whenComplete((t, throwable) -> {
+ releaseLocks.run();
  if (throwable != null) {
  future.completeExceptionally(throwable);
  } else {

diff --git a/c2me-base/src/main/java/com/ishland/c2me/base/common/scheduler/WrappingTask.java b/c2me-base/src/main/java/com/ishland/c2me/base/common/scheduler/WrappingTask.java
@@ -16,10 +16,11 @@ public WrappingTask(long pos, Runnable wrapped) {
  }
 
  @Override
- public void run() {
+ public void run(Runnable releaseLocks) {
  try {
  wrapped.run();
  } finally {
+ releaseLocks.run();
  for (Runnable runnable : this.postExec) {
  try {
  runnable.run();