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 #ifndef CORE_THREADING_DISPATHCER_IMPL_H
17 #define CORE_THREADING_DISPATHCER_IMPL_H
18 
19 #include <base/containers/vector.h>
20 #include <core/namespace.h>
21 #include <core/threading/intf_thread_pool.h>
22 
23 #include "threading/dispatcher_task_queue.h"
24 
CORE_BEGIN_NAMESPACE()25 CORE_BEGIN_NAMESPACE()
26 // Wrapper for DispatcherTaskQueue exposing a subset of the functionality.
27 class DispatcherImpl final : public IDispatcherTaskQueue {
28 public:
29     ~DispatcherImpl();
30     explicit DispatcherImpl(const IThreadPool::Ptr& threads);
31 
32     BASE_NS::vector<uint64_t> CollectFinishedTasks() override;
33 
34     void Submit(uint64_t taskIdentifier, IThreadPool::ITask::Ptr&& task) override;
35     void SubmitAfter(uint64_t afterIdentifier, uint64_t taskIdentifier, IThreadPool::ITask::Ptr&& task) override;
36     void SubmitAfter(BASE_NS::array_view<const uint64_t> afterIdentifiers, uint64_t taskIdentifier,
37         IThreadPool::ITask::Ptr&& task) override;
38     void Execute() override;
39     void Clear() override;
40 
41 protected:
42     void Destroy() override;
43 
44 private:
45     DispatcherTaskQueue queue_;
46 };
47 CORE_END_NAMESPACE()
48 
49 #endif // CORE_THREADING_DISPATHCER_IMPL_H
50