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 #include "app_event_observer_dao.h"
16
17 #include "app_event_cache_common.h"
18 #include "app_event_store.h"
19 #include "hilog/log.h"
20 #include "rdb_helper.h"
21 #include "sql_util.h"
22
23 namespace OHOS {
24 namespace HiviewDFX {
25 using namespace AppEventCacheCommon;
26 namespace {
27 const HiLogLabel LABEL = { LOG_CORE, HIAPPEVENT_DOMAIN, "HiAppEvent_AppEventObserverDao" };
28 }
29
AppEventObserverDao(std::shared_ptr<NativeRdb::RdbStore> dbStore)30 AppEventObserverDao::AppEventObserverDao(std::shared_ptr<NativeRdb::RdbStore> dbStore) : dbStore_(dbStore)
31 {
32 if (Create() != DB_SUCC) {
33 HiLog::Error(LABEL, "failed to create table=%{public}s", Observers::TABLE.c_str());
34 }
35 }
36
Create()37 int AppEventObserverDao::Create()
38 {
39 /**
40 * table: observers
41 *
42 * |-------|------|------|
43 * | seq | name | hash |
44 * |-------|------|------|
45 * | INT64 | TEXT | INT64|
46 * |-------|------|------|
47 */
48 const std::vector<std::pair<std::string, std::string>> fields = {
49 {Observers::FIELD_NAME, SqlUtil::SQL_TEXT_TYPE},
50 {Observers::FIELD_HASH, SqlUtil::SQL_INT_TYPE},
51 };
52 std::string sql = SqlUtil::CreateTable(Observers::TABLE, fields);
53 if (dbStore_->ExecuteSql(sql) != NativeRdb::E_OK) {
54 return DB_FAILED;
55 }
56 return DB_SUCC;
57 }
58
Insert(const std::string & observer,int64_t hashCode)59 int64_t AppEventObserverDao::Insert(const std::string& observer, int64_t hashCode)
60 {
61 NativeRdb::ValuesBucket bucket;
62 bucket.PutString(Observers::FIELD_NAME, observer);
63 bucket.PutLong(Observers::FIELD_HASH, hashCode);
64 int64_t seq = 0;
65 if (dbStore_->Insert(seq, Observers::TABLE, bucket) != NativeRdb::E_OK) {
66 return DB_FAILED;
67 }
68 return seq;
69 }
70
QuerySeq(const std::string & observer,int64_t hashCode)71 int64_t AppEventObserverDao::QuerySeq(const std::string& observer, int64_t hashCode)
72 {
73 NativeRdb::AbsRdbPredicates predicates(Observers::TABLE);
74 predicates.EqualTo(Observers::FIELD_NAME, observer);
75 predicates.EqualTo(Observers::FIELD_HASH, hashCode);
76 auto resultSet = dbStore_->Query(predicates, {Observers::FIELD_SEQ});
77 if (resultSet == nullptr) {
78 HiLog::Error(LABEL, "failed to query table, observer name=%{public}s, hash code=%{public}" PRId64,
79 observer.c_str(), hashCode);
80 return DB_FAILED;
81 }
82
83 // the hash code is unique, so get only the first
84 int64_t observerSeq = 0;
85 if (resultSet->GoToNextRow() == NativeRdb::E_OK && resultSet->GetLong(0, observerSeq) == NativeRdb::E_OK) {
86 HiLog::Info(LABEL, "succ to query observer seq=%{public}" PRId64 ", name=%{public}s, hash=%{public}" PRId64,
87 observerSeq, observer.c_str(), hashCode);
88 resultSet->Close();
89 return observerSeq;
90 }
91 resultSet->Close();
92 return DB_FAILED;
93 }
94
QuerySeqs(const std::string & observer,std::vector<int64_t> & observerSeqs)95 int AppEventObserverDao::QuerySeqs(const std::string& observer, std::vector<int64_t>& observerSeqs)
96 {
97 NativeRdb::AbsRdbPredicates predicates(Observers::TABLE);
98 predicates.EqualTo(Observers::FIELD_NAME, observer);
99 auto resultSet = dbStore_->Query(predicates, {Observers::FIELD_SEQ});
100 if (resultSet == nullptr) {
101 HiLog::Error(LABEL, "failed to query table, observer=%{public}s", observer.c_str());
102 return DB_FAILED;
103 }
104 while (resultSet->GoToNextRow() == NativeRdb::E_OK) {
105 int64_t observerSeq = 0;
106 if (resultSet->GetLong(0, observerSeq) != NativeRdb::E_OK) {
107 HiLog::Error(LABEL, "failed to get seq value from resultSet, observer=%{public}s", observer.c_str());
108 continue;
109 }
110 observerSeqs.emplace_back(observerSeq);
111 }
112 resultSet->Close();
113 return DB_SUCC;
114 }
115
Delete(const std::string & observer)116 int AppEventObserverDao::Delete(const std::string& observer)
117 {
118 int deleteRows = 0;
119 NativeRdb::AbsRdbPredicates predicates(Observers::TABLE);
120 predicates.EqualTo(Observers::FIELD_NAME, observer);
121 if (dbStore_->Delete(deleteRows, predicates) != NativeRdb::E_OK) {
122 HiLog::Error(LABEL, "failed to delete records, observer=%{public}s", observer.c_str());
123 return DB_FAILED;
124 }
125 HiLog::Info(LABEL, "delete %{public}d records, observer=%{public}s", deleteRows, observer.c_str());
126 return deleteRows;
127 }
128
Delete(int64_t observerSeq)129 int AppEventObserverDao::Delete(int64_t observerSeq)
130 {
131 int deleteRows = 0;
132 NativeRdb::AbsRdbPredicates predicates(Observers::TABLE);
133 predicates.EqualTo(Observers::FIELD_SEQ, observerSeq);
134 if (dbStore_->Delete(deleteRows, predicates) != NativeRdb::E_OK) {
135 HiLog::Error(LABEL, "failed to delete records, observer seq=%{public}" PRId64, observerSeq);
136 return DB_FAILED;
137 }
138 HiLog::Info(LABEL, "delete %{public}d records, observerSeq=%{public}" PRId64, deleteRows, observerSeq);
139 return deleteRows;
140 }
141 } // namespace HiviewDFX
142 } // namespace OHOS
143