From 1e1173c4227a3f0e6c7688ee5c5d533c9921dc50 Mon Sep 17 00:00:00 2001
From: shedaniel <daniel@shedaniel.me>
Date: Wed, 9 Aug 2023 03:23:39 +0800
Subject: Fix concurrency issues with recipe lookup

---
 .../rei/api/common/registry/ParentReloadable.java          | 14 ++++++++++++++
 .../me/shedaniel/rei/api/common/registry/Reloadable.java   |  6 ++++++
 2 files changed, 20 insertions(+)

(limited to 'api/src/main/java')

diff --git a/api/src/main/java/me/shedaniel/rei/api/common/registry/ParentReloadable.java b/api/src/main/java/me/shedaniel/rei/api/common/registry/ParentReloadable.java
index eef0ae964..f128673ef 100644
--- a/api/src/main/java/me/shedaniel/rei/api/common/registry/ParentReloadable.java
+++ b/api/src/main/java/me/shedaniel/rei/api/common/registry/ParentReloadable.java
@@ -60,6 +60,20 @@ public interface ParentReloadable<P extends REIPlugin<?>> extends Reloadable<P>
         }
     }
     
+    @Override
+    default void preStage(ReloadStage stage) {
+        for (Reloadable<P> reloadable : getReloadables()) {
+            reloadable.preStage(stage);
+        }
+    }
+    
+    @Override
+    default void postStage(ReloadStage stage) {
+        for (Reloadable<P> reloadable : getReloadables()) {
+            reloadable.postStage(stage);
+        }
+    }
+    
     @Override
     default void beforeReloadable(ReloadStage stage, Reloadable<P> other) {
         for (Reloadable<P> reloadable : getReloadables()) {
diff --git a/api/src/main/java/me/shedaniel/rei/api/common/registry/Reloadable.java b/api/src/main/java/me/shedaniel/rei/api/common/registry/Reloadable.java
index 100a054d6..62d0df7b7 100644
--- a/api/src/main/java/me/shedaniel/rei/api/common/registry/Reloadable.java
+++ b/api/src/main/java/me/shedaniel/rei/api/common/registry/Reloadable.java
@@ -48,6 +48,12 @@ public interface Reloadable<P extends REIPlugin<?>> {
         }
     }
     
+    @ApiStatus.Experimental
+    default void preStage(ReloadStage stage) {}
+    
+    @ApiStatus.Experimental
+    default void postStage(ReloadStage stage) {}
+    
     @ApiStatus.Experimental
     default void beforeReloadable(ReloadStage stage, Reloadable<P> other) {}
     
-- 
cgit