From ae238f22ab8cfe2c4d252a99b3d33d9a2b64f214 Mon Sep 17 00:00:00 2001 From: fangjinliang1 Date: Fri, 14 Jan 2022 12:04:21 +0800 Subject: [PATCH] Multi users modify Signed-off-by: fangjinliang1 Change-Id: I0ed2f845206b7a4362a7bbaafe0ad5488e9c049a --- frameworks/ans/core/common/include/ans_const_define.h | 3 +++ services/ans/src/advanced_notification_service.cpp | 5 ++++- services/ans/src/notification_subscriber_manager.cpp | 7 +++++-- 3 files changed, 12 insertions(+), 3 deletions(-) diff --git a/frameworks/ans/core/common/include/ans_const_define.h b/frameworks/ans/core/common/include/ans_const_define.h index c498d2bcb..bda3a1ffe 100644 --- a/frameworks/ans/core/common/include/ans_const_define.h +++ b/frameworks/ans/core/common/include/ans_const_define.h @@ -32,8 +32,11 @@ constexpr uint32_t MAX_ICON_SIZE = 50 * 1024; constexpr uint32_t MAX_PICTURE_SIZE = 2 * 1024 * 1024; constexpr bool SUPPORT_DO_NOT_DISTRUB = true; constexpr uint32_t SYSTEM_SERVICE_UID = 1000; + constexpr int32_t SUBSCRIBE_USER_INIT = -1; constexpr int32_t SUBSCRIBE_USER_ALL = -2; +constexpr int32_t SUBSCRIBE_USER_SYSTEM_BEGIN = 0; +constexpr int32_t SUBSCRIBE_USER_SYSTEM_END = 99; // Default sound for notification const static Uri DEFAULT_NOTIFICATION_SOUND("file://system/etc/Light.ogg"); diff --git a/services/ans/src/advanced_notification_service.cpp b/services/ans/src/advanced_notification_service.cpp index f936f4655..953a519df 100644 --- a/services/ans/src/advanced_notification_service.cpp +++ b/services/ans/src/advanced_notification_service.cpp @@ -144,7 +144,7 @@ inline ErrCode CheckPictureSize(const sptr &request) return result; } -inline ErrCode PrepereNotificationRequest(const sptr &request) +ErrCode PrepereNotificationRequest(const sptr &request) { std::string bundle = GetClientBundleName(); if (bundle.empty()) { @@ -161,6 +161,9 @@ inline ErrCode PrepereNotificationRequest(const sptr &reque int userId = SUBSCRIBE_USER_INIT; OHOS::AccountSA::OsAccountManager::GetOsAccountLocalIdFromUid(uid, userId); + if (userId >= SUBSCRIBE_USER_SYSTEM_BEGIN && userId <= SUBSCRIBE_USER_SYSTEM_END) { + userId = SUBSCRIBE_USER_ALL; + } request->SetCreatorUserId(userId); ErrCode result = CheckPictureSize(request); diff --git a/services/ans/src/notification_subscriber_manager.cpp b/services/ans/src/notification_subscriber_manager.cpp index 8485bea32..1e850b74b 100644 --- a/services/ans/src/notification_subscriber_manager.cpp +++ b/services/ans/src/notification_subscriber_manager.cpp @@ -206,7 +206,8 @@ void NotificationSubscriberManager::AddRecordInfo( record->subscribedAll = false; } record->userId = subscribeInfo->GetAppUserId(); - if (record->userId == SUBSCRIBE_USER_INIT) { + if (record->userId == SUBSCRIBE_USER_INIT || + (record->userId >= SUBSCRIBE_USER_SYSTEM_BEGIN && record->userId <= SUBSCRIBE_USER_SYSTEM_END)) { record->userId = SUBSCRIBE_USER_ALL; } } else { @@ -286,7 +287,9 @@ void NotificationSubscriberManager::NotifyConsumedInner( auto BundleNames = notification->GetBundleName(); auto iter = std::find(record->bundleList_.begin(), record->bundleList_.end(), BundleNames); if (!record->subscribedAll == (iter != record->bundleList_.end()) && - (notification->GetUserId() == record->userId || record->userId == SUBSCRIBE_USER_ALL)) { + (notification->GetUserId() == record->userId || + notification->GetUserId() == SUBSCRIBE_USER_ALL || + record->userId == SUBSCRIBE_USER_ALL)) { record->subscriber->OnConsumed(notification, notificationMap); record->subscriber->OnConsumed(notification); } -- Gitee