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 #ifndef ECMASCRIPT_TASKPOOL_TASKPOOL_H 17 #define ECMASCRIPT_TASKPOOL_TASKPOOL_H 18 19 #include <memory> 20 21 #include "ecmascript/common.h" 22 #include "ecmascript/taskpool/runner.h" 23 #include "libpandabase/os/mutex.h" 24 25 namespace panda::ecmascript { 26 class Taskpool { 27 public: 28 PUBLIC_API static Taskpool *GetCurrentTaskpool(); 29 30 Taskpool() = default; ~Taskpool()31 PUBLIC_API ~Taskpool() 32 { 33 os::memory::LockHolder lock(mutex_); 34 runner_->TerminateThread(); 35 isInitialized_ = 0; 36 } 37 38 NO_COPY_SEMANTIC(Taskpool); 39 NO_MOVE_SEMANTIC(Taskpool); 40 41 void Initialize(int threadNum = DEFAULT_TASKPOOL_THREAD_NUM); 42 void Destroy(int32_t id); 43 PostTask(std::unique_ptr<Task> task)44 void PostTask(std::unique_ptr<Task> task) const 45 { 46 ASSERT(isInitialized_ > 0); 47 runner_->PostTask(std::move(task)); 48 } 49 50 // Terminate a task of a specified type 51 void TerminateTask(int32_t id, TaskType type = TaskType::ALL); 52 GetTotalThreadNum()53 uint32_t GetTotalThreadNum() const 54 { 55 return runner_->GetTotalThreadNum(); 56 } 57 IsInThreadPool(std::thread::id id)58 bool IsInThreadPool(std::thread::id id) const 59 { 60 return runner_->IsInThreadPool(id); 61 } 62 63 private: 64 uint32_t TheMostSuitableThreadNum(uint32_t threadNum) const; 65 66 std::unique_ptr<Runner> runner_; 67 int isInitialized_ = 0; 68 os::memory::Mutex mutex_; 69 }; 70 } // namespace panda::ecmascript 71 #endif // ECMASCRIPT_PALTFORM_PLATFORM_H 72