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 "common/include/task_scheduler.h"
17 #include "hitrace_meter.h"
18 #include "window_manager_hilog.h"
19
20 namespace OHOS::Rosen {
TaskScheduler(const std::string & threadName)21 TaskScheduler::TaskScheduler(const std::string& threadName)
22 {
23 auto runner = AppExecFwk::EventRunner::Create(threadName);
24 handler_ = std::make_shared<AppExecFwk::EventHandler>(runner);
25 }
26
GetEventHandler()27 std::shared_ptr<AppExecFwk::EventHandler> TaskScheduler::GetEventHandler()
28 {
29 return handler_;
30 }
31
PostAsyncTask(Task && task,const std::string & name,int64_t delayTime)32 void TaskScheduler::PostAsyncTask(Task&& task, const std::string& name, int64_t delayTime)
33 {
34 if (delayTime == 0 && handler_->GetEventRunner()->IsCurrentRunnerThread()) {
35 HITRACE_METER_FMT(HITRACE_TAG_WINDOW_MANAGER, "ssm:%s", name.c_str());
36 task();
37 return;
38 }
39 auto localTask = [this, task = std::move(task), name] {
40 HITRACE_METER_FMT(HITRACE_TAG_WINDOW_MANAGER, "ssm:%s", name.c_str());
41 task();
42 ExecuteExportTask();
43 };
44 handler_->PostTask(std::move(localTask), "wms:" + name, delayTime, AppExecFwk::EventQueue::Priority::IMMEDIATE);
45 }
46
PostTask(Task && task,const std::string & name,int64_t delayTime)47 void TaskScheduler::PostTask(Task&& task, const std::string& name, int64_t delayTime)
48 {
49 PostAsyncTask(std::move(task), name, delayTime);
50 }
51
RemoveTask(const std::string & name)52 void TaskScheduler::RemoveTask(const std::string& name)
53 {
54 handler_->RemoveTask("wms:" + name);
55 }
56
PostVoidSyncTask(Task && task,const std::string & name)57 void TaskScheduler::PostVoidSyncTask(Task&& task, const std::string& name)
58 {
59 if (handler_->GetEventRunner()->IsCurrentRunnerThread()) {
60 HITRACE_METER_FMT(HITRACE_TAG_WINDOW_MANAGER, "ssm:%s", name.c_str());
61 task();
62 return;
63 }
64 auto localTask = [this, &task, &name] {
65 HITRACE_METER_FMT(HITRACE_TAG_WINDOW_MANAGER, "ssm:%s", name.c_str());
66 task();
67 ExecuteExportTask();
68 };
69 handler_->PostSyncTask(std::move(localTask), "wms:" + name, AppExecFwk::EventQueue::Priority::IMMEDIATE);
70 }
71
SetExportHandler(const std::shared_ptr<AppExecFwk::EventHandler> & handler)72 void TaskScheduler::SetExportHandler(const std::shared_ptr<AppExecFwk::EventHandler>& handler)
73 {
74 exportHandler_ = handler;
75 }
76
AddExportTask(std::string taskName,Task && task)77 void TaskScheduler::AddExportTask(std::string taskName, Task&& task)
78 {
79 if (handler_->GetEventRunner()->IsCurrentRunnerThread()) {
80 exportFuncMap_[std::move(taskName)] = std::move(task);
81 } else {
82 task();
83 }
84 }
85
ExecuteExportTask()86 void TaskScheduler::ExecuteExportTask()
87 {
88 if (!exportHandler_ || exportFuncMap_.empty()) {
89 return;
90 }
91 auto task = [funcMap = std::move(exportFuncMap_)] {
92 for (auto iter = funcMap.begin(); iter != funcMap.end(); iter++) {
93 HITRACE_METER_FMT(HITRACE_TAG_WINDOW_MANAGER, "ssm:%s", iter->first.c_str());
94 iter->second();
95 }
96 };
97 exportFuncMap_.clear();
98 exportHandler_->PostTask(std::move(task), "wms:exportTask");
99 }
100
StartTraceForSyncTask(const std::string & name)101 void StartTraceForSyncTask(const std::string& name)
102 {
103 StartTraceArgs(HITRACE_TAG_WINDOW_MANAGER, "ssm:%s", name.c_str());
104 }
FinishTraceForSyncTask()105 void FinishTraceForSyncTask()
106 {
107 FinishTrace(HITRACE_TAG_WINDOW_MANAGER);
108 }
109 } // namespace OHOS::Rosen