File tree 1 file changed +4
-2
lines changed
src/main/java/meteordevelopment/meteorclient/systems/modules
1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -66,6 +66,7 @@ public class Modules extends System<Modules> {
66
66
private static final List <Category > CATEGORIES = new ArrayList <>();
67
67
68
68
private final List <Module > modules = new ArrayList <>();
69
+ private final Map <Class <? extends Module >, Module > moduleInstances = new Reference2ReferenceOpenHashMap <>();
69
70
private final Map <Category , List <Module >> groups = new Reference2ReferenceOpenHashMap <>();
70
71
71
72
private final List <Module > active = new ArrayList <>();
@@ -127,9 +128,9 @@ public static Category getCategoryByHash(int hash) {
127
128
return null ;
128
129
}
129
130
130
- @ SuppressWarnings ({ "unchecked" , "OptionalGetWithoutIsPresent" } )
131
+ @ SuppressWarnings ("unchecked" )
131
132
public <T extends Module > T get (Class <T > klass ) {
132
- return (T ) modules . stream (). filter ( module -> module . getClass (). equals ( klass )). findFirst (). orElse ( null );
133
+ return (T ) moduleInstances . get ( klass );
133
134
}
134
135
135
136
public Module get (String name ) {
@@ -394,6 +395,7 @@ public void add(Module module) {
394
395
395
396
// Add the module
396
397
modules .add (module );
398
+ moduleInstances .put (module .getClass (), module );
397
399
getGroup (module .category ).add (module );
398
400
399
401
// Register color settings for the module
You can’t perform that action at this time.
0 commit comments