1 /*
2 * Copyright (c) 2024 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 "startup_task.h"
17 #include "hilog_tag_wrapper.h"
18
19 namespace OHOS {
20 namespace AbilityRuntime {
StartupTask(const std::string & name)21 StartupTask::StartupTask(const std::string &name) : name_(name), state_(State::CREATED)
22 {}
23
24 StartupTask::~StartupTask() = default;
25
GetName() const26 const std::string& StartupTask::GetName() const
27 {
28 return name_;
29 }
30
GetDependencies() const31 std::vector<std::string> StartupTask::GetDependencies() const
32 {
33 return dependencies_;
34 }
35
SetDependencies(const std::vector<std::string> & dependencies)36 void StartupTask::SetDependencies(const std::vector<std::string> &dependencies)
37 {
38 dependencies_ = dependencies;
39 }
40
SaveResult(const std::shared_ptr<StartupTaskResult> & result)41 void StartupTask::SaveResult(const std::shared_ptr<StartupTaskResult> &result)
42 {
43 if (result == nullptr) {
44 TAG_LOGE(AAFwkTag::STARTUP, "startup task: %{public}s, result null", name_.c_str());
45 return;
46 }
47 TAG_LOGD(AAFwkTag::STARTUP,
48 "startup task: %{public}s, result code: %{public}d", name_.c_str(), result->GetResultCode());
49 result_ = result;
50 if (result->GetResultCode() == ERR_OK) {
51 state_ = State::INITIALIZED;
52 } else {
53 state_ = State::CREATED;
54 }
55 }
56
RemoveResult()57 int32_t StartupTask::RemoveResult()
58 {
59 if (state_ != State::INITIALIZED) {
60 TAG_LOGE(AAFwkTag::STARTUP, "%{public}s, result not init", name_.c_str());
61 return ERR_STARTUP_INTERNAL_ERROR;
62 }
63 result_ = nullptr;
64 state_ = State::CREATED;
65 return ERR_OK;
66 }
67
GetResult() const68 const std::shared_ptr<StartupTaskResult>& StartupTask::GetResult() const
69 {
70 return result_;
71 }
72
GetState() const73 StartupTask::State StartupTask::GetState() const
74 {
75 return state_;
76 }
77
DumpDependencies() const78 std::string StartupTask::DumpDependencies() const
79 {
80 if (dependencies_.empty()) {
81 return "";
82 }
83 bool isFirst = true;
84 std::string dumpResult;
85 for (const auto &iter : dependencies_) {
86 if (isFirst) {
87 dumpResult = iter;
88 isFirst = false;
89 } else {
90 dumpResult += ", " + iter;
91 }
92 }
93 return dumpResult;
94 }
95
GetDependenciesCount() const96 uint32_t StartupTask::GetDependenciesCount() const
97 {
98 return dependencies_.size();
99 }
100
GetWaitOnMainThread() const101 bool StartupTask::GetWaitOnMainThread() const
102 {
103 return waitOnMainThread_;
104 }
105
SetWaitOnMainThread(bool waitOnMainThread)106 void StartupTask::SetWaitOnMainThread(bool waitOnMainThread)
107 {
108 waitOnMainThread_ = waitOnMainThread;
109 }
110
GetCallCreateOnMainThread() const111 bool StartupTask::GetCallCreateOnMainThread() const
112 {
113 return callCreateOnMainThread_;
114 }
115
SetCallCreateOnMainThread(bool callCreateOnMainThread)116 void StartupTask::SetCallCreateOnMainThread(bool callCreateOnMainThread)
117 {
118 callCreateOnMainThread_ = callCreateOnMainThread;
119 }
120
AddExtraCallback(std::unique_ptr<StartupTaskResultCallback> callback)121 int32_t StartupTask::AddExtraCallback(std::unique_ptr<StartupTaskResultCallback> callback)
122 {
123 if (state_ != State::INITIALIZING) {
124 TAG_LOGE(AAFwkTag::STARTUP, "state not INITIALIZING");
125 return ERR_STARTUP_INTERNAL_ERROR;
126 }
127 // extra callback will called while init done
128 extraCallbacks_.emplace_back(std::move(callback));
129 return ERR_OK;
130 }
131
CallExtraCallback(const std::shared_ptr<StartupTaskResult> & result)132 void StartupTask::CallExtraCallback(const std::shared_ptr<StartupTaskResult> &result)
133 {
134 for (auto &callback : extraCallbacks_) {
135 if (callback != nullptr) {
136 callback->Call(result);
137 }
138 }
139 extraCallbacks_.clear();
140 }
141
RunTaskPreInit(std::unique_ptr<StartupTaskResultCallback> & callback)142 int32_t StartupTask::RunTaskPreInit(std::unique_ptr<StartupTaskResultCallback>& callback)
143 {
144 if (state_ != State::CREATED) {
145 TAG_LOGE(AAFwkTag::STARTUP, "%{public}s, state wrong %{public}d", name_.c_str(), static_cast<int32_t>(state_));
146 return ERR_STARTUP_INTERNAL_ERROR;
147 }
148 state_ = State::INITIALIZING;
149 callback->Push([weak = weak_from_this()](const std::shared_ptr<StartupTaskResult>& result) {
150 auto startupTask = weak.lock();
151 if (startupTask == nullptr) {
152 TAG_LOGE(AAFwkTag::STARTUP, "startupTask null");
153 return;
154 }
155 startupTask->SaveResult(result);
156 startupTask->CallExtraCallback(result);
157 });
158 return ERR_OK;
159 }
160 } // namespace AbilityRuntime
161 } // namespace OHOS
162