trying to optimise it

1 files changed, 15 insertions, 7 deletions

diff --git a/src/main/java/me/shedaniel/rei/tests/plugin/REITestPlugin.java b/src/main/java/me/shedaniel/rei/tests/plugin/REITestPlugin.java
index 9064fd499..9eebeaee0 100644
--- a/src/main/java/me/shedaniel/rei/tests/plugin/REITestPlugin.java
+++ b/src/main/java/me/shedaniel/rei/tests/plugin/REITestPlugin.java
@@ -15,10 +15,14 @@ import net.minecraft.util.registry.Registry;
 import org.apache.logging.log4j.LogManager;
 import org.jetbrains.annotations.TestOnly;
 
+import java.util.Random;
+
 @TestOnly
 @Deprecated
 public class REITestPlugin implements REIPluginV0 {
 
+    private Random random = new Random();
+
     @Override
     public void preRegister() {
         LogManager.getLogger().error("REI Test Plugin is enabled! If you see this unintentionally, please report this!");
@@ -32,17 +36,21 @@ public class REITestPlugin implements REIPluginV0 {
     @Override
     public void registerEntries(EntryRegistry entryRegistry) {
         for (Item item : Registry.ITEM) {
-            entryRegistry.registerEntryAfter(null, EntryStack.create(item), false);
-            entryRegistry.registerEntryAfter(null, EntryStack.create(item), false);
-            entryRegistry.registerEntryAfter(null, EntryStack.create(item), false);
+            for (int i = 0; i < 15; i++)
+                entryRegistry.registerEntry(transformStack(EntryStack.create(item)));
             try {
-                for (ItemStack stack : entryRegistry.getAllStacksFromItem(item)) {
-                    entryRegistry.registerEntryAfter(null, EntryStack.create(stack), false);
-                    entryRegistry.registerEntryAfter(null, EntryStack.create(stack), false);
-                    entryRegistry.registerEntryAfter(null, EntryStack.create(stack), false);
+                for (ItemStack stack : entryRegistry.appendStacksForItem(item)) {
+                    for (int i = 0; i < 15; i++)
+                        entryRegistry.registerEntry(transformStack(EntryStack.create(stack)));
                 }
             } catch (Exception e) {
             }
         }
     }
+
+    public EntryStack transformStack(EntryStack stack) {
+        stack.setAmount(random.nextInt(Integer.MAX_VALUE));
+        return stack;
+    }
+
 }