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