1 /*
2  * Copyright (c) 2022 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 #include "is_last_work_time_out.h"
16 
17 #include <new>
18 
19 #include "common.h"
20 #include "workscheduler_srv_client.h"
21 #include "work_sched_hilog.h"
22 #include "work_sched_errors.h"
23 
24 namespace OHOS {
25 namespace WorkScheduler {
26 const uint32_t WORK_ID_INDEX = 0;
27 const uint32_t CALLBACK_INDEX = 1;
28 const uint32_t IS_LAST_WORK_TIME_OUT_MIN_PARAMS = 1;
29 const uint32_t IS_LAST_WORK_TIME_OUT_MAX_PARAMS = 2;
30 
31 struct IsLastWorkTimeOutParamsInfo {
32     int32_t workId = -1;
33     napi_ref callback = nullptr;
34 };
35 
36 struct AsyncCallbackIsLastWorkTimeOut : public AsyncWorkData {
AsyncCallbackIsLastWorkTimeOutOHOS::WorkScheduler::AsyncCallbackIsLastWorkTimeOut37     explicit AsyncCallbackIsLastWorkTimeOut(napi_env env) : AsyncWorkData(env) {}
38     int32_t workId {-1};
39     bool result {false};
40 };
41 
ParseParameters(const napi_env & env,const napi_callback_info & info,IsLastWorkTimeOutParamsInfo & params)42 napi_value ParseParameters(const napi_env &env, const napi_callback_info &info, IsLastWorkTimeOutParamsInfo &params)
43 {
44     size_t argc = IS_LAST_WORK_TIME_OUT_MAX_PARAMS;
45     napi_value argv[IS_LAST_WORK_TIME_OUT_MAX_PARAMS] = {nullptr};
46     NAPI_CALL(env, napi_get_cb_info(env, info, &argc, argv, NULL, NULL));
47     if (argc != IS_LAST_WORK_TIME_OUT_MAX_PARAMS && argc != IS_LAST_WORK_TIME_OUT_MIN_PARAMS) {
48         Common::HandleParamErr(env, E_PARAM_NUMBER_ERR);
49         return nullptr;
50     }
51 
52     // argv[0] : workId
53     if (!Common::MatchValueType(env, argv[WORK_ID_INDEX], napi_number)) {
54         Common::HandleParamErr(env, E_WORKID_ERR);
55         return nullptr;
56     }
57     napi_get_value_int32(env, argv[WORK_ID_INDEX], &params.workId);
58 
59     // argv[1]: callback
60     if (argc == IS_LAST_WORK_TIME_OUT_MAX_PARAMS) {
61         napi_valuetype valuetype = napi_undefined;
62         NAPI_CALL(env, napi_typeof(env, argv[CALLBACK_INDEX], &valuetype));
63         if (valuetype != napi_function) {
64             Common::HandleParamErr(env, E_CALLBACK_TYPE_ERR);
65             return nullptr;
66         }
67         napi_create_reference(env, argv[CALLBACK_INDEX], 1, &params.callback);
68     }
69     return Common::NapiGetNull(env);
70 }
71 
IsLastWorkTimeOut(napi_env env,napi_callback_info info)72 napi_value IsLastWorkTimeOut(napi_env env, napi_callback_info info)
73 {
74     WS_HILOGD("Is LastWork TimeOut napi begin.");
75 
76     // Get params.
77     IsLastWorkTimeOutParamsInfo params;
78     if (ParseParameters(env, info, params) == nullptr) {
79         return Common::JSParaError(env, params.callback);
80     }
81 
82     napi_value promise = nullptr;
83     AsyncCallbackIsLastWorkTimeOut *asyncCallbackInfo = new (std::nothrow) AsyncCallbackIsLastWorkTimeOut(env);
84     if (!asyncCallbackInfo) {
85         return Common::JSParaError(env, params.callback);
86     }
87     std::unique_ptr<AsyncCallbackIsLastWorkTimeOut> callbackPtr {asyncCallbackInfo};
88     asyncCallbackInfo->workId = params.workId;
89     WS_HILOGD("asyncCallbackInfo->workId: %{public}d", asyncCallbackInfo->workId);
90     Common::PaddingAsyncWorkData(env, params.callback, *asyncCallbackInfo, promise);
91 
92     napi_value resourceName = nullptr;
93     NAPI_CALL(env, napi_create_string_latin1(env, "IsLastWorkTimeOut", NAPI_AUTO_LENGTH, &resourceName));
94 
95     NAPI_CALL(env, napi_create_async_work(env, nullptr, resourceName,
96         [](napi_env env, void *data) {
97             AsyncCallbackIsLastWorkTimeOut *asyncCallbackInfo = static_cast<AsyncCallbackIsLastWorkTimeOut *>(data);
98             asyncCallbackInfo->errorCode =
99                 WorkSchedulerSrvClient::GetInstance().IsLastWorkTimeout(asyncCallbackInfo->workId,
100                 asyncCallbackInfo->result);
101             asyncCallbackInfo->errorMsg = Common::FindErrMsg(env, asyncCallbackInfo->errorCode);
102         },
103         [](napi_env env, napi_status status, void *data) {
104             AsyncCallbackIsLastWorkTimeOut *asyncCallbackInfo = static_cast<AsyncCallbackIsLastWorkTimeOut *>(data);
105             std::unique_ptr<AsyncCallbackIsLastWorkTimeOut> callbackPtr {asyncCallbackInfo};
106             if (asyncCallbackInfo != nullptr) {
107                 napi_value result = nullptr;
108                 if (asyncCallbackInfo->errorCode != ERR_OK) {
109                     result = Common::NapiGetNull(env);
110                 } else {
111                     napi_get_boolean(env, asyncCallbackInfo->result, &result);
112                 }
113                 Common::ReturnCallbackPromise(env, *asyncCallbackInfo, result);
114             }
115         },
116         static_cast<AsyncCallbackIsLastWorkTimeOut *>(asyncCallbackInfo),
117         &asyncCallbackInfo->asyncWork));
118 
119     NAPI_CALL(env, napi_queue_async_work(env, asyncCallbackInfo->asyncWork));
120     callbackPtr.release();
121 
122     WS_HILOGD("Is LastWork TimeOut napi end.");
123     if (asyncCallbackInfo->isCallback) {
124         return Common::NapiGetNull(env);
125     } else {
126         return promise;
127     }
128 }
129 } // namespace WorkScheduler
130 } // namespace OHOS