1 /*
2 * Copyright (c) 2021 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 "ecmascript/taskpool/taskpool.h"
17
18 #include "ecmascript/platform/os.h"
19
20 namespace panda::ecmascript {
GetCurrentTaskpool()21 Taskpool *Taskpool::GetCurrentTaskpool()
22 {
23 static Taskpool *taskpool = new Taskpool();
24 return taskpool;
25 }
26
Initialize(int threadNum,std::function<void (os::thread::native_handle_type)> prologueHook,const std::function<void (os::thread::native_handle_type)> epilogueHook)27 void Taskpool::Initialize(int threadNum,
28 std::function<void(os::thread::native_handle_type)> prologueHook,
29 const std::function<void(os::thread::native_handle_type)> epilogueHook)
30 {
31 LockHolder lock(mutex_);
32 if (isInitialized_++ <= 0) {
33 runner_ = std::make_unique<Runner>(TheMostSuitableThreadNum(threadNum), prologueHook, epilogueHook);
34 }
35 }
36
Destroy(int32_t id)37 void Taskpool::Destroy(int32_t id)
38 {
39 ASSERT(id != 0);
40 LockHolder lock(mutex_);
41 if (isInitialized_ <= 0) {
42 return;
43 }
44 isInitialized_--;
45 if (isInitialized_ == 0) {
46 runner_->TerminateThread();
47 } else {
48 runner_->TerminateTask(id, TaskType::ALL);
49 }
50 }
51
TerminateTask(int32_t id,TaskType type)52 void Taskpool::TerminateTask(int32_t id, TaskType type)
53 {
54 if (isInitialized_ <= 0) {
55 return;
56 }
57 runner_->TerminateTask(id, type);
58 }
59
TheMostSuitableThreadNum(uint32_t threadNum) const60 uint32_t Taskpool::TheMostSuitableThreadNum(uint32_t threadNum) const
61 {
62 if (threadNum > 0) {
63 return std::min<uint32_t>(threadNum, MAX_TASKPOOL_THREAD_NUM);
64 }
65 uint32_t numOfThreads = std::min<uint32_t>(NumberOfCpuCore() / 2, MAX_TASKPOOL_THREAD_NUM);
66 if (numOfThreads > MIN_TASKPOOL_THREAD_NUM) {
67 return numOfThreads - 1; // 1 for daemon thread.
68 }
69 return MIN_TASKPOOL_THREAD_NUM; // At least MIN_TASKPOOL_THREAD_NUM GC threads, and 1 extra daemon thread.
70 }
71
ForEachTask(const std::function<void (Task *)> & f)72 void Taskpool::ForEachTask(const std::function<void(Task*)> &f)
73 {
74 if (isInitialized_ <= 0) {
75 return;
76 }
77 runner_->ForEachTask(f);
78 }
79 } // namespace panda::ecmascript
80