diff --git a/feature/pagedesktop/src/main/ets/default/common/components/SwiperPage.ets b/feature/pagedesktop/src/main/ets/default/common/components/SwiperPage.ets index be4882061c799244246dd457499b3d379538a26e..3914c5edd02cbb39689c366488aad28cb762c621 100644 --- a/feature/pagedesktop/src/main/ets/default/common/components/SwiperPage.ets +++ b/feature/pagedesktop/src/main/ets/default/common/components/SwiperPage.ets @@ -21,7 +21,8 @@ import { localEventManager, EventConstants, CardItemInfo, - FolderData + FolderData, + BadgeManager } from '@ohos/common'; import AppItem from './AppItem'; import FormItem from './FormItem'; @@ -262,6 +263,10 @@ export default struct SwiperPage { const dragResult = PageDesktopDragHandler.getInstance().onDragDrop(event.getWindowX(), event.getWindowY()); Log.showInfo(TAG, `onDrop dragResult: ${dragResult}`); AppStorage.setOrCreate('selectAppIndex', null); + const dragItemInfo: LauncherDragItemInfo = AppStorage.get('dragItemInfo') as LauncherDragItemInfo; + if (dragItemInfo.bundleName) { + BadgeManager.getInstance().updateBadgeNumber(dragItemInfo.bundleName, dragItemInfo.badgeNumber); + } if (!dragResult) { AppStorage.setOrCreate('dragItemInfo', new LauncherDragItemInfo()); } else { diff --git a/feature/smartdock/src/main/ets/default/layout/ResidentLayout.ets b/feature/smartdock/src/main/ets/default/layout/ResidentLayout.ets index 237bb5247698c57a57093cffa1c9ecb6518a2e91..94e3b3aaa02fb23bfb3fad7fc836c65cda1a39cb 100644 --- a/feature/smartdock/src/main/ets/default/layout/ResidentLayout.ets +++ b/feature/smartdock/src/main/ets/default/layout/ResidentLayout.ets @@ -24,7 +24,8 @@ import { StyleConstants, ResourceManager, Log, - MenuInfo + MenuInfo, + BadgeManager } from '@ohos/common'; import { SmartDockStyleConfig } from '../config/SmartDockStyleConfig'; import SmartDockDragHandler from '../common/SmartDockDragHandler'; @@ -122,6 +123,10 @@ export default struct ResidentLayout { Log.showInfo(TAG, `onDrop event: [${event.getWindowX()}, ${event.getWindowY()}]`); const dragResult = SmartDockDragHandler.getInstance().onDragDrop(event.getWindowX(), event.getWindowY()); AppStorage.setOrCreate('selectAppIndex', null); + const dragItemInfo: LauncherDragItemInfo = AppStorage.get('dragItemInfo') as LauncherDragItemInfo; + if (dragItemInfo.bundleName) { + BadgeManager.getInstance().updateBadgeNumber(dragItemInfo.bundleName, dragItemInfo.badgeNumber); + } if (!dragResult) { AppStorage.setOrCreate('dragItemInfo', new LauncherDragItemInfo()); } else {