1 /*
2 * Copyright (c) 2021 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 "values_bucket.h"
17 #include "sqlite_global_config.h"
18 #include "logger.h"
19
20 namespace OHOS {
21 namespace NativeRdb {
ValuesBucket()22 ValuesBucket::ValuesBucket()
23 {
24 }
25
ValuesBucket(std::map<std::string,ValueObject> & valuesMap)26 ValuesBucket::ValuesBucket(std::map<std::string, ValueObject> &valuesMap) : valuesMap(valuesMap)
27 {
28 }
29
~ValuesBucket()30 ValuesBucket::~ValuesBucket()
31 {
32 }
33
PutString(const std::string & columnName,const std::string & value)34 void ValuesBucket::PutString(const std::string &columnName, const std::string &value)
35 {
36 valuesMap.insert(std::make_pair(columnName, ValueObject(value)));
37 }
38
PutInt(const std::string & columnName,int value)39 void ValuesBucket::PutInt(const std::string &columnName, int value)
40 {
41 valuesMap.insert(std::make_pair(columnName, ValueObject(value)));
42 }
43
PutLong(const std::string & columnName,int64_t value)44 void ValuesBucket::PutLong(const std::string &columnName, int64_t value)
45 {
46 valuesMap.insert(std::make_pair(columnName, ValueObject(value)));
47 }
48
PutDouble(const std::string & columnName,double value)49 void ValuesBucket::PutDouble(const std::string &columnName, double value)
50 {
51 valuesMap.insert(std::make_pair(columnName, ValueObject(value)));
52 }
53
PutBool(const std::string & columnName,bool value)54 void ValuesBucket::PutBool(const std::string &columnName, bool value)
55 {
56 valuesMap.insert(std::make_pair(columnName, ValueObject(value)));
57 }
58
PutBlob(const std::string & columnName,const std::vector<uint8_t> & value)59 void ValuesBucket::PutBlob(const std::string &columnName, const std::vector<uint8_t> &value)
60 {
61 valuesMap.insert(std::make_pair(columnName, ValueObject(value)));
62 }
63
PutNull(const std::string & columnName)64 void ValuesBucket::PutNull(const std::string &columnName)
65 {
66 valuesMap.insert(std::make_pair(columnName, ValueObject()));
67 }
68
Delete(const std::string & columnName)69 void ValuesBucket::Delete(const std::string &columnName)
70 {
71 valuesMap.erase(columnName);
72 }
73
Clear()74 void ValuesBucket::Clear()
75 {
76 valuesMap.clear();
77 }
78
Size() const79 int ValuesBucket::Size() const
80 {
81 return valuesMap.size();
82 }
83
IsEmpty() const84 bool ValuesBucket::IsEmpty() const
85 {
86 return valuesMap.empty();
87 }
88
HasColumn(const std::string & columnName) const89 bool ValuesBucket::HasColumn(const std::string &columnName) const
90 {
91 auto iter = valuesMap.find(columnName);
92 if (iter == valuesMap.end()) {
93 return false;
94 }
95 return true;
96 }
97
GetObject(const std::string & columnName,ValueObject & value) const98 bool ValuesBucket::GetObject(const std::string &columnName, ValueObject &value) const
99 {
100 auto iter = valuesMap.find(columnName);
101 if (iter == valuesMap.end()) {
102 return false;
103 }
104 value = iter->second;
105 return true;
106 }
107
GetAll(std::map<std::string,ValueObject> & outValuesMap) const108 void ValuesBucket::GetAll(std::map<std::string, ValueObject> &outValuesMap) const
109 {
110 outValuesMap = valuesMap;
111 }
112
113 #if !defined(WINDOWS_PLATFORM) && !defined(MAC_PLATFORM)
Marshalling(Parcel & parcel) const114 bool ValuesBucket::Marshalling(Parcel &parcel) const
115 {
116 parcel.WriteInt32(valuesMap.size());
117 for (auto &it : valuesMap) {
118 parcel.WriteString(it.first);
119 parcel.WriteParcelable(&it.second);
120 }
121 return true;
122 }
123
Unmarshalling(Parcel & parcel)124 ValuesBucket *ValuesBucket::Unmarshalling(Parcel &parcel)
125 {
126 int mapSize = parcel.ReadInt32();
127 std::map<std::string, ValueObject> valuesMap;
128 ValueObject *value = nullptr;
129 if (mapSize > GlobalExpr::SQLITE_MAX_COLUMN) {
130 LOG_ERROR("The mapSize is %{public}d ", mapSize);
131 return new ValuesBucket(valuesMap);
132 }
133 for (int i = 0; i < mapSize; i++) {
134 std::string key = parcel.ReadString();
135 value = parcel.ReadParcelable<ValueObject>();
136 valuesMap.insert(std::make_pair(key, *value));
137 delete value;
138 }
139 value = nullptr;
140 return new ValuesBucket(valuesMap);
141 }
142 #endif
143 } // namespace NativeRdb
144 } // namespace OHOS
145