/* * Copyright (c) Huawei Technologies Co., Ltd. 2023. All rights reserved. * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ #include "frame_slice_table.h" namespace SysTuning { namespace TraceStreamer { enum class Index : int32_t { ID = 0, TS, VSYNC, IPID, ITID, CALLSTACK_ID, DUR, SRC, DST, TYPE, TYPE_DESC, FLAG, DEPTH, FRAME_NO }; FrameSliceTable::FrameSliceTable(const TraceDataCache *dataCache) : TableBase(dataCache) { tableColumn_.push_back(TableBase::ColumnInfo("id", "INTEGER")); tableColumn_.push_back(TableBase::ColumnInfo("ts", "INTEGER")); tableColumn_.push_back(TableBase::ColumnInfo("vsync", "INTEGER")); tableColumn_.push_back(TableBase::ColumnInfo("ipid", "INTEGER")); tableColumn_.push_back(TableBase::ColumnInfo("itid", "INTEGER")); tableColumn_.push_back(TableBase::ColumnInfo("callstack_id", "INTEGER")); tableColumn_.push_back(TableBase::ColumnInfo("dur", "INTEGER")); tableColumn_.push_back(TableBase::ColumnInfo("src", "TEXT")); tableColumn_.push_back(TableBase::ColumnInfo("dst", "INTEGER")); tableColumn_.push_back(TableBase::ColumnInfo("type", "INTEGER")); tableColumn_.push_back(TableBase::ColumnInfo("type_desc", "TEXT")); tableColumn_.push_back(TableBase::ColumnInfo("flag", "INTEGER")); tableColumn_.push_back(TableBase::ColumnInfo("depth", "INTEGER")); tableColumn_.push_back(TableBase::ColumnInfo("frame_no", "INTEGER")); tablePriKey_.push_back("id"); } FrameSliceTable::~FrameSliceTable() {} void FrameSliceTable::FilterByConstraint(FilterConstraints &slicefc, double &slicefilterCost, size_t slicerowCount, uint32_t slicecurrenti) { // To use the EstimateFilterCost function in the TableBase parent class function to calculate the i-value of each // for loop const auto &slicec = slicefc.GetConstraints()[slicecurrenti]; switch (static_cast(slicec.col)) { case Index::ID: { if (CanFilterId(slicec.op, slicerowCount)) { slicefc.UpdateConstraint(slicecurrenti, true); slicefilterCost += 1; // id can position by 1 step } else { slicefilterCost += slicerowCount; // scan all rows } break; } default: // other column slicefilterCost += slicerowCount; // scan all rows break; } } std::unique_ptr FrameSliceTable::CreateCursor() { return std::make_unique(dataCache_, this); } FrameSliceTable::Cursor::Cursor(const TraceDataCache *dataCache, TableBase *table) : TableBase::Cursor(dataCache, table, static_cast(dataCache->GetConstFrameSliceData().Size())), frameSliceObj_(dataCache->GetConstFrameSliceData()) { } FrameSliceTable::Cursor::~Cursor() {} int32_t FrameSliceTable::Cursor::Filter(const FilterConstraints &fc, sqlite3_value **argv) { // reset indexMap_ indexMap_ = std::make_unique(0, rowCount_); if (rowCount_ <= 0) { return SQLITE_OK; } HandleIndex(fc, argv); auto frameSliceTabOrderbys = fc.GetOrderBys(); for (auto i = frameSliceTabOrderbys.size(); i > 0;) { i--; switch (static_cast(frameSliceTabOrderbys[i].iColumn)) { case Index::ID: indexMap_->SortBy(frameSliceTabOrderbys[i].desc); break; default: break; } } return SQLITE_OK; } void FrameSliceTable::Cursor::HandleIndex(const FilterConstraints &fc, sqlite3_value **argv) { auto &cs = fc.GetConstraints(); for (size_t i = 0; i < cs.size(); i++) { const auto &c = cs[i]; switch (static_cast(c.col)) { case Index::ID: FilterId(c.op, argv[i]); break; case Index::ITID: indexMap_->MixRange(c.op, static_cast(sqlite3_value_int(argv[i])), frameSliceObj_.InternalTidsData()); break; case Index::IPID: indexMap_->MixRange(c.op, static_cast(sqlite3_value_int(argv[i])), frameSliceObj_.Ipids()); break; case Index::VSYNC: indexMap_->MixRange(c.op, static_cast(sqlite3_value_int(argv[i])), frameSliceObj_.VsyncIds()); break; case Index::CALLSTACK_ID: indexMap_->MixRange(c.op, static_cast(sqlite3_value_int(argv[i])), frameSliceObj_.CallStackIds()); break; case Index::DUR: indexMap_->MixRange(c.op, static_cast(sqlite3_value_int(argv[i])), frameSliceObj_.Durs()); break; case Index::TYPE: indexMap_->MixRange(c.op, static_cast(sqlite3_value_int(argv[i])), frameSliceObj_.Types()); break; case Index::FLAG: indexMap_->MixRange(c.op, static_cast(sqlite3_value_int(argv[i])), frameSliceObj_.Flags()); break; case Index::DEPTH: indexMap_->MixRange(c.op, static_cast(sqlite3_value_int(argv[i])), frameSliceObj_.Depths()); break; case Index::FRAME_NO: indexMap_->MixRange(c.op, static_cast(sqlite3_value_int(argv[i])), frameSliceObj_.FrameNos()); break; default: break; } } } int32_t FrameSliceTable::Cursor::Column(int32_t column) const { switch (static_cast(column)) { case Index::ID: sqlite3_result_int64(context_, static_cast(frameSliceObj_.IdsData()[CurrentRow()])); break; case Index::TS: SetTypeColumnInt64(frameSliceObj_.TimeStampData()[CurrentRow()], INVALID_UINT64); break; case Index::VSYNC: sqlite3_result_int64(context_, static_cast(frameSliceObj_.VsyncIds()[CurrentRow()])); break; case Index::IPID: sqlite3_result_int64(context_, static_cast(frameSliceObj_.Ipids()[CurrentRow()])); break; case Index::ITID: sqlite3_result_int64(context_, static_cast(frameSliceObj_.InternalTidsData()[CurrentRow()])); break; case Index::CALLSTACK_ID: SetTypeColumnInt64(frameSliceObj_.CallStackIds()[CurrentRow()], INVALID_UINT64); break; case Index::DUR: SetTypeColumnInt64(frameSliceObj_.Durs()[CurrentRow()], INVALID_UINT64); break; default: HandleTypeColumns(column); } return SQLITE_OK; } void FrameSliceTable::Cursor::HandleTypeColumns(int32_t column) const { switch (static_cast(column)) { case Index::SRC: sqlite3_result_text(context_, frameSliceObj_.Srcs()[CurrentRow()].c_str(), frameSliceObj_.Srcs()[CurrentRow()].length(), nullptr); break; case Index::DST: SetTypeColumnInt64(frameSliceObj_.Dsts()[CurrentRow()], INVALID_UINT64); break; case Index::TYPE: SetTypeColumnInt64(frameSliceObj_.Types()[CurrentRow()], INVALID_UINT64); break; case Index::TYPE_DESC: sqlite3_result_text(context_, frameSliceObj_.Types()[CurrentRow()] == 0 ? "actural" : "expect", STR_DEFAULT_LEN, nullptr); break; case Index::FLAG: SetTypeColumnInt32(frameSliceObj_.Flags()[CurrentRow()], INVALID_UINT8); break; case Index::DEPTH: SetTypeColumnInt32(frameSliceObj_.Depths()[CurrentRow()], INVALID_UINT8); break; case Index::FRAME_NO: SetTypeColumnInt32(frameSliceObj_.FrameNos()[CurrentRow()], INVALID_UINT32); break; default: TS_LOGF("Unregistered column : %d", column); break; } } void FrameSliceTable::GetOrbyes(FilterConstraints &slicefc, EstimatedIndexInfo &sliceei) { auto sliceorderbys = slicefc.GetOrderBys(); for (auto i = 0; i < sliceorderbys.size(); i++) { switch (static_cast(sliceorderbys[i].iColumn)) { case Index::ID: break; default: // other columns can be sorted by SQLite sliceei.isOrdered = false; break; } } } } // namespace TraceStreamer } // namespace SysTuning