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