Merge elements when getting new composition data

NordicSemiconductor · Oct 17, 2022 · 90acbda · 90acbda
1 parent 71dd487
commit 90acbda
Showing 1 changed file with 40 additions and 11 deletions.
diff --git a/mesh/src/main/java/no/nordicsemi/android/mesh/transport/ProvisionedMeshNode.java b/mesh/src/main/java/no/nordicsemi/android/mesh/transport/ProvisionedMeshNode.java
@@ -22,23 +22,27 @@
 
 package no.nordicsemi.android.mesh.transport;
 
+import static androidx.room.ForeignKey.CASCADE;
+
 import android.annotation.SuppressLint;
 import android.os.Parcel;
 
-import java.util.ArrayList;
-import java.util.Collections;
-import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
-import java.util.Set;
-
 import androidx.annotation.NonNull;
 import androidx.annotation.RestrictTo;
 import androidx.annotation.VisibleForTesting;
 import androidx.room.Entity;
 import androidx.room.ForeignKey;
 import androidx.room.Ignore;
 import androidx.room.Index;
+
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.LinkedHashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.Set;
+
 import no.nordicsemi.android.mesh.ApplicationKey;
 import no.nordicsemi.android.mesh.Features;
 import no.nordicsemi.android.mesh.MeshNetwork;
@@ -54,8 +58,6 @@
 import no.nordicsemi.android.mesh.utils.SecureUtils;
 import no.nordicsemi.android.mesh.utils.SparseIntArrayParcelable;
 
-import static androidx.room.ForeignKey.CASCADE;
-
 @SuppressWarnings({"WeakerAccess"})
 @Entity(tableName = "nodes",
         foreignKeys = @ForeignKey(entity = MeshNetwork.class,
@@ -100,7 +102,7 @@ public ProvisionedMeshNode(final UnprovisionedMeshNode node) {
         ttl = node.getTtl();
         mTimeStampInMillis = node.getTimeStamp();
         // Here we add some dummy elements with empty models to occupy the addresses in use.
-        for(int i = 0; i < node.getProvisioningCapabilities().getNumberOfElements(); i++){
+        for (int i = 0; i < node.getProvisioningCapabilities().getNumberOfElements(); i++) {
             mElements.put(unicastAddress + i, new Element(unicastAddress + i, 0, new HashMap<>()));
         }
         security = node.isSecure() ? HIGH : LOW;
@@ -506,7 +508,34 @@ void setCompositionData(
                 lowPowerFeatureSupported ? Features.DISABLED : Features.UNSUPPORTED,
                 proxyFeatureSupported ? Features.DISABLED : Features.UNSUPPORTED,
                 relayFeatureSupported ? Features.DISABLED : Features.UNSUPPORTED);
-        mElements.putAll(configCompositionDataStatus.getElements());
+
+        LinkedHashMap<Integer, Element> modifiableNewElements = new LinkedHashMap<>(configCompositionDataStatus.getElements());
+
+        // For each old element
+        for (Map.Entry<Integer, Element> newElementEntry : configCompositionDataStatus.getElements().entrySet()) {
+            final Element newElement = newElementEntry.getValue();
+            // Fetch same element in new composition data
+            final Element oldElement = mElements.get(newElementEntry.getKey());
+
+            LinkedHashMap<Integer, MeshModel> updatedElementModels = new LinkedHashMap<>(newElement.getMeshModels());
+
+            // Old element existed previously, may have same mesh models
+            if (oldElement != null) {
+                // For each model in that element
+                for (Map.Entry<Integer, MeshModel> modelEntry : newElement.getMeshModels().entrySet()) {
+
+                    // Override new element models with old to ensure we keep mesh model info
+                    if (oldElement.getMeshModels().containsKey(modelEntry.getKey())) {
+                        updatedElementModels.put(modelEntry.getKey(), oldElement.meshModels.get(modelEntry.getKey()));
+                    }
+                }
+
+                Element newUpdatedElement = new Element(newElement.elementAddress, newElement.locationDescriptor, updatedElementModels, newElement.name);
+
+                modifiableNewElements.put(newElementEntry.getKey(), newUpdatedElement);
+            }
+        }
+        mElements.putAll(modifiableNewElements);
     }
 
     /**