Update selected ViewHolder after notifying

rubensousa · Oct 16, 2024 · 7d764d7 · 7d764d7
1 parent cb7226c
commit 7d764d7
Showing 1 changed file with 7 additions and 9 deletions.
diff --git a/...recyclerview/src/main/java/com/rubensousa/dpadrecyclerview/layoutmanager/PivotSelector.kt b/...recyclerview/src/main/java/com/rubensousa/dpadrecyclerview/layoutmanager/PivotSelector.kt
@@ -213,6 +213,11 @@ internal class PivotSelector(
     }
 
     fun onLayoutCompleted() {
+        if (isSelectionUpdatePending) {
+            isSelectionUpdatePending = false
+            dispatchViewHolderSelected()
+            dispatchViewHolderSelectedAndAligned()
+        }
         /**
          * Always update the selected ViewHolder on every layout update
          */
@@ -223,11 +228,6 @@ internal class PivotSelector(
                 }
             }
         }
-        if (isSelectionUpdatePending) {
-            isSelectionUpdatePending = false
-            dispatchViewHolderSelected()
-            dispatchViewHolderSelectedAndAligned()
-        }
     }
 
     fun getCurrentSubPositions(): Int {
@@ -350,15 +350,13 @@ internal class PivotSelector(
                     listener.onViewHolderDeselected(recyclerView, viewHolder)
                 }
             }
-
             if (viewHolder is DpadViewHolder) {
-                selectedViewHolder = viewHolder
                 viewHolder.onViewHolderSelected()
-            } else {
-                selectedViewHolder = null
             }
         }
 
+        selectedViewHolder = viewHolder
+
         if (viewHolder != null) {
             selectionListeners.forEach { listener ->
                 listener.onViewHolderSelected(