1 /*
2 * Copyright (C) 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 "net_datashare_utils.h"
17
18 #include <vector>
19
20 #include "net_manager_constants.h"
21 #include "net_mgr_log_wrapper.h"
22
23 namespace OHOS {
24 namespace NetManagerStandard {
25 namespace {
26 constexpr const char *SETTINGS_DATASHARE_URI =
27 "datashare:///com.ohos.settingsdata/entry/settingsdata/SETTINGSDATA?Proxy=true";
28 constexpr const char *SETTINGS_DATA_COLUMN_KEYWORD = "KEYWORD";
29 constexpr const char *SETTINGS_DATA_COLUMN_VALUE = "VALUE";
30
31 constexpr int INVALID_VALUE = -1;
32 } // namespace
33
NetDataShareHelperUtils()34 NetDataShareHelperUtils::NetDataShareHelperUtils()
35 {
36 dataShareHelper_ = CreateDataShareHelper();
37 }
38
CreateDataShareHelper()39 std::shared_ptr<DataShare::DataShareHelper> NetDataShareHelperUtils::CreateDataShareHelper()
40 {
41 sptr<ISystemAbilityManager> saManager = SystemAbilityManagerClient::GetInstance().GetSystemAbilityManager();
42 if (saManager == nullptr) {
43 NETMGR_LOG_E("NetDataShareHelperUtils GetSystemAbilityManager failed.");
44 return nullptr;
45 }
46 sptr<IRemoteObject> remoteObj = saManager->GetSystemAbility(COMM_NET_CONN_MANAGER_SYS_ABILITY_ID);
47 if (remoteObj == nullptr) {
48 NETMGR_LOG_E("NetDataShareHelperUtils GetSystemAbility Service Failed.");
49 return nullptr;
50 }
51 return DataShare::DataShareHelper::Creator(remoteObj, SETTINGS_DATASHARE_URI);
52 }
53
Query(Uri & uri,const std::string & key,std::string & value)54 int32_t NetDataShareHelperUtils::Query(Uri &uri, const std::string &key, std::string &value)
55 {
56 if (dataShareHelper_ == nullptr) {
57 NETMGR_LOG_E("dataShareHelper_ is nullptr");
58 return NETMANAGER_ERROR;
59 }
60 DataShare::DataSharePredicates predicates;
61 std::vector<std::string> columns;
62 predicates.EqualTo(SETTINGS_DATA_COLUMN_KEYWORD, key);
63 auto result = dataShareHelper_->Query(uri, predicates, columns);
64 if (result == nullptr) {
65 NETMGR_LOG_E("DataShareHelper query error, result is null");
66 return NETMANAGER_ERROR;
67 }
68
69 if (result->GoToFirstRow() != DataShare::E_OK) {
70 NETMGR_LOG_E("DataShareHelper query failed,go to first row error");
71 result->Close();
72 return NETMANAGER_ERROR;
73 }
74
75 int columnIndex;
76 result->GetColumnIndex(SETTINGS_DATA_COLUMN_VALUE, columnIndex);
77 result->GetString(columnIndex, value);
78 result->Close();
79 NETMGR_LOG_I("DataShareHelper query success,value[%{public}s]", value.c_str());
80 return NETMANAGER_SUCCESS;
81 }
82
Insert(Uri & uri,const std::string & key,const std::string & value)83 int32_t NetDataShareHelperUtils::Insert(Uri &uri, const std::string &key, const std::string &value)
84 {
85 if (dataShareHelper_ == nullptr) {
86 NETMGR_LOG_E("dataShareHelper_ is nullptr");
87 return NETMANAGER_ERROR;
88 }
89 DataShare::DataShareValuesBucket valuesBucket;
90 DataShare::DataShareValueObject keyObj(key);
91 DataShare::DataShareValueObject valueObj(value);
92 valuesBucket.Put(SETTINGS_DATA_COLUMN_KEYWORD, keyObj);
93 valuesBucket.Put(SETTINGS_DATA_COLUMN_VALUE, valueObj);
94 int32_t result = dataShareHelper_->Insert(uri, valuesBucket);
95 if (result == INVALID_VALUE) {
96 NETMGR_LOG_E("DataShareHelper insert failed, insert result:%{public}d", result);
97 return NETMANAGER_ERROR;
98 }
99 dataShareHelper_->NotifyChange(uri);
100 NETMGR_LOG_I("DataShareHelper insert success");
101 return NETMANAGER_SUCCESS;
102 }
103
Update(Uri & uri,const std::string & key,const std::string & value)104 int32_t NetDataShareHelperUtils::Update(Uri &uri, const std::string &key, const std::string &value)
105 {
106 if (dataShareHelper_ == nullptr) {
107 NETMGR_LOG_E("dataShareHelper_ is nullptr");
108 return NETMANAGER_ERROR;
109 }
110 std::string queryValue;
111 int32_t ret = Query(uri, key, queryValue);
112 if (ret == NETMANAGER_ERROR) {
113 return Insert(uri, key, value);
114 }
115
116 DataShare::DataShareValuesBucket valuesBucket;
117 DataShare::DataShareValueObject valueObj(value);
118 valuesBucket.Put(SETTINGS_DATA_COLUMN_VALUE, valueObj);
119 DataShare::DataSharePredicates predicates;
120 predicates.EqualTo(SETTINGS_DATA_COLUMN_KEYWORD, key);
121 int32_t result = dataShareHelper_->Update(uri, predicates, valuesBucket);
122 if (result == INVALID_VALUE) {
123 return NETMANAGER_ERROR;
124 }
125 dataShareHelper_->NotifyChange(uri);
126 NETMGR_LOG_I("DataShareHelper update success");
127 return NETMANAGER_SUCCESS;
128 }
129 } // namespace NetManagerStandard
130 } // namespace OHOS