commit fa37c6742e63f1b19b8efa51309c17c07c7446cb
author Alina Zaidi <alinazaidi@google.com> Mon Oct 25 16:10:45 2021 +0100
committer Alina Zaidi <alinazaidi@google.com> Mon Oct 25 16:10:45 2021 +0100
tree d2a3e1bb3c9d4658effacbf005ed4ce356393b5e
parent f0f2b75adaa1c530991b573859bdc7b761261ac3

Make sure distinct widgets items are used when mapping widget component
names to widget items.

This is to prevent a crash on duplicate key when creating map.

Bug: 204065759
Test: Enabled flag and got no crash
Change-Id: Ice9f93af76c02eeba53c934687358ed806c60b60

quickstep/src/com/android/launcher3/model/WidgetsPredictionUpdateTask.java

diff --git a/quickstep/src/com/android/launcher3/model/WidgetsPredictionUpdateTask.java b/quickstep/src/com/android/launcher3/model/WidgetsPredictionUpdateTask.java
index 3d891e8..4be83dc 100644
--- a/quickstep/src/com/android/launcher3/model/WidgetsPredictionUpdateTask.java
+++ b/quickstep/src/com/android/launcher3/model/WidgetsPredictionUpdateTask.java
@@ -83,7 +83,7 @@
             }
         } else {
             Map<ComponentKey, WidgetItem> widgetItems =
-                    allWidgets.values().stream().flatMap(List::stream)
+                    allWidgets.values().stream().flatMap(List::stream).distinct()
                             .collect(Collectors.toMap(widget -> (ComponentKey) widget,
                                     widget -> widget));
             for (AppTarget app : mTargets) {