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 "ffrt_utils.h"
17
18 namespace OHOS {
19 namespace PowerMgr {
SubmitTask(const FFRTTask & task)20 void FFRTUtils::SubmitTask(const FFRTTask& task)
21 {
22 ffrt::submit(task);
23 }
24
SubmitTaskSync(const FFRTTask & task)25 void FFRTUtils::SubmitTaskSync(const FFRTTask& task)
26 {
27 ffrt::submit(task);
28 ffrt::wait();
29 }
30
SubmitQueueTasks(const std::vector<FFRTTask> & tasks,FFRTQueue & queue)31 void FFRTUtils::SubmitQueueTasks(const std::vector<FFRTTask>& tasks, FFRTQueue& queue)
32 {
33 if (tasks.empty()) {
34 return;
35 }
36 for (auto task : tasks) {
37 queue.submit(task);
38 }
39 }
40
SubmitDelayTask(FFRTTask & task,uint32_t delayMs,FFRTQueue & queue)41 FFRTHandle FFRTUtils::SubmitDelayTask(FFRTTask& task, uint32_t delayMs, FFRTQueue& queue)
42 {
43 using namespace std::chrono;
44 milliseconds ms(delayMs);
45 microseconds us = duration_cast<microseconds>(ms);
46 return queue.submit_h(task, ffrt::task_attr().delay(us.count()));
47 }
48
SubmitDelayTask(FFRTTask & task,uint32_t delayMs,std::shared_ptr<FFRTQueue> queue)49 FFRTHandle FFRTUtils::SubmitDelayTask(FFRTTask& task, uint32_t delayMs, std::shared_ptr<FFRTQueue> queue)
50 {
51 using namespace std::chrono;
52 milliseconds ms(delayMs);
53 microseconds us = duration_cast<microseconds>(ms);
54 return queue->submit_h(task, ffrt::task_attr().delay(us.count()));
55 }
56
SubmitTimeoutTask(const FFRTTask & task,uint32_t timeoutMs)57 bool FFRTUtils::SubmitTimeoutTask(const FFRTTask& task, uint32_t timeoutMs)
58 {
59 ffrt::future<void> future = ffrt::async(task);
60 auto status = future.wait_for(std::chrono::milliseconds(timeoutMs));
61 return status == ffrt::future_status::ready;
62 }
63
CancelTask(FFRTHandle & handle,FFRTQueue & queue)64 void FFRTUtils::CancelTask(FFRTHandle& handle, FFRTQueue& queue)
65 {
66 queue.cancel(handle);
67 }
68
CancelTask(FFRTHandle & handle,std::shared_ptr<FFRTQueue> queue)69 void FFRTUtils::CancelTask(FFRTHandle& handle, std::shared_ptr<FFRTQueue> queue)
70 {
71 queue->cancel(handle);
72 }
73
Mutex()74 FFRTUtils::Mutex::Mutex()
75 {
76 mutex_ = new ffrt::mutex();
77 }
78
~Mutex()79 FFRTUtils::Mutex::~Mutex()
80 {
81 if (!mutex_) {
82 return;
83 }
84 mutex_ = nullptr;
85 delete mutex_;
86 }
87
Lock()88 void FFRTUtils::Mutex::Lock()
89 {
90 if (!mutex_) {
91 return;
92 }
93 mutex_->lock();
94 }
95
TryLock()96 bool FFRTUtils::Mutex::TryLock()
97 {
98 if (!mutex_) {
99 return false;
100 }
101 return mutex_->try_lock();
102 }
103
Unlock()104 void FFRTUtils::Mutex::Unlock()
105 {
106 if (!mutex_) {
107 return;
108 }
109 mutex_->unlock();
110 }
111 } // namespace PowerMgr
112 } // namespace OHOS