1 /*
2 * Copyright (c) 2023 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 * http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16 #include "sleep_callback_holder.h"
17 #include "power_common.h"
18
19 namespace OHOS {
20 namespace PowerMgr {
SleepCallbackHolder()21 SleepCallbackHolder::SleepCallbackHolder() {}
22 SleepCallbackHolder::~SleepCallbackHolder() = default;
23
AddCallback(const sptr<ISyncSleepCallback> & callback,SleepPriority priority)24 void SleepCallbackHolder::AddCallback(const sptr<ISyncSleepCallback>& callback, SleepPriority priority)
25 {
26 std::lock_guard<std::mutex> lock(mutex_);
27 switch (priority) {
28 case SleepPriority::LOW: {
29 auto iter = lowPriorityCallbacks_.insert(callback);
30 break;
31 }
32 case SleepPriority::DEFAULT: {
33 auto iter = defaultPriorityCallbacks_.insert(callback);
34 break;
35 }
36 case SleepPriority::HIGH: {
37 auto iter = highPriorityCallbacks_.insert(callback);
38 break;
39 }
40 default: {
41 break;
42 }
43 }
44 }
45
GetHighPriorityCallbacks()46 SleepCallbackHolder::SleepCallbackContainerType SleepCallbackHolder::GetHighPriorityCallbacks()
47 {
48 std::lock_guard<std::mutex> lock(mutex_);
49 return highPriorityCallbacks_;
50 }
51
GetDefaultPriorityCallbacks()52 SleepCallbackHolder::SleepCallbackContainerType SleepCallbackHolder::GetDefaultPriorityCallbacks()
53 {
54 std::lock_guard<std::mutex> lock(mutex_);
55 return defaultPriorityCallbacks_;
56 }
57
GetLowPriorityCallbacks()58 SleepCallbackHolder::SleepCallbackContainerType SleepCallbackHolder::GetLowPriorityCallbacks()
59 {
60 std::lock_guard<std::mutex> lock(mutex_);
61 return lowPriorityCallbacks_;
62 }
63
RemoveCallback(const sptr<ISyncSleepCallback> & callback)64 void SleepCallbackHolder::RemoveCallback(const sptr<ISyncSleepCallback>& callback)
65 {
66 std::lock_guard<std::mutex> lock(mutex_);
67 RemoveCallback(lowPriorityCallbacks_, callback);
68 RemoveCallback(defaultPriorityCallbacks_, callback);
69 RemoveCallback(highPriorityCallbacks_, callback);
70 }
71
RemoveCallback(SleepCallbackHolder::SleepCallbackContainerType & callbacks,const sptr<ISyncSleepCallback> & callback)72 void SleepCallbackHolder::RemoveCallback(
73 SleepCallbackHolder::SleepCallbackContainerType& callbacks, const sptr<ISyncSleepCallback>& callback)
74 {
75 auto iter = callbacks.find(callback);
76 if (iter == callbacks.end()) {
77 POWER_HILOGE(FEATURE_SUSPEND, "Cannot find the sleep callback");
78 return;
79 }
80 callbacks.erase(iter);
81 }
82
83 } // namespace PowerMgr
84 } // namespace OHOS
85