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 "rdb_data_ability_utils.h"
17 #include "result_set_utils.h"
18
19 using namespace OHOS::RdbDataAbilityAdapter;
20 using namespace OHOS::DataShare;
21 using namespace OHOS::NativeRdb;
22
RdbDataAbilityUtils()23 RdbDataAbilityUtils::RdbDataAbilityUtils()
24 {
25 }
26
~RdbDataAbilityUtils()27 RdbDataAbilityUtils::~RdbDataAbilityUtils()
28 {
29 }
30
ToDataShareValuesBucket(const ValuesBucket & valuesBucket)31 DataShareValuesBucket RdbDataAbilityUtils::ToDataShareValuesBucket(const ValuesBucket &valuesBucket)
32 {
33 std::map<std::string, DataShareValueObject> values;
34 std::map<std::string, ValueObject> valuesMap;
35 valuesBucket.GetAll(valuesMap);
36 for (auto &[key, value] : valuesMap) {
37 if (value.GetType() == ValueObjectType::TYPE_BOOL) {
38 values.insert(std::make_pair(key, DataShareValueObject(value.operator bool())));
39 } else if (value.GetType() == ValueObjectType::TYPE_INT) {
40 values.insert(std::make_pair(key, DataShareValueObject(value.operator int())));
41 } else if (value.GetType() == ValueObjectType::TYPE_DOUBLE) {
42 values.insert(std::make_pair(key, DataShareValueObject(value.operator double())));
43 } else if (value.GetType() == ValueObjectType::TYPE_STRING) {
44 values.insert(std::make_pair(key, DataShareValueObject(value.operator std::string())));
45 } else if (value.GetType() == ValueObjectType::TYPE_BLOB) {
46 values.insert(std::make_pair(key, DataShareValueObject(value.operator std::vector<uint8_t>())));
47 }
48 }
49 return DataShareValuesBucket(values);
50 }
51
ToDataSharePredicates(const DataAbilityPredicates & predicates)52 DataSharePredicates RdbDataAbilityUtils::ToDataSharePredicates(const DataAbilityPredicates &predicates)
53 {
54 DataSharePredicates dataSharePredicates;
55
56 if (predicates.IsDistinct()) {
57 dataSharePredicates.Distinct();
58 }
59 if (!predicates.GetGroup().empty()) {
60 std::vector<std::string> groups;
61 groups.push_back(predicates.GetGroup());
62 dataSharePredicates.GroupBy(groups);
63 }
64 if (!predicates.GetIndex().empty()) {
65 dataSharePredicates.IndexedBy(predicates.GetIndex());
66 }
67 if (predicates.GetLimit() != 0 || predicates.GetOffset()) {
68 dataSharePredicates.Limit(predicates.GetLimit(), predicates.GetOffset());
69 }
70
71 dataSharePredicates.SetSettingMode(QUERY_LANGUAGE);
72 dataSharePredicates.SetWhereClause(predicates.GetWhereClause());
73 dataSharePredicates.SetWhereArgs(predicates.GetWhereArgs());
74 dataSharePredicates.SetOrder(predicates.GetOrder());
75 return dataSharePredicates;
76 }
77
ToAbsSharedResultSet(std::shared_ptr<DSResultSet> resultSet)78 std::shared_ptr<AbsSharedResultSet> RdbDataAbilityUtils::ToAbsSharedResultSet(std::shared_ptr<DSResultSet> resultSet)
79 {
80 return std::make_shared<ResultSetUtils>(resultSet);
81 }
82