Remove badge info from map when count is 0.
Bug: 35803112
Change-Id: I680b92aac24bf5d083d64e9a4cd35471637a3809
diff --git a/src/com/android/launcher3/popup/PopupDataProvider.java b/src/com/android/launcher3/popup/PopupDataProvider.java
index ee2930f..f0ccb1b 100644
--- a/src/com/android/launcher3/popup/PopupDataProvider.java
+++ b/src/com/android/launcher3/popup/PopupDataProvider.java
@@ -75,6 +75,9 @@
notificationWasAddedOrRemoved = shouldBeFilteredOut
? badgeInfo.removeNotificationKey(notificationKey)
: badgeInfo.addNotificationKeyIfNotExists(notificationKey);
+ if (badgeInfo.getNotificationCount() == 0) {
+ mPackageUserToBadgeInfos.remove(postedPackageUserKey);
+ }
}
updateLauncherIconBadges(Utilities.singletonHashSet(postedPackageUserKey),
notificationWasAddedOrRemoved);