1 /* 2 * Copyright (c) 2022-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 #ifndef OHOS_TASK_QUEUE_H 17 #define OHOS_TASK_QUEUE_H 18 19 #include <condition_variable> 20 #include <mutex> 21 #include <queue> 22 #include <thread> 23 24 #include "task_impl_interface.h" 25 26 namespace OHOS { 27 namespace DistributedHardware { 28 class TaskQueue { 29 public: TaskQueue(uint32_t maxSize)30 explicit TaskQueue(uint32_t maxSize) : maxSize_(maxSize) {}; 31 ~TaskQueue() = default; 32 int32_t Produce(std::shared_ptr<TaskImplInterface> &task); 33 void Start(); 34 void Stop(); 35 36 private: 37 void Consume(std::shared_ptr<TaskImplInterface> &task); 38 void Run(); 39 40 private: 41 static constexpr uint8_t TASK_WAIT_TIME = 20; 42 static constexpr const char* MAIN_THREAD_LOOP = "mainThreadLoop"; 43 44 std::mutex taskQueueMutex_; 45 std::condition_variable taskQueueCond_; 46 std::queue<std::shared_ptr<TaskImplInterface>> taskQueue_; 47 bool taskQueueReady_ = false; 48 bool isQuitTaskQueue_ = false; 49 std::thread mainThreadLoop_; 50 uint32_t maxSize_; 51 }; 52 } // DistributedHardware 53 } // OHOS 54 #endif // OHOS_TASK_QUEUE_H 55