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 #include "statistics_observer_wrapper.h"
17 
18 #include "constant.h"
19 #include "module_template.h"
20 #include "napi_constant.h"
21 #include "net_stats_client.h"
22 #include "net_stats_constants.h"
23 #include "netmanager_base_log.h"
24 
25 namespace OHOS {
26 namespace NetManagerStandard {
StatisticsObserverWrapper()27 StatisticsObserverWrapper::StatisticsObserverWrapper()
28     : observer_(new StatisticsCallbackObserver()), manager_(new EventManager()), registered_(false)
29 {
30 }
31 
~StatisticsObserverWrapper()32 StatisticsObserverWrapper::~StatisticsObserverWrapper()
33 {
34     delete manager_;
35 }
36 
On(napi_env env,napi_callback_info info,const std::initializer_list<std::string> & events,bool asyncCallback)37 napi_value StatisticsObserverWrapper::On(napi_env env, napi_callback_info info,
38                                          const std::initializer_list<std::string> &events, bool asyncCallback)
39 {
40     size_t paramsCount = MAX_PARAM_NUM;
41     napi_value params[MAX_PARAM_NUM] = {nullptr};
42     NAPI_CALL(env, napi_get_cb_info(env, info, &paramsCount, params, nullptr, nullptr));
43     if (paramsCount != PARAM_OPTIONS_AND_CALLBACK ||
44         NapiUtils::GetValueType(env, params[ARG_INDEX_0]) != napi_string ||
45         NapiUtils::GetValueType(env, params[ARG_INDEX_1]) != napi_function) {
46         NETMANAGER_BASE_LOGE("on off once interface para: [string, function]");
47         napi_throw_error(env, std::to_string(NETMANAGER_ERR_PARAMETER_ERROR).c_str(), "Parameter error");
48         return NapiUtils::GetUndefined(env);
49     }
50 
51     const auto event = NapiUtils::GetStringFromValueUtf8(env, params[ARG_INDEX_0]);
52     if (std::find(events.begin(), events.end(), event) == events.end()) {
53         return NapiUtils::GetUndefined(env);
54     }
55     if (!registered_) {
56         int32_t ret = NetStatsClient::GetInstance().RegisterNetStatsCallback(observer_);
57         NETMANAGER_BASE_LOGI("ret = [%{public}d]", ret);
58         registered_ = (ret == NETMANAGER_SUCCESS);
59         if (!registered_) {
60             NETMANAGER_BASE_LOGE("register callback error");
61             NetBaseErrorCodeConvertor convertor;
62             std::string errorMsg = convertor.ConvertErrorCode(ret);
63             napi_throw_error(env, std::to_string(ret).c_str(), errorMsg.c_str());
64         }
65     }
66     if (registered_) {
67         manager_->AddListener(env, event, params[ARG_INDEX_1], false, asyncCallback);
68     }
69     return NapiUtils::GetUndefined(env);
70 }
71 
Off(napi_env env,napi_callback_info info,const std::initializer_list<std::string> & events,bool asyncCallback)72 napi_value StatisticsObserverWrapper::Off(napi_env env, napi_callback_info info,
73                                           const std::initializer_list<std::string> &events, bool asyncCallback)
74 {
75     napi_value thisVal = nullptr;
76     size_t paramsCount = MAX_PARAM_NUM;
77     napi_value params[MAX_PARAM_NUM] = {nullptr};
78     NAPI_CALL(env, napi_get_cb_info(env, info, &paramsCount, params, &thisVal, nullptr));
79 
80     if (!(paramsCount == PARAM_JUST_OPTIONS || paramsCount == PARAM_OPTIONS_AND_CALLBACK) ||
81         NapiUtils::GetValueType(env, params[ARG_INDEX_0]) != napi_string) {
82         NETMANAGER_BASE_LOGE("on off once interface para: [string, function?]");
83         napi_throw_error(env, std::to_string(NETMANAGER_ERR_PARAMETER_ERROR).c_str(), "Parameter error");
84         return NapiUtils::GetUndefined(env);
85     }
86 
87     if (paramsCount == PARAM_OPTIONS_AND_CALLBACK &&
88         NapiUtils::GetValueType(env, params[ARG_INDEX_1]) != napi_function) {
89         NETMANAGER_BASE_LOGE("on off once interface para: [string, function]");
90         return NapiUtils::GetUndefined(env);
91     }
92 
93     std::string event = NapiUtils::GetStringFromValueUtf8(env, params[ARG_INDEX_0]);
94     if (std::find(events.begin(), events.end(), event) == events.end()) {
95         return NapiUtils::GetUndefined(env);
96     }
97 
98     if (paramsCount == PARAM_OPTIONS_AND_CALLBACK) {
99         manager_->DeleteListener(event, params[ARG_INDEX_1]);
100     } else {
101         manager_->DeleteListener(event);
102     }
103 
104     if (manager_->IsListenerListEmpty()) {
105         auto ret = NetStatsClient::GetInstance().UnregisterNetStatsCallback(observer_);
106         if (ret != NETMANAGER_SUCCESS) {
107             NETMANAGER_BASE_LOGE("unregister ret = %{public}d", ret);
108             NetBaseErrorCodeConvertor convertor;
109             std::string errorMsg = convertor.ConvertErrorCode(ret);
110             napi_throw_error(env, std::to_string(ret).c_str(), errorMsg.c_str());
111         }
112         registered_ = false;
113     }
114     return NapiUtils::GetUndefined(env);
115 }
116 } // namespace NetManagerStandard
117 } // namespace OHOS
118