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