Refactored generate() for pairs. (#1501)

Refactored `generate()` for pairs to introduce `toJSON()`, for cases where we just want to generate the body of a JSON object.
p4lang · Sep 20, 2018 · 5096e1c · 5096e1c
1 parent d37330b
commit 5096e1c
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/ir/json_generator.h b/ir/json_generator.h
@@ -80,11 +80,18 @@ class JSONGenerator {
 
     template<typename T, typename U>
     void generate(const std::pair<T, U> &v) {
-        out << "{" << std::endl << ++indent << "\"first\" : ";
+        ++indent;
+        out << "{" << std::endl;
+        toJSON(v);
+        out << std::endl << --indent << "}";
+    }
+
+    template<typename T, typename U>
+    void toJSON(const std::pair<T, U> &v) {
+        out << indent << "\"first\" : ";
         generate(v.first);
         out << "," << std::endl << indent << "\"second\" : ";
         generate(v.second);
-        out << std::endl << --indent << "}";
     }
 
     template<typename T>