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 "disk_io_table.h"
17
18 namespace SysTuning {
19 namespace TraceStreamer {
20 namespace {
21 enum Index {
22 TS = 0,
23 DUR,
24 RD,
25 WR,
26 RD_SPEED,
27 WR_SPEED,
28 RD_COUNT,
29 WR_COUNT,
30 RD_COUNT_SPEED,
31 WR_COUNT_SPEED,
32 };
33 }
DiskIOTable(const TraceDataCache * dataCache)34 DiskIOTable::DiskIOTable(const TraceDataCache* dataCache) : TableBase(dataCache)
35 {
36 tableColumn_.push_back(TableBase::ColumnInfo("ts", "INTEGER"));
37 tableColumn_.push_back(TableBase::ColumnInfo("dur", "INTEGER"));
38 tableColumn_.push_back(TableBase::ColumnInfo("rd", "INTEGER"));
39 tableColumn_.push_back(TableBase::ColumnInfo("wr", "INTEGER"));
40 tableColumn_.push_back(TableBase::ColumnInfo("rd_speed", "REAL"));
41 tableColumn_.push_back(TableBase::ColumnInfo("wr_speed", "REAL"));
42 tableColumn_.push_back(TableBase::ColumnInfo("rd_count", "INTEGER"));
43 tableColumn_.push_back(TableBase::ColumnInfo("wr_count", "INTEGER"));
44 tableColumn_.push_back(TableBase::ColumnInfo("rd_count_speed", "REAL"));
45 tableColumn_.push_back(TableBase::ColumnInfo("wr_count_speed", "REAL"));
46 tablePriKey_.push_back("ts");
47 }
48
~DiskIOTable()49 DiskIOTable::~DiskIOTable() {}
50
CreateCursor()51 std::unique_ptr<TableBase::Cursor> DiskIOTable::CreateCursor()
52 {
53 return std::make_unique<Cursor>(dataCache_, this);
54 }
55
Cursor(const TraceDataCache * dataCache,TableBase * table)56 DiskIOTable::Cursor::Cursor(const TraceDataCache* dataCache, TableBase* table)
57 : TableBase::Cursor(dataCache, table, static_cast<uint32_t>(dataCache->GetConstDiskIOData().Size())),
58 diskIODataObj_(dataCache->GetConstDiskIOData())
59 {
60 }
61
~Cursor()62 DiskIOTable::Cursor::~Cursor() {}
63
Column(int column) const64 int DiskIOTable::Cursor::Column(int column) const
65 {
66 switch (column) {
67 case TS: {
68 sqlite3_result_int64(context_, static_cast<int64_t>(diskIODataObj_.TimeStamData()[CurrentRow()]));
69 break;
70 }
71 case DUR: {
72 sqlite3_result_int64(context_, static_cast<int64_t>(diskIODataObj_.Durs()[CurrentRow()]));
73 break;
74 }
75 case RD: {
76 sqlite3_result_int64(context_, static_cast<int64_t>(diskIODataObj_.RdDatas()[CurrentRow()]));
77 break;
78 }
79 case WR: {
80 sqlite3_result_int64(context_, static_cast<int64_t>(diskIODataObj_.WrDatas()[CurrentRow()]));
81 break;
82 }
83 case RD_SPEED: {
84 sqlite3_result_double(context_, static_cast<double>(diskIODataObj_.RdSpeedDatas()[CurrentRow()]));
85 break;
86 }
87 case WR_SPEED: {
88 sqlite3_result_double(context_, static_cast<double>(diskIODataObj_.WrSpeedDatas()[CurrentRow()]));
89 break;
90 }
91 case RD_COUNT: {
92 sqlite3_result_int64(context_, static_cast<int64_t>(diskIODataObj_.RdCountDatas()[CurrentRow()]));
93 break;
94 }
95 case WR_COUNT: {
96 sqlite3_result_int64(context_, static_cast<int64_t>(diskIODataObj_.WrCountDatas()[CurrentRow()]));
97 break;
98 }
99 case RD_COUNT_SPEED: {
100 sqlite3_result_double(context_, static_cast<double>(diskIODataObj_.RdCountPerSecDatas()[CurrentRow()]));
101 break;
102 }
103 case WR_COUNT_SPEED: {
104 sqlite3_result_double(context_, static_cast<double>(diskIODataObj_.WrCountPerSecDatas()[CurrentRow()]));
105 break;
106 }
107 default:
108 TS_LOGF("Unregistered column : %d", column);
109 break;
110 }
111 return SQLITE_OK;
112 }
113 } // namespace TraceStreamer
114 } // namespace SysTuning
115