| /* |
| * Copyright (C) 2018 The Android Open Source Project |
| * |
| * 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 "src/trace_processor/thread_table.h" |
| |
| #include "perfetto/base/logging.h" |
| #include "src/trace_processor/query_constraints.h" |
| #include "src/trace_processor/sqlite_utils.h" |
| |
| namespace perfetto { |
| namespace trace_processor { |
| |
| namespace { |
| |
| using namespace sqlite_utils; |
| |
| } // namespace |
| |
| ThreadTable::ThreadTable(sqlite3*, const TraceStorage* storage) |
| : storage_(storage) {} |
| |
| void ThreadTable::RegisterTable(sqlite3* db, const TraceStorage* storage) { |
| Table::Register<ThreadTable>(db, storage, "thread"); |
| } |
| |
| std::string ThreadTable::CreateTableStmt(int, const char* const*) { |
| return "CREATE TABLE x(" |
| "utid UNSIGNED INT, " |
| "upid UNSIGNED INT, " |
| "name TEXT, " |
| "tid UNSIGNED INT, " |
| "PRIMARY KEY(utid)" |
| ") WITHOUT ROWID;"; |
| } |
| |
| std::unique_ptr<Table::Cursor> ThreadTable::CreateCursor() { |
| return std::unique_ptr<Table::Cursor>(new Cursor(storage_)); |
| } |
| |
| int ThreadTable::BestIndex(const QueryConstraints& qc, BestIndexInfo* info) { |
| info->estimated_cost = static_cast<uint32_t>(storage_->thread_count()); |
| |
| // If the query has a constraint on the |utid| field, return a reduced cost |
| // because we can do that filter efficiently. |
| const auto& constraints = qc.constraints(); |
| if (constraints.size() == 1 && constraints.front().iColumn == Column::kUtid) { |
| info->estimated_cost = IsOpEq(constraints.front().op) ? 1 : 10; |
| } |
| |
| return SQLITE_OK; |
| } |
| |
| ThreadTable::Cursor::Cursor(const TraceStorage* storage) : storage_(storage) {} |
| |
| int ThreadTable::Cursor::Column(sqlite3_context* context, int N) { |
| const auto& thread = storage_->GetThread(utid_filter_.current); |
| switch (N) { |
| case Column::kUtid: { |
| sqlite3_result_int64(context, utid_filter_.current); |
| break; |
| } |
| case Column::kUpid: { |
| sqlite3_result_int64(context, thread.upid); |
| break; |
| } |
| case Column::kName: { |
| const auto& name = storage_->GetString(thread.name_id); |
| sqlite3_result_text(context, name.c_str(), |
| static_cast<int>(name.length()), nullptr); |
| break; |
| } |
| case Column::kTid: { |
| sqlite3_result_int64(context, thread.tid); |
| break; |
| } |
| default: { |
| PERFETTO_FATAL("Unknown column %d", N); |
| break; |
| } |
| } |
| return SQLITE_OK; |
| } |
| |
| int ThreadTable::Cursor::Filter(const QueryConstraints& qc, |
| sqlite3_value** argv) { |
| utid_filter_.min = 1; |
| utid_filter_.max = static_cast<uint32_t>(storage_->thread_count()); |
| utid_filter_.desc = false; |
| utid_filter_.current = utid_filter_.min; |
| for (size_t j = 0; j < qc.constraints().size(); j++) { |
| const auto& cs = qc.constraints()[j]; |
| if (cs.iColumn == Column::kUtid) { |
| UniqueTid constraint_utid = |
| static_cast<UniqueTid>(sqlite3_value_int(argv[j])); |
| // Filter the range of utids that we are interested in, based on the |
| // constraints in the query. Everything between min and max (inclusive) |
| // will be returned. |
| if (IsOpEq(cs.op)) { |
| utid_filter_.min = constraint_utid; |
| utid_filter_.max = constraint_utid; |
| } else if (IsOpGe(cs.op) || IsOpGt(cs.op)) { |
| utid_filter_.min = |
| IsOpGt(cs.op) ? constraint_utid + 1 : constraint_utid; |
| } else if (IsOpLe(cs.op) || IsOpLt(cs.op)) { |
| utid_filter_.max = |
| IsOpLt(cs.op) ? constraint_utid - 1 : constraint_utid; |
| } |
| } |
| } |
| for (const auto& ob : qc.order_by()) { |
| if (ob.iColumn == Column::kUtid) { |
| utid_filter_.desc = ob.desc; |
| utid_filter_.current = |
| utid_filter_.desc ? utid_filter_.max : utid_filter_.min; |
| } |
| } |
| |
| return SQLITE_OK; |
| } |
| |
| int ThreadTable::Cursor::Next() { |
| if (utid_filter_.desc) { |
| --utid_filter_.current; |
| } else { |
| ++utid_filter_.current; |
| } |
| |
| return SQLITE_OK; |
| } |
| |
| int ThreadTable::Cursor::Eof() { |
| return utid_filter_.desc ? utid_filter_.current < utid_filter_.min |
| : utid_filter_.current > utid_filter_.max; |
| } |
| } // namespace trace_processor |
| } // namespace perfetto |