1 /*
2 * Copyright (c) 2021-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 "data_ability_operation_builder.h"
17 #include "data_ability_predicates.h"
18 #include "hilog_wrapper.h"
19 #include "values_bucket.h"
20
21 namespace OHOS {
22 namespace AppExecFwk {
DataAbilityOperationBuilder(const int type,const std::shared_ptr<Uri> & uri)23 DataAbilityOperationBuilder::DataAbilityOperationBuilder(const int type, const std::shared_ptr<Uri> &uri)
24 {
25 type_ = type;
26 uri_ = uri;
27 expectedCount_ = 0;
28 interrupted_ = false;
29 valuesBucket_ = nullptr;
30 dataAbilityPredicates_ = nullptr;
31 valuesBucketReferences_ = nullptr;
32 dataAbilityPredicatesBackReferences_.clear();
33 }
~DataAbilityOperationBuilder()34 DataAbilityOperationBuilder::~DataAbilityOperationBuilder()
35 {
36 dataAbilityPredicatesBackReferences_.clear();
37 }
38
Build()39 std::shared_ptr<DataAbilityOperation> DataAbilityOperationBuilder::Build()
40 {
41 HILOG_DEBUG("DataAbilityOperationBuilder::Build start");
42 if (type_ != DataAbilityOperation::TYPE_UPDATE || (valuesBucket_ != nullptr && !valuesBucket_->IsEmpty())) {
43 std::shared_ptr<DataAbilityOperation> operation = std::make_shared<DataAbilityOperation>(shared_from_this());
44 HILOG_DEBUG("DataAbilityOperationBuilder::Build end");
45 return operation;
46 }
47 HILOG_ERROR("DataAbilityOperationBuilder::Build return nullptr");
48 return nullptr;
49 }
50
WithValuesBucket(std::shared_ptr<NativeRdb::ValuesBucket> & values)51 std::shared_ptr<DataAbilityOperationBuilder> DataAbilityOperationBuilder::WithValuesBucket(
52 std::shared_ptr<NativeRdb::ValuesBucket> &values)
53 {
54 HILOG_DEBUG("DataAbilityOperationBuilder::WithValuesBucket start");
55 if (type_ != DataAbilityOperation::TYPE_INSERT && type_ != DataAbilityOperation::TYPE_UPDATE &&
56 type_ != DataAbilityOperation::TYPE_ASSERT) {
57 HILOG_ERROR("DataAbilityOperationBuilder::WithValuesBucket only inserts, updates and assert can have values,"
58 " type=%{public}d", type_);
59 return nullptr;
60 }
61
62 valuesBucket_ = std::make_shared<NativeRdb::ValuesBucket>(*values);
63 HILOG_DEBUG("DataAbilityOperationBuilder::WithValuesBucket end");
64 return shared_from_this();
65 }
66
WithPredicates(std::shared_ptr<NativeRdb::DataAbilityPredicates> & predicates)67 std::shared_ptr<DataAbilityOperationBuilder> DataAbilityOperationBuilder::WithPredicates(
68 std::shared_ptr<NativeRdb::DataAbilityPredicates> &predicates)
69 {
70 HILOG_DEBUG("DataAbilityOperationBuilder::WithPredicates start");
71 if (type_ != DataAbilityOperation::TYPE_DELETE && type_ != DataAbilityOperation::TYPE_UPDATE &&
72 type_ != DataAbilityOperation::TYPE_ASSERT) {
73 HILOG_ERROR(
74 "DataAbilityOperationBuilder::withPredicates only deletes, updates and assert can have selections,"
75 " type=%{public}d", type_);
76 return nullptr;
77 }
78 dataAbilityPredicates_ = predicates;
79 HILOG_DEBUG("DataAbilityOperationBuilder::WithPredicates end");
80 return shared_from_this();
81 }
WithExpectedCount(int count)82 std::shared_ptr<DataAbilityOperationBuilder> DataAbilityOperationBuilder::WithExpectedCount(int count)
83 {
84 HILOG_DEBUG("DataAbilityOperationBuilder::WithExpectedCount start");
85 HILOG_INFO("DataAbilityOperationBuilder::WithExpectedCount expectedCount:%{public}d", count);
86 if (type_ != DataAbilityOperation::TYPE_UPDATE && type_ != DataAbilityOperation::TYPE_DELETE &&
87 type_ != DataAbilityOperation::TYPE_ASSERT) {
88 HILOG_ERROR("DataAbilityOperationBuilder::withExpectedCount only updates, deletes and assert "
89 "can have expected counts, "
90 "type=%{public}d", type_);
91 return nullptr;
92 }
93 expectedCount_ = count;
94 HILOG_DEBUG("DataAbilityOperationBuilder::WithExpectedCount end");
95 return shared_from_this();
96 }
WithPredicatesBackReference(int requestArgIndex,int previousResult)97 std::shared_ptr<DataAbilityOperationBuilder> DataAbilityOperationBuilder::WithPredicatesBackReference(
98 int requestArgIndex, int previousResult)
99 {
100 HILOG_DEBUG("DataAbilityOperationBuilder::WithPredicatesBackReference start");
101 HILOG_INFO("DataAbilityOperationBuilder::WithPredicatesBackReference requestArgIndex:%{public}d, "
102 "previousResult:%{public}d", requestArgIndex, previousResult);
103 if (type_ != DataAbilityOperation::TYPE_UPDATE && type_ != DataAbilityOperation::TYPE_DELETE &&
104 type_ != DataAbilityOperation::TYPE_ASSERT) {
105 HILOG_ERROR("DataAbilityOperationBuilder::withPredicatesBackReference only updates, deletes, "
106 "and asserts can have select back-references, type=%{public}d", type_);
107 return nullptr;
108 }
109 dataAbilityPredicatesBackReferences_.insert(std::make_pair(requestArgIndex, previousResult));
110 HILOG_DEBUG("DataAbilityOperationBuilder::WithPredicatesBackReference end");
111 return shared_from_this();
112 }
WithValueBackReferences(std::shared_ptr<NativeRdb::ValuesBucket> & backReferences)113 std::shared_ptr<DataAbilityOperationBuilder> DataAbilityOperationBuilder::WithValueBackReferences(
114 std::shared_ptr<NativeRdb::ValuesBucket> &backReferences)
115 {
116 HILOG_DEBUG("DataAbilityOperationBuilder::WithValueBackReferences start");
117 if (type_ != DataAbilityOperation::TYPE_INSERT && type_ != DataAbilityOperation::TYPE_UPDATE &&
118 type_ != DataAbilityOperation::TYPE_ASSERT) {
119 HILOG_ERROR("DataAbilityOperationBuilder::withValueBackReferences only inserts, updates, and asserts can have "
120 "value back-references, type=%{public}d", type_);
121 return nullptr;
122 }
123 valuesBucketReferences_ = backReferences;
124 HILOG_DEBUG("DataAbilityOperationBuilder::WithValueBackReferences end");
125 return shared_from_this();
126 }
WithInterruptionAllowed(bool interrupted)127 std::shared_ptr<DataAbilityOperationBuilder> DataAbilityOperationBuilder::WithInterruptionAllowed(bool interrupted)
128 {
129 HILOG_DEBUG("DataAbilityOperationBuilder::WithInterruptionAllowed start");
130 HILOG_INFO("DataAbilityOperationBuilder::WithInterruptionAllowed interrupted=%{public}d", interrupted);
131 if (type_ != DataAbilityOperation::TYPE_INSERT && type_ != DataAbilityOperation::TYPE_UPDATE &&
132 type_ != DataAbilityOperation::TYPE_ASSERT && type_ != DataAbilityOperation::TYPE_DELETE) {
133 HILOG_ERROR("DataAbilityOperationBuilder::withInterruptionAllowed only inserts, updates, delete, "
134 "and asserts can have value back-references, type=%{public}d", type_);
135 return nullptr;
136 }
137 interrupted_ = interrupted;
138 HILOG_DEBUG("DataAbilityOperationBuilder::WithInterruptionAllowed end");
139 return shared_from_this();
140 }
141 } // namespace AppExecFwk
142 } // namespace OHOS
143