Optimize scene/scene collection lookups.

Signed-off-by: Jacob Laursen <[email protected]>
jlaur · Jan 2, 2022 · 27dca2b · 27dca2b
1 parent ab28adb
commit 27dca2b
Showing 1 changed file with 20 additions and 20 deletions.
diff --git a/...main/java/org/openhab/binding/hdpowerview/internal/builders/AutomationChannelBuilder.java b/...main/java/org/openhab/binding/hdpowerview/internal/builders/AutomationChannelBuilder.java
@@ -17,8 +17,11 @@
 import java.time.format.TextStyle;
 import java.util.ArrayList;
 import java.util.EnumSet;
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
 import java.util.StringJoiner;
+import java.util.stream.Collectors;
 
 import org.eclipse.jdt.annotation.NonNullByDefault;
 import org.eclipse.jdt.annotation.Nullable;
@@ -53,8 +56,8 @@ public class AutomationChannelBuilder {
             HDPowerViewBindingConstants.CHANNELTYPE_AUTOMATION_ENABLED);
 
     private List<Channel> channels;
-    private List<Scene> scenes;
-    private List<SceneCollection> sceneCollections;
+    private Map<Integer, Scene> scenes;
+    private Map<Integer, SceneCollection> sceneCollections;
     private List<ScheduledEvent> scheduledEvents;
 
     public AutomationChannelBuilder(HDPowerViewTranslationProvider translationProvider,
@@ -63,8 +66,8 @@ public AutomationChannelBuilder(HDPowerViewTranslationProvider translationProvid
         this.channelGroupUid = channelGroupUid;
         this.channels = new ArrayList<>(0);
         this.scheduledEvents = new ArrayList<>(0);
-        this.scenes = new ArrayList<>(0);
-        this.sceneCollections = new ArrayList<>(0);
+        this.scenes = new HashMap<>(0);
+        this.sceneCollections = new HashMap<>(0);
     }
 
     /**
@@ -98,7 +101,7 @@ public AutomationChannelBuilder withChannels(List<Channel> channels) {
      * @return channel builder
      */
     public AutomationChannelBuilder withScenes(List<Scene> scenes) {
-        this.scenes = scenes;
+        this.scenes = scenes.stream().collect(Collectors.toMap(scene -> scene.id, scene -> scene));
         return this;
     }
 
@@ -109,7 +112,8 @@ public AutomationChannelBuilder withScenes(List<Scene> scenes) {
      * @return channel builder
      */
     public AutomationChannelBuilder withSceneCollections(List<SceneCollection> sceneCollections) {
-        this.sceneCollections = sceneCollections;
+        this.sceneCollections = sceneCollections.stream()
+                .collect(Collectors.toMap(sceneCollection -> sceneCollection.id, sceneCollection -> sceneCollection));
         return this;
     }
 
@@ -131,7 +135,7 @@ public AutomationChannelBuilder withScheduledEvents(List<ScheduledEvent> schedul
      */
     public List<Channel> build() {
         scheduledEvents.stream().forEach(scheduledEvent -> {
-            Channel channel = createChannel(scheduledEvent, scenes, sceneCollections);
+            Channel channel = createChannel(scheduledEvent);
             if (channel != null) {
                 channels.add(channel);
             }
@@ -140,9 +144,8 @@ public List<Channel> build() {
         return channels;
     }
 
-    private @Nullable Channel createChannel(ScheduledEvent scheduledEvent, List<Scene> scenes,
-            List<SceneCollection> sceneCollections) {
-        String referencedName = getReferencedSceneOrSceneCollectionName(scheduledEvent, scenes, sceneCollections);
+    private @Nullable Channel createChannel(ScheduledEvent scheduledEvent) {
+        String referencedName = getReferencedSceneOrSceneCollectionName(scheduledEvent);
         if (referencedName == null) {
             return null;
         }
@@ -156,22 +159,19 @@ public List<Channel> build() {
         return channel;
     }
 
-    private @Nullable String getReferencedSceneOrSceneCollectionName(ScheduledEvent scheduledEvent, List<Scene> scenes,
-            List<SceneCollection> sceneCollections) {
+    private @Nullable String getReferencedSceneOrSceneCollectionName(ScheduledEvent scheduledEvent) {
         if (scheduledEvent.sceneId > 0) {
-            for (Scene scene : scenes) {
-                if (scene.id == scheduledEvent.sceneId) {
-                    return scene.getName();
-                }
+            Scene scene = scenes.get(scheduledEvent.sceneId);
+            if (scene != null) {
+                return scene.getName();
             }
             logger.error("Scene '{}' was not found for scheduled event '{}'", scheduledEvent.sceneId,
                     scheduledEvent.id);
             return null;
         } else if (scheduledEvent.sceneCollectionId > 0) {
-            for (SceneCollection sceneCollection : sceneCollections) {
-                if (sceneCollection.id == scheduledEvent.sceneCollectionId) {
-                    return sceneCollection.getName();
-                }
+            SceneCollection sceneCollection = sceneCollections.get(scheduledEvent.sceneCollectionId);
+            if (sceneCollection != null) {
+                return sceneCollection.getName();
             }
             logger.error("Scene collection '{}' was not found for scheduled event '{}'",
                     scheduledEvent.sceneCollectionId, scheduledEvent.id);