Use absl::flat_hash_map in StatisticsWriter::GenerateStatistics

PiperOrigin-RevId: 408846483 Change-Id: Ic22876fa24216af32cf62d4ee5037b139527b242
google · Nov 10, 2021 · c2d8716 · c2d8716
1 parent d13ee1f
commit c2d8716
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 7 deletions.
diff --git a/statistics_writer.cc b/statistics_writer.cc
@@ -15,9 +15,9 @@
 #include "third_party/zynamics/binexport/statistics_writer.h"
 
 #include <iomanip>
-#include <map>
 #include <string>
 
+#include "third_party/absl/container/flat_hash_map.h"
 #include "third_party/zynamics/binexport/call_graph.h"
 #include "third_party/zynamics/binexport/flow_graph.h"
 
@@ -30,8 +30,8 @@ StatisticsWriter::StatisticsWriter(const std::string& filename)
 
 void StatisticsWriter::GenerateStatistics(
     const CallGraph& call_graph, const FlowGraph& flow_graph,
-    std::map<std::string, size_t>* statistics_ptr) const {
-  std::map<std::string, size_t>& statistics = *statistics_ptr;
+    absl::flat_hash_map<std::string, size_t>* statistics_ptr) const {
+  absl::flat_hash_map<std::string, size_t>& statistics = *statistics_ptr;
   statistics.clear();
   const Functions& functions = flow_graph.GetFunctions();
   statistics["callgraph nodes (functions)"] = functions.size();
@@ -96,7 +96,7 @@ absl::Status StatisticsWriter::Write(const CallGraph& call_graph,
                                      const Instructions&,
                                      const AddressReferences&,
                                      const AddressSpace&) {
-  std::map<std::string, size_t> statistics;
+  absl::flat_hash_map<std::string, size_t> statistics;
   GenerateStatistics(call_graph, flow_graph, &statistics);
   for (const auto& entry : statistics) {
     const std::string padding(32 - entry.first.size(), '.');

diff --git a/statistics_writer.h b/statistics_writer.h
@@ -17,6 +17,7 @@
 
 #include <fstream>
 
+#include "third_party/absl/container/flat_hash_map.h"
 #include "third_party/zynamics/binexport/writer.h"
 
 namespace security::binexport {
@@ -26,9 +27,9 @@ class StatisticsWriter : public Writer {
   explicit StatisticsWriter(std::ostream& stream);
   explicit StatisticsWriter(const std::string& filename);
 
-  void GenerateStatistics(const CallGraph& call_graph,
-                          const FlowGraph& flow_graph,
-                          std::map<std::string, size_t>* statistics) const;
+  void GenerateStatistics(
+      const CallGraph& call_graph, const FlowGraph& flow_graph,
+      absl::flat_hash_map<std::string, size_t>* statistics) const;
 
   absl::Status Write(const CallGraph& call_graph, const FlowGraph& flow_graph,
                      const Instructions&, const AddressReferences&,