1 /*
2 * Copyright (c) 2025 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 "trace_behavior_storage.h"
17
18 #include "hiview_logger.h"
19 #include "rdb_store.h"
20 #include "sql_util.h"
21
22 namespace OHOS {
23 namespace HiviewDFX {
24 DEFINE_LOG_TAG("TraceBehaviorDbHelper");
25 namespace {
26 constexpr char DB_PATH[] = "/data/log/hiview/unified_collection/trace/";
27 constexpr char DB_NAME[] = "trace_flow_control";
28 constexpr int32_t DB_VERSION = 1;
29 constexpr char TABLE_NAME_BEHAVIOR[] = "trace_behavior_db_helper";
30 constexpr char COLUMN_ID[] = "id";
31 constexpr char COLUMN_BEHAVIOR_ID[] = "behavior_id ";
32 constexpr char COLUMN_DATE[] = "task_date";
33 constexpr char COLUMN_USED_QUOTA[] = "used_quota";
34
InnerGetBucket(const BehaviorRecord & behaviorRecord)35 NativeRdb::ValuesBucket InnerGetBucket(const BehaviorRecord &behaviorRecord)
36 {
37 NativeRdb::ValuesBucket bucket;
38 bucket.PutInt(COLUMN_BEHAVIOR_ID, behaviorRecord.behaviorId);
39 bucket.PutString(COLUMN_DATE, behaviorRecord.dateNum);
40 bucket.PutInt(COLUMN_USED_QUOTA, behaviorRecord.usedQuota);
41 return bucket;
42 }
43 }
44
GetRecord(BehaviorRecord & behaviorRecord)45 bool TraceBehaviorStorage::GetRecord(BehaviorRecord &behaviorRecord)
46 {
47 if (dbStore_ == nullptr) {
48 HIVIEW_LOGE("db store is null, path=%{public}s", DB_PATH);
49 return false;
50 }
51 NativeRdb::AbsRdbPredicates predicates(TABLE_NAME_BEHAVIOR);
52 predicates.EqualTo(COLUMN_BEHAVIOR_ID, behaviorRecord.behaviorId);
53 predicates.EqualTo(COLUMN_DATE, behaviorRecord.dateNum);
54 auto resultSet = dbStore_->Query(predicates, {COLUMN_USED_QUOTA});
55 if (resultSet == nullptr) {
56 HIVIEW_LOGE("failed to query from table %{public}s", TABLE_NAME_BEHAVIOR);
57 return false;
58 }
59
60 if (resultSet->GoToNextRow() == NativeRdb::E_OK) {
61 resultSet->GetInt(0, behaviorRecord.usedQuota); // 0 means used_quota field
62 } else {
63 HIVIEW_LOGW("fail to get record for date %{public}s, set usedQuota to 0", behaviorRecord.dateNum.c_str());
64 behaviorRecord.usedQuota = 0;
65 resultSet->Close();
66 return false;
67 }
68 resultSet->Close();
69 return true;
70 }
71
InsertRecord(BehaviorRecord & behaviorRecord)72 bool TraceBehaviorStorage::InsertRecord(BehaviorRecord &behaviorRecord)
73 {
74 if (dbStore_ == nullptr) {
75 HIVIEW_LOGE("db store is null, path=%{public}s", DB_PATH);
76 return false;
77 }
78 NativeRdb::ValuesBucket bucket;
79 bucket.PutInt(COLUMN_BEHAVIOR_ID, behaviorRecord.behaviorId);
80 bucket.PutString(COLUMN_DATE, behaviorRecord.dateNum);
81 bucket.PutInt(COLUMN_USED_QUOTA, behaviorRecord.usedQuota);
82 int64_t seq = 0;
83 if (dbStore_->Insert(seq, TABLE_NAME_BEHAVIOR, bucket) != NativeRdb::E_OK) {
84 HIVIEW_LOGW("failed to insert table");
85 return false;
86 }
87 return true;
88 }
89
UpdateRecord(BehaviorRecord & behaviorRecord)90 bool TraceBehaviorStorage::UpdateRecord(BehaviorRecord &behaviorRecord)
91 {
92 if (dbStore_ == nullptr) {
93 HIVIEW_LOGE("db store is null, path=%{public}s", DB_PATH);
94 return false;
95 }
96 HIVIEW_LOGD("Update usedQuota:%{public}d", behaviorRecord.usedQuota);
97 NativeRdb::ValuesBucket bucket = InnerGetBucket(behaviorRecord);
98 NativeRdb::AbsRdbPredicates predicates(TABLE_NAME_BEHAVIOR);
99 predicates.EqualTo(COLUMN_BEHAVIOR_ID, behaviorRecord.behaviorId);
100 predicates.EqualTo(COLUMN_DATE, behaviorRecord.dateNum);
101 int32_t changedRows = 0;
102 if (dbStore_->Update(changedRows, bucket, predicates) != NativeRdb::E_OK) {
103 HIVIEW_LOGW("failed to update table");
104 return false;
105 }
106 return true;
107 }
108 } // namespace HiviewDFX
109 } // namespace OHOS
110