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 "location_data_rdb_helper.h"
17
18 #include "if_system_ability_manager.h"
19 #include "iservice_registry.h"
20 #include "location_sa_load_manager.h"
21 #include "system_ability_definition.h"
22
23 #include "datashare_predicates.h"
24 #include "datashare_values_bucket.h"
25 #include "common_utils.h"
26 #include "i_locator.h"
27 #include "location_log.h"
28 #include "location_data_rdb_manager.h"
29
30 namespace OHOS {
31 namespace Location {
32 constexpr const char* LOCATION_DATA_COLUMN_KEYWORD = "KEYWORD";
33 constexpr const char* LOCATION_DATA_COLUMN_VALUE = "VALUE";
34
GetInstance()35 LocationDataRdbHelper* LocationDataRdbHelper::GetInstance()
36 {
37 static LocationDataRdbHelper data;
38 return &data;
39 }
40
LocationDataRdbHelper()41 LocationDataRdbHelper::LocationDataRdbHelper()
42 {
43 Initialize();
44 }
45
~LocationDataRdbHelper()46 LocationDataRdbHelper::~LocationDataRdbHelper()
47 {
48 remoteObj_ = nullptr;
49 }
50
Initialize()51 void LocationDataRdbHelper::Initialize()
52 {
53 auto remote = sptr<ILocator>(new (std::nothrow) IRemoteStub<ILocator>());
54 if (remote == nullptr) {
55 LBSLOGE(LOCATOR_STANDARD, "%{public}s: remoteObject is nullptr", __func__);
56 return;
57 }
58 remoteObj_ = remote->AsObject();
59 }
60
CreateDataShareHelper()61 std::shared_ptr<DataShare::DataShareHelper> LocationDataRdbHelper::CreateDataShareHelper()
62 {
63 if (remoteObj_ == nullptr) {
64 LBSLOGE(LOCATOR_STANDARD, "%{public}s: remoteObject is nullptr, reInitialize", __func__);
65 Initialize();
66 }
67 return DataShare::DataShareHelper::Creator(remoteObj_,
68 LocationDataRdbManager::GetLocationDataUriByCurrentUserId("location_enable"));
69 }
70
ReleaseDataShareHelper(std::shared_ptr<DataShare::DataShareHelper> & dataShareHelper)71 void LocationDataRdbHelper::ReleaseDataShareHelper(std::shared_ptr<DataShare::DataShareHelper>& dataShareHelper)
72 {
73 if (dataShareHelper == nullptr) {
74 return;
75 }
76 dataShareHelper->Release();
77 }
78
RegisterDataObserver(const Uri & uri,const sptr<AAFwk::IDataAbilityObserver> & dataObserver)79 LocationErrCode LocationDataRdbHelper::RegisterDataObserver(
80 const Uri &uri, const sptr<AAFwk::IDataAbilityObserver> &dataObserver)
81 {
82 auto dataShareHelper = CreateDataShareHelper();
83 if (dataShareHelper == nullptr) {
84 return ERRCODE_SERVICE_UNAVAILABLE;
85 }
86 dataShareHelper->RegisterObserver(uri, dataObserver);
87 ReleaseDataShareHelper(dataShareHelper);
88 return ERRCODE_SUCCESS;
89 }
90
UnregisterDataObserver(const Uri & uri,const sptr<AAFwk::IDataAbilityObserver> & dataObserver)91 LocationErrCode LocationDataRdbHelper::UnregisterDataObserver(
92 const Uri &uri, const sptr<AAFwk::IDataAbilityObserver> &dataObserver)
93 {
94 auto dataShareHelper = CreateDataShareHelper();
95 if (dataShareHelper == nullptr) {
96 return ERRCODE_SERVICE_UNAVAILABLE;
97 }
98 dataShareHelper->UnregisterObserver(uri, dataObserver);
99 ReleaseDataShareHelper(dataShareHelper);
100 return ERRCODE_SUCCESS;
101 }
102
GetValue(Uri & uri,const std::string & column,int32_t & value)103 LocationErrCode LocationDataRdbHelper::GetValue(Uri &uri, const std::string &column, int32_t &value)
104 {
105 auto dataShareHelper = CreateDataShareHelper();
106 if (dataShareHelper == nullptr) {
107 return ERRCODE_SERVICE_UNAVAILABLE;
108 }
109 DataShare::DataSharePredicates predicates;
110 std::vector<std::string> columns;
111 predicates.EqualTo(LOCATION_DATA_COLUMN_KEYWORD, column);
112 auto rows = dataShareHelper->Query(uri, predicates, columns);
113 if (rows == nullptr) {
114 LBSLOGE(LOCATOR_STANDARD, "%{public}s can not get rows", __func__);
115 ReleaseDataShareHelper(dataShareHelper);
116 return ERRCODE_SERVICE_UNAVAILABLE;
117 }
118 rows->GoToFirstRow();
119 int32_t columnIndex;
120 rows->GetColumnIndex(LOCATION_DATA_COLUMN_VALUE, columnIndex);
121 std::string valueStr;
122 int32_t ret = rows->GetString(columnIndex, valueStr);
123 if (ret != 0) {
124 LBSLOGE(LOCATOR_STANDARD, "%{public}s can not get value", __func__);
125 rows->Close();
126 ReleaseDataShareHelper(dataShareHelper);
127 return ERRCODE_SERVICE_UNAVAILABLE;
128 }
129 rows->Close();
130 ReleaseDataShareHelper(dataShareHelper);
131 value = atoi(valueStr.c_str());
132 LBSLOGD(LOCATOR_STANDARD, "LocationDataRdbHelper:%{public}s success, value = %{public}d", __func__, value);
133 return ERRCODE_SUCCESS;
134 }
135
SetValue(Uri & uri,const std::string & column,int & value)136 LocationErrCode LocationDataRdbHelper::SetValue(Uri &uri, const std::string &column, int &value)
137 {
138 auto dataShareHelper = CreateDataShareHelper();
139 if (dataShareHelper == nullptr) {
140 return ERRCODE_SERVICE_UNAVAILABLE;
141 }
142 int32_t oldVal = 0;
143 LocationErrCode errorCode = GetValue(uri, column, oldVal);
144 DataShare::DataShareValueObject keyObj(column);
145 DataShare::DataShareValueObject valueObj(std::to_string(value));
146 DataShare::DataShareValuesBucket bucket;
147 bucket.Put(LOCATION_DATA_COLUMN_VALUE, valueObj);
148 bucket.Put(LOCATION_DATA_COLUMN_KEYWORD, keyObj);
149 int32_t err;
150 if (errorCode != ERRCODE_SUCCESS) {
151 err = dataShareHelper->Insert(uri, bucket);
152 } else {
153 DataShare::DataSharePredicates predicates;
154 predicates.EqualTo(LOCATION_DATA_COLUMN_KEYWORD, column);
155 err = dataShareHelper->Update(uri, predicates, bucket);
156 }
157 if (err < 0) {
158 LBSLOGE(LOCATOR_STANDARD, "%{public}s: can not set value", __func__);
159 ReleaseDataShareHelper(dataShareHelper);
160 return ERRCODE_SERVICE_UNAVAILABLE;
161 }
162 dataShareHelper->NotifyChange(uri);
163 ReleaseDataShareHelper(dataShareHelper);
164 return ERRCODE_SUCCESS;
165 }
166 } // namespace Location
167 } // namespace OHOS