1 /*
2 * Copyright (c) 2023-2024 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_dao.h"
16
17 #include "app_event_cache_common.h"
18 #include "app_event_store.h"
19 #include "hiappevent_base.h"
20 #include "hilog/log.h"
21 #include "rdb_helper.h"
22 #include "sql_util.h"
23
24 #undef LOG_DOMAIN
25 #define LOG_DOMAIN 0xD002D07
26
27 #undef LOG_TAG
28 #define LOG_TAG "EventDao"
29
30 namespace OHOS {
31 namespace HiviewDFX {
32 using namespace AppEventCacheCommon;
33
AppEventDao(std::shared_ptr<NativeRdb::RdbStore> dbStore)34 AppEventDao::AppEventDao(std::shared_ptr<NativeRdb::RdbStore> dbStore) : dbStore_(dbStore)
35 {
36 if (Create() != DB_SUCC) {
37 HILOG_ERROR(LOG_CORE, "failed to create table=%{public}s", Events::TABLE.c_str());
38 }
39 }
40
Create()41 int AppEventDao::Create()
42 {
43 /**
44 * table: events
45 *
46 * |-------|--------|------|------|------|-----|-----|----------|---------|----------|------------|--------|
47 * ------------|
48 * | seq | domain | name | type | tz | pid | tid | trace_id | span_id | pspan_id | trace_flag | params |
49 * running_id |
50 * |-------|--------|------|------|------|-----|-----|----------|---------|----------|------------|--------|
51 * ------------|
52 * | INT64 | TEXT | TEXT | INT | TEXT | INT | INT | INT64 | INT64 | INT64 | INT | TEXT |
53 * TEXT |
54 * |-------|--------|------|------|------|-----|-----|----------|---------|----------|------------|--------|
55 * ------------|
56 */
57 const std::vector<std::pair<std::string, std::string>> fields = {
58 {Events::FIELD_DOMAIN, SqlUtil::SQL_TEXT_TYPE},
59 {Events::FIELD_NAME, SqlUtil::SQL_TEXT_TYPE},
60 {Events::FIELD_TYPE, SqlUtil::SQL_INT_TYPE},
61 {Events::FIELD_TIME, SqlUtil::SQL_INT_TYPE},
62 {Events::FIELD_TZ, SqlUtil::SQL_TEXT_TYPE},
63 {Events::FIELD_PID, SqlUtil::SQL_INT_TYPE},
64 {Events::FIELD_TID, SqlUtil::SQL_INT_TYPE},
65 {Events::FIELD_TRACE_ID, SqlUtil::SQL_INT_TYPE},
66 {Events::FIELD_SPAN_ID, SqlUtil::SQL_INT_TYPE},
67 {Events::FIELD_PSPAN_ID, SqlUtil::SQL_INT_TYPE},
68 {Events::FIELD_TRACE_FLAG, SqlUtil::SQL_INT_TYPE},
69 {Events::FIELD_PARAMS, SqlUtil::SQL_TEXT_TYPE},
70 {Events::FIELD_RUNNING_ID, SqlUtil::SQL_TEXT_TYPE},
71 };
72 std::string sql = SqlUtil::CreateTable(Events::TABLE, fields);
73 if (dbStore_->ExecuteSql(sql) != NativeRdb::E_OK) {
74 return DB_FAILED;
75 }
76 return DB_SUCC;
77 }
78
Insert(std::shared_ptr<AppEventPack> event)79 int64_t AppEventDao::Insert(std::shared_ptr<AppEventPack> event)
80 {
81 NativeRdb::ValuesBucket bucket;
82 bucket.PutString(Events::FIELD_DOMAIN, event->GetDomain());
83 bucket.PutString(Events::FIELD_NAME, event->GetName());
84 bucket.PutInt(Events::FIELD_TYPE, event->GetType());
85 bucket.PutLong(Events::FIELD_TIME, event->GetTime());
86 bucket.PutString(Events::FIELD_TZ, event->GetTimeZone());
87 bucket.PutInt(Events::FIELD_PID, event->GetPid());
88 bucket.PutInt(Events::FIELD_TID, event->GetTid());
89 bucket.PutLong(Events::FIELD_TRACE_ID, event->GetTraceId());
90 bucket.PutLong(Events::FIELD_SPAN_ID, event->GetSpanId());
91 bucket.PutLong(Events::FIELD_PSPAN_ID, event->GetPspanId());
92 bucket.PutInt(Events::FIELD_TRACE_FLAG, event->GetTraceFlag());
93 bucket.PutString(Events::FIELD_PARAMS, event->GetParamStr());
94 bucket.PutString(Events::FIELD_RUNNING_ID, event->GetRunningId());
95 int64_t seq = 0;
96 if (dbStore_->Insert(seq, Events::TABLE, bucket) != NativeRdb::E_OK) {
97 return DB_FAILED;
98 }
99 return seq;
100 }
101
Delete(int64_t eventSeq)102 int AppEventDao::Delete(int64_t eventSeq)
103 {
104 NativeRdb::AbsRdbPredicates predicates(Events::TABLE);
105 if (eventSeq > 0) {
106 predicates.EqualTo(Events::FIELD_SEQ, eventSeq);
107 }
108 int deleteRows = 0;
109 if (dbStore_->Delete(deleteRows, predicates) != NativeRdb::E_OK) {
110 return DB_FAILED;
111 }
112 HILOG_INFO(LOG_CORE, "delete %{public}d records, eventSeq=%{public}" PRId64, deleteRows, eventSeq);
113 return deleteRows;
114 }
115
Delete(const std::vector<int64_t> & eventSeqs)116 int AppEventDao::Delete(const std::vector<int64_t>& eventSeqs)
117 {
118 if (eventSeqs.empty()) {
119 return DB_SUCC;
120 }
121 NativeRdb::AbsRdbPredicates predicates(Events::TABLE);
122 std::vector<std::string> eventSeqStrs(eventSeqs.size());
123 std::transform(eventSeqs.begin(), eventSeqs.end(), eventSeqStrs.begin(), [](int64_t eventSeq) {
124 return std::to_string(eventSeq);
125 });
126 predicates.In(Events::FIELD_SEQ, eventSeqStrs);
127
128 int deleteRows = 0;
129 if (dbStore_->Delete(deleteRows, predicates) != NativeRdb::E_OK) {
130 return DB_FAILED;
131 }
132 HILOG_INFO(LOG_CORE, "delete %{public}d records", deleteRows);
133 return deleteRows;
134 }
135 } // namespace HiviewDFX
136 } // namespace OHOS
137