Add binary protocol support to thrift2ast

Summary: This is a better choice when the output will be compressed or size does not matter.

Reviewed By: thedavekwon

Differential Revision: D59119356

fbshipit-source-id: 6afb0c657abe66307e22021eec8d82515a729789

Author: iahs

third-party/thrift/src/thrift/compiler/generate/t_ast_generator.cc

@@ -31,6 +31,7 @@
 
 #include <folly/compression/Compression.h>
 #include <thrift/lib/cpp2/op/Sha256Hasher.h>
+#include <thrift/lib/cpp2/protocol/BinaryProtocol.h>
 #include <thrift/lib/cpp2/protocol/CompactProtocol.h>
 #include <thrift/lib/cpp2/protocol/DebugProtocol.h>
 #include <thrift/lib/cpp2/protocol/Serializer.h>
@@ -46,6 +47,7 @@ enum class ast_protocol {
   json,
   debug,
   compact,
+  binary,
 };
 
 template <typename Writer, typename T>
@@ -85,6 +87,8 @@ class t_ast_generator : public t_generator {
           protocol_ = ast_protocol::debug;
         } else if (pair.second == "compact") {
           protocol_ = ast_protocol::compact;
+        } else if (pair.second == "binary") {
+          protocol_ = ast_protocol::binary;
         } else {
           throw std::runtime_error(
               fmt::format("Unknown protocol `{}`", pair.second));
@@ -512,6 +516,9 @@ void t_ast_generator::generate_program() {
     case ast_protocol::compact:
       f_out_ << serialize<CompactProtocolWriter>(ast);
       break;
+    case ast_protocol::binary:
+      f_out_ << serialize<BinaryProtocolWriter>(ast);
+      break;
   }
   f_out_.close();
 }