1 /*
2 * Copyright (c) Huawei Technologies Co., Ltd. 2023. All rights reserved.
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 class Index : int32_t {
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 TASK_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("task_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 (static_cast<Index>(column)) {
69 case Index::ID:
70 sqlite3_result_int64(context_, static_cast<sqlite3_int64>(CurrentRow()));
71 break;
72 case Index::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 Index::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 Index::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 Index::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 Index::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 default:
104 HandleTypeColumns(column);
105 }
106 return SQLITE_OK;
107 }
HandleTypeColumns(int32_t taskPoolTabColumn) const108 void TaskPoolTable::Cursor::HandleTypeColumns(int32_t taskPoolTabColumn) const
109 {
110 switch (static_cast<Index>(taskPoolTabColumn)) {
111 case Index::RETURN_ITID:
112 if (taskPoolObj_.ReturnItids()[CurrentRow()] != INVALID_INT32) {
113 sqlite3_result_int64(context_, static_cast<sqlite3_int64>(
114 dataCache_->GetConstTaskPoolData().ReturnItids()[CurrentRow()]));
115 }
116 break;
117 case Index::TASK_ID:
118 if (taskPoolObj_.TaskIds()[CurrentRow()] != INVALID_INT64) {
119 sqlite3_result_int64(
120 context_, static_cast<sqlite3_int64>(dataCache_->GetConstTaskPoolData().TaskIds()[CurrentRow()]));
121 }
122 break;
123 case Index::PRIORITY:
124 if (taskPoolObj_.Prioritys()[CurrentRow()] != INVALID_INT32) {
125 sqlite3_result_int64(
126 context_, static_cast<sqlite3_int64>(dataCache_->GetConstTaskPoolData().Prioritys()[CurrentRow()]));
127 }
128 break;
129 case Index::EXECUTE_STATE:
130 if (taskPoolObj_.ExecuteStates()[CurrentRow()] != INVALID_INT32) {
131 sqlite3_result_int64(context_, static_cast<sqlite3_int64>(
132 dataCache_->GetConstTaskPoolData().ExecuteStates()[CurrentRow()]));
133 }
134 break;
135 case Index::RETURN_STATE:
136 if (taskPoolObj_.ReturnStates()[CurrentRow()] != INVALID_INT32) {
137 sqlite3_result_int64(context_, static_cast<sqlite3_int64>(
138 dataCache_->GetConstTaskPoolData().ReturnStates()[CurrentRow()]));
139 }
140 break;
141 case Index::TIMEOUT_ROW:
142 if (taskPoolObj_.TimeoutRows()[CurrentRow()] != INVALID_INT32) {
143 sqlite3_result_int64(context_, static_cast<sqlite3_int64>(
144 dataCache_->GetConstTaskPoolData().TimeoutRows()[CurrentRow()]));
145 }
146 break;
147 default:
148 TS_LOGF("Unregistered taskPoolTabColumn : %d", taskPoolTabColumn);
149 break;
150 }
151 }
152 } // namespace TraceStreamer
153 } // namespace SysTuning
154