/*
|
* Copyright (C) 2019 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/args_tracker.h"
|
|
#include <algorithm>
|
|
namespace perfetto {
|
namespace trace_processor {
|
|
ArgsTracker::ArgsTracker(TraceProcessorContext* context) : context_(context) {}
|
|
void ArgsTracker::AddArg(RowId row_id,
|
StringId flat_key,
|
StringId key,
|
Variadic value) {
|
args_.emplace_back();
|
|
auto* rid_arg = &args_.back();
|
rid_arg->row_id = row_id;
|
rid_arg->flat_key = flat_key;
|
rid_arg->key = key;
|
rid_arg->value = value;
|
}
|
|
void ArgsTracker::Flush() {
|
using Arg = TraceStorage::Args::Arg;
|
|
// We sort here because a single packet may add multiple args with different
|
// rowids.
|
auto comparator = [](const Arg& f, const Arg& s) {
|
return f.row_id < s.row_id;
|
};
|
std::sort(args_.begin(), args_.end(), comparator);
|
|
auto* storage = context_->storage.get();
|
for (uint32_t i = 0; i < args_.size();) {
|
const auto& args = args_[i];
|
RowId rid = args.row_id;
|
|
uint32_t next_rid_idx = i + 1;
|
while (next_rid_idx < args_.size() && rid == args_[next_rid_idx].row_id)
|
next_rid_idx++;
|
|
auto set_id = storage->mutable_args()->AddArgSet(args_, i, next_rid_idx);
|
auto pair = TraceStorage::ParseRowId(rid);
|
switch (pair.first) {
|
case TableId::kRawEvents:
|
storage->mutable_raw_events()->set_arg_set_id(pair.second, set_id);
|
break;
|
case TableId::kCounterValues:
|
storage->mutable_counter_values()->set_arg_set_id(pair.second, set_id);
|
break;
|
case TableId::kInstants:
|
storage->mutable_instants()->set_arg_set_id(pair.second, set_id);
|
break;
|
default:
|
PERFETTO_FATAL("Unsupported table to insert args into");
|
}
|
i = next_rid_idx;
|
}
|
args_.clear();
|
}
|
|
} // namespace trace_processor
|
} // namespace perfetto
|
