14d6c458bSopenharmony_ci/*
24d6c458bSopenharmony_ci * Copyright (c) 2022 Huawei Device Co., Ltd.
34d6c458bSopenharmony_ci * Licensed under the Apache License, Version 2.0 (the "License");
44d6c458bSopenharmony_ci * you may not use this file except in compliance with the License.
54d6c458bSopenharmony_ci * You may obtain a copy of the License at
64d6c458bSopenharmony_ci *
74d6c458bSopenharmony_ci *     http://www.apache.org/licenses/LICENSE-2.0
84d6c458bSopenharmony_ci *
94d6c458bSopenharmony_ci * Unless required by applicable law or agreed to in writing, software
104d6c458bSopenharmony_ci * distributed under the License is distributed on an "AS IS" BASIS,
114d6c458bSopenharmony_ci * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
124d6c458bSopenharmony_ci * See the License for the specific language governing permissions and
134d6c458bSopenharmony_ci * limitations under the License.
144d6c458bSopenharmony_ci */
154d6c458bSopenharmony_ci
164d6c458bSopenharmony_ci#ifndef JS_CONCURRENT_MODULE_TASKPOOL_TASK_RUNNER_H
174d6c458bSopenharmony_ci#define JS_CONCURRENT_MODULE_TASKPOOL_TASK_RUNNER_H
184d6c458bSopenharmony_ci
194d6c458bSopenharmony_ci#include <functional>
204d6c458bSopenharmony_ci
214d6c458bSopenharmony_ci#include "native_engine/native_engine.h"
224d6c458bSopenharmony_ci#include "thread.h"
234d6c458bSopenharmony_ci
244d6c458bSopenharmony_cinamespace Commonlibrary::Concurrent::TaskPoolModule {
254d6c458bSopenharmony_cistruct TaskStartCallback {
264d6c458bSopenharmony_ci    using CallbackFunction = std::function<void(void*)>;
274d6c458bSopenharmony_ci
284d6c458bSopenharmony_ci    explicit TaskStartCallback(CallbackFunction function = nullptr, void* dataArgs = nullptr)
294d6c458bSopenharmony_ci        : callback(function), data(dataArgs)
304d6c458bSopenharmony_ci    {}
314d6c458bSopenharmony_ci
324d6c458bSopenharmony_ci    CallbackFunction callback;
334d6c458bSopenharmony_ci    void* data;
344d6c458bSopenharmony_ci};
354d6c458bSopenharmony_ci
364d6c458bSopenharmony_ciclass TaskRunner {
374d6c458bSopenharmony_cipublic:
384d6c458bSopenharmony_ci    // real thread execute
394d6c458bSopenharmony_ci    class TaskInnerRunner : public Thread {
404d6c458bSopenharmony_ci    public:
414d6c458bSopenharmony_ci        explicit TaskInnerRunner(const TaskRunner* runner);
424d6c458bSopenharmony_ci        ~TaskInnerRunner() = default;
434d6c458bSopenharmony_ci        void Run() override;
444d6c458bSopenharmony_ci
454d6c458bSopenharmony_ci    private:
464d6c458bSopenharmony_ci        const TaskRunner* runner_ {nullptr};
474d6c458bSopenharmony_ci    };
484d6c458bSopenharmony_ci
494d6c458bSopenharmony_ci    explicit TaskRunner(TaskStartCallback callback);
504d6c458bSopenharmony_ci    ~TaskRunner();
514d6c458bSopenharmony_ci
524d6c458bSopenharmony_ci    bool Execute();
534d6c458bSopenharmony_ci    void Run() const;
544d6c458bSopenharmony_ci    void Stop();
554d6c458bSopenharmony_ci
564d6c458bSopenharmony_ciprivate:
574d6c458bSopenharmony_ci    TaskInnerRunner* taskInnerRunner_ {nullptr};
584d6c458bSopenharmony_ci    TaskStartCallback callback_;
594d6c458bSopenharmony_ci    uv_thread_t selfThreadId_ {0};
604d6c458bSopenharmony_ci};
614d6c458bSopenharmony_ci} // namespace Commonlibrary::Concurrent::TaskPoolModule
624d6c458bSopenharmony_ci#endif // JS_CONCURRENT_MODULE_TASKPOOL_TASK_RUNNER_H
63