[tracing] Switch to new heap dump format.

base/trace_event/heap_profiler_event_writer.cc

+// Copyright 2017 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "base/trace_event/heap_profiler_event_writer.h"
+
+#include <stdint.h>
+
+#include <tuple>
+#include <unordered_map>
+
+#include "base/memory/ptr_util.h"
+#include "base/numerics/safe_conversions.h"
+#include "base/trace_event/heap_profiler_allocation_register.h"
+#include "base/trace_event/heap_profiler_serialization_state.h"
+#include "base/trace_event/heap_profiler_stack_frame_deduplicator.h"
+#include "base/trace_event/heap_profiler_string_deduplicator.h"
+#include "base/trace_event/heap_profiler_type_name_deduplicator.h"
+#include "base/trace_event/trace_event_argument.h"
+
+namespace base {
+namespace trace_event {
+
+namespace {
+
+struct AggregationKey {
+  int backtrace_id;
+  int type_id;
+
+  struct Hasher {
+    size_t operator()(const AggregationKey& key) const {
+      return base::HashInts(key.backtrace_id, key.type_id);
+    }
+  };
+
+  bool operator==(const AggregationKey& other) const {
+    return backtrace_id == other.backtrace_id && type_id == other.type_id;
+  }
+};
+
+}  // namespace
+
+std::unique_ptr<TracedValue> SerializeHeapDump(
+    const AllocationRegister& allocation_register,
+    const HeapProfilerSerializationState& serialization_state) {
+  // Aggregate allocations by {backtrace_id, type_id} key.
+  std::unordered_map<AggregationKey, AllocationMetrics, AggregationKey::Hasher>
+      metrics_by_key;
+  for (const auto& allocation : allocation_register) {
+    int backtrace_id = serialization_state.stack_frame_deduplicator()->Insert(
+        std::begin(allocation.context.backtrace.frames),
+        std::begin(allocation.context.backtrace.frames) +
+            allocation.context.backtrace.frame_count);
+
+    int type_id = serialization_state.type_name_deduplicator()->Insert(
+        allocation.context.type_name);
+
+    AggregationKey key = {backtrace_id, type_id};
+    AllocationMetrics& metrics = metrics_by_key[key];
+    metrics.size += allocation.size;
+    metrics.count += 1;
+  }
+
+  auto traced_value = MakeUnique<TracedValue>();
+
+  traced_value->BeginArray("nodes");
+  for (const auto& key_and_metrics : metrics_by_key)
+    traced_value->AppendInteger(key_and_metrics.first.backtrace_id);
+  traced_value->EndArray();
+
+  traced_value->BeginArray("types");
+  for (const auto& key_and_metrics : metrics_by_key)
+    traced_value->AppendInteger(key_and_metrics.first.type_id);
+  traced_value->EndArray();
+
+  traced_value->BeginArray("counts");
+  for (const auto& key_and_metrics : metrics_by_key)
+    traced_value->AppendInteger(
+        saturated_cast<int>(key_and_metrics.second.count));
+  traced_value->EndArray();
+
+  traced_value->BeginArray("sizes");
+  for (const auto& key_and_metrics : metrics_by_key)
+    traced_value->AppendInteger(
+        saturated_cast<int>(key_and_metrics.second.size));
+  traced_value->EndArray();
+
+  return traced_value;
+}
+
+std::unique_ptr<TracedValue> SerializeHeapProfileEventData(
+    const SerializedHeapDumpsMap& heap_dumps,
+    const HeapProfilerSerializationState& serialization_state) {
+  auto traced_value = MakeUnique<TracedValue>();
+
+  // See brief description of the format in the header file.
+  traced_value->SetInteger("version", 1);
+
+  traced_value->BeginDictionary("allocators");
+  for (const auto& name_and_dump : heap_dumps) {
+    traced_value->SetValueWithCopiedName(name_and_dump.first.c_str(),
+                                         *name_and_dump.second);
+  }
+  traced_value->EndDictionary();
+
+  traced_value->BeginDictionary("maps");
+
+  if (auto* deduplicator = serialization_state.stack_frame_deduplicator()) {
+    traced_value->BeginArray("nodes");
+    deduplicator->SerializeIncrementally(&*traced_value);
+    traced_value->EndArray();
+  }
+
+  if (auto* deduplicator = serialization_state.type_name_deduplicator()) {
+    traced_value->BeginArray("types");
+    deduplicator->SerializeIncrementally(&*traced_value);
+    traced_value->EndArray();
+  }
+
+  if (auto* deduplicator = serialization_state.string_deduplicator()) {
+    traced_value->BeginArray("strings");
+    deduplicator->SerializeIncrementally(&*traced_value);
+    traced_value->EndArray();
+  }
+
+  traced_value->EndDictionary();
+
+  return traced_value;
+}
+
+}  // namespace trace_event
+}  // namespace base