Fix DataModel crash after deserialization (#249)

Fixes #248
wayfair · Feb 15, 2021 · 814e250 · 814e250
1 parent dc39575
commit 814e250
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/BrickKit/bricks/src/main/java/com/wayfair/brickkit/brick/DataModel.kt b/BrickKit/bricks/src/main/java/com/wayfair/brickkit/brick/DataModel.kt
@@ -11,15 +11,21 @@ import java.io.Serializable
  */
 abstract class DataModel : Serializable {
     @Transient
-    private val updateListeners = mutableSetOf<DataModelUpdateListener>()
+    private var updateListeners: MutableSet<DataModelUpdateListener>? = null
+        get() {
+            if (field == null) {
+                field = mutableSetOf()
+            }
+            return field
+        }
 
     /**
      * Add a [DataModelUpdateListener] to the list of listeners.
      *
      * @param updateListener the listener to add
      */
     open fun addUpdateListener(updateListener: DataModelUpdateListener) {
-        updateListeners.add(updateListener)
+        updateListeners!!.add(updateListener)
     }
 
     /**
@@ -28,7 +34,7 @@ abstract class DataModel : Serializable {
      * @param updateListener the listener to remove
      */
     open fun removeUpdateListener(updateListener: DataModelUpdateListener) {
-        updateListeners.remove(updateListener)
+        updateListeners!!.remove(updateListener)
     }
 
     /**
@@ -37,7 +43,7 @@ abstract class DataModel : Serializable {
     open fun notifyChange() {
         val handler = Handler(Looper.getMainLooper())
 
-        updateListeners.forEach { updateListener ->
+        updateListeners!!.forEach { updateListener ->
             handler.post { updateListener.notifyChange() }
         }
     }