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 "task_pool_table.h"
17
18 namespace SysTuning {
19 namespace TraceStreamer {
20 enum Index {
21 ID = 0,
22 ALLOCATION_TASK_ROW,
23 EXECUTE_TASK_ROW,
24 RETURN_TASK_ROW,
25 ALLOCATION_ITID,
26 EXECUTE_ITID,
27 RETURN_ITID,
28 EXECUTE_ID,
29 PRIORITY,
30 EXECUTE_STATE,
31 RETURN_STATE,
32 TIMEOUT_ROW
33 };
TaskPoolTable(const TraceDataCache * dataCache)34 TaskPoolTable::TaskPoolTable(const TraceDataCache* dataCache) : TableBase(dataCache)
35 {
36 tableColumn_.push_back(TableBase::ColumnInfo("id", "INTEGER"));
37 tableColumn_.push_back(TableBase::ColumnInfo("allocation_task_row", "INTEGER"));
38 tableColumn_.push_back(TableBase::ColumnInfo("execute_task_row", "INTEGER"));
39 tableColumn_.push_back(TableBase::ColumnInfo("return_task_row", "INTEGER"));
40 tableColumn_.push_back(TableBase::ColumnInfo("allocation_itid", "INTEGER"));
41 tableColumn_.push_back(TableBase::ColumnInfo("execute_itid", "INTEGER"));
42 tableColumn_.push_back(TableBase::ColumnInfo("return_itid", "INTEGER"));
43 tableColumn_.push_back(TableBase::ColumnInfo("execute_id", "INTEGER"));
44 tableColumn_.push_back(TableBase::ColumnInfo("priority", "INTEGER"));
45 tableColumn_.push_back(TableBase::ColumnInfo("execute_state", "INTEGER"));
46 tableColumn_.push_back(TableBase::ColumnInfo("return_state", "INTEGER"));
47 tableColumn_.push_back(TableBase::ColumnInfo("timeout_row", "INTEGER"));
48 tablePriKey_.push_back("id");
49 }
50
~TaskPoolTable()51 TaskPoolTable::~TaskPoolTable() {}
52
CreateCursor()53 std::unique_ptr<TableBase::Cursor> TaskPoolTable::CreateCursor()
54 {
55 return std::make_unique<Cursor>(dataCache_, this);
56 }
57
Cursor(const TraceDataCache * dataCache,TableBase * table)58 TaskPoolTable::Cursor::Cursor(const TraceDataCache* dataCache, TableBase* table)
59 : TableBase::Cursor(dataCache, table, static_cast<uint32_t>(dataCache->GetConstTaskPoolData().Size())),
60 taskPoolObj_(dataCache->GetConstTaskPoolData())
61 {
62 }
63
~Cursor()64 TaskPoolTable::Cursor::~Cursor() {}
65
Column(int32_t column) const66 int32_t TaskPoolTable::Cursor::Column(int32_t column) const
67 {
68 switch (column) {
69 case ID:
70 sqlite3_result_int64(context_, static_cast<sqlite3_int64>(CurrentRow()));
71 break;
72 case ALLOCATION_TASK_ROW:
73 if (taskPoolObj_.AllocationTaskRows()[CurrentRow()] != INVALID_INT32) {
74 sqlite3_result_int64(
75 context_,
76 static_cast<sqlite3_int64>(dataCache_->GetConstTaskPoolData().AllocationTaskRows()[CurrentRow()]));
77 }
78 break;
79 case EXECUTE_TASK_ROW:
80 if (taskPoolObj_.ExecuteTaskRows()[CurrentRow()] != INVALID_INT32) {
81 sqlite3_result_int64(context_, static_cast<sqlite3_int64>(
82 dataCache_->GetConstTaskPoolData().ExecuteTaskRows()[CurrentRow()]));
83 }
84 break;
85 case RETURN_TASK_ROW:
86 if (taskPoolObj_.ReturnTaskRows()[CurrentRow()] != INVALID_INT32) {
87 sqlite3_result_int64(context_, static_cast<sqlite3_int64>(
88 dataCache_->GetConstTaskPoolData().ReturnTaskRows()[CurrentRow()]));
89 }
90 break;
91 case ALLOCATION_ITID:
92 if (taskPoolObj_.AllocationItids()[CurrentRow()] != INVALID_INT32) {
93 sqlite3_result_int64(context_, static_cast<sqlite3_int64>(
94 dataCache_->GetConstTaskPoolData().AllocationItids()[CurrentRow()]));
95 }
96 break;
97 case EXECUTE_ITID:
98 if (taskPoolObj_.ExecuteItids()[CurrentRow()] != INVALID_INT32) {
99 sqlite3_result_int64(context_, static_cast<sqlite3_int64>(
100 dataCache_->GetConstTaskPoolData().ExecuteItids()[CurrentRow()]));
101 }
102 break;
103 case RETURN_ITID:
104 if (taskPoolObj_.ReturnItids()[CurrentRow()] != INVALID_INT32) {
105 sqlite3_result_int64(context_, static_cast<sqlite3_int64>(
106 dataCache_->GetConstTaskPoolData().ReturnItids()[CurrentRow()]));
107 }
108 break;
109 case EXECUTE_ID:
110 if (taskPoolObj_.ExecuteIds()[CurrentRow()] != INVALID_INT32) {
111 sqlite3_result_int64(context_, static_cast<sqlite3_int64>(
112 dataCache_->GetConstTaskPoolData().ExecuteIds()[CurrentRow()]));
113 }
114 break;
115 case PRIORITY:
116 if (taskPoolObj_.Prioritys()[CurrentRow()] != INVALID_INT32) {
117 sqlite3_result_int64(
118 context_, static_cast<sqlite3_int64>(dataCache_->GetConstTaskPoolData().Prioritys()[CurrentRow()]));
119 }
120 break;
121 case EXECUTE_STATE:
122 if (taskPoolObj_.ExecuteStates()[CurrentRow()] != INVALID_INT32) {
123 sqlite3_result_int64(context_, static_cast<sqlite3_int64>(
124 dataCache_->GetConstTaskPoolData().ExecuteStates()[CurrentRow()]));
125 }
126 break;
127 case RETURN_STATE:
128 if (taskPoolObj_.ReturnStates()[CurrentRow()] != INVALID_INT32) {
129 sqlite3_result_int64(context_, static_cast<sqlite3_int64>(
130 dataCache_->GetConstTaskPoolData().ReturnStates()[CurrentRow()]));
131 }
132 break;
133 case TIMEOUT_ROW:
134 if (taskPoolObj_.TimeoutRows()[CurrentRow()] != INVALID_INT32) {
135 sqlite3_result_int64(context_, static_cast<sqlite3_int64>(
136 dataCache_->GetConstTaskPoolData().TimeoutRows()[CurrentRow()]));
137 }
138 break;
139 default:
140 TS_LOGF("Unregistered column : %d", column);
141 break;
142 }
143 return SQLITE_OK;
144 }
145 } // namespace TraceStreamer
146 } // namespace SysTuning
147