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 HIDEBUG_ASYNCTASK_H_
17#define HIDEBUG_ASYNCTASK_H_
18
19#include "napi/native_api.h"
20
21#include <functional>
22#include <string>
23#include <type_traits>
24
25namespace OHOS {
26namespace HiviewDFX {
27
28napi_value CreateErrorMessage(napi_env env, const std::string& msg);
29
30napi_value CreateErrorMessage(napi_env env, const std::string& errCode, const std::string& msg);
31
32napi_value CreateUndefined(napi_env env);
33
34class AsyncTask {
35public:
36    explicit AsyncTask(const std::string& resourceName): resourceName_(resourceName) {};
37    virtual ~AsyncTask() = default;
38
39    template<typename T, typename = typename std::enable_if<std::is_base_of<AsyncTask, T>::value>::type>
40    static napi_value GetPromise(napi_env env, std::function<void(T*)> setReqParam = [](T*) {})
41    {
42        napi_value promise = nullptr;
43        T* asyncTask = new (std::nothrow) T();
44        if (asyncTask == nullptr) {
45            return nullptr;
46        }
47        setReqParam(asyncTask);
48        if (!asyncTask->CreatePromise(env, promise)) {
49            delete asyncTask;
50            return nullptr;
51        }
52        return promise;
53    };
54
55protected:
56    napi_async_work worker_ = nullptr;
57    napi_deferred deferred_ = nullptr;
58    std::string resourceName_;
59    virtual void Work(napi_env env) = 0;
60    virtual void Done(napi_env env, napi_status status) = 0;
61
62private:
63    bool CreatePromise(napi_env env, napi_value& promise);
64    static void ExecuteCallBack(napi_env env, void* data);
65    static void CompletedCallBack(napi_env env, napi_status status, void* data);
66};
67}
68}
69
70#endif //HIDEBUG_ASYNCTASK_H_
71