remove annotation for drift.recursive_reference when generating json

Author: vhsu14

presto-native-execution/presto_cpp/main/thrift/presto_thrift.thrift

@@ -495,7 +495,7 @@ struct TaskUpdateRequest {
 struct ExecutionFailureInfo {
   1: string type;
   2: string message;
-  3: optional ExecutionFailureInfo cause (cpp.ref_type="shared");
+  3: optional ExecutionFailureInfo cause (cpp.ref_type="shared", drift.recursive_reference=true);
   4: list<ExecutionFailureInfo> suppressed;
   5: list<string> stack;
   6: ErrorLocation errorLocation;

presto-native-execution/presto_cpp/main/thrift/thrift2json.py

@@ -13,6 +13,7 @@
 
 import json
 import sys
+import os
 
 from ptsd_jbroll import ast
 from ptsd_jbroll.loader import Loader
@@ -31,7 +32,19 @@ class attrdict(dict):
     __setattr__ = dict.__setitem__
 
 
-loader = Loader(sys.argv[1], logger=devnull)
+def preprocess(file_path, output_temp_path):
+    with open(file_path, 'r') as file:
+        lines = file.readlines()
+    modified_lines = []
+    for line in lines:
+        if '(cpp.ref_type="shared", drift.recursive_reference=true)' in line:
+            modified_line = line.replace('(cpp.ref_type="shared", drift.recursive_reference=true)', '(cpp.ref_type="shared")')
+            modified_lines.append(modified_line)
+        else:
+            modified_lines.append(line)
+    with open(output_temp_path, 'w') as file:
+        file.writelines(modified_lines)
+
 
 typeMap = {
     "byte": "byte",
@@ -104,8 +117,16 @@ def items(module):
     return i
 
 
-everything = [{"comment": "// This file is generated DO NOT EDIT @" + "generated"}]
-for module in loader.modules.values():
-    everything.extend(items(module))
+def main():
+    preprocess(sys.argv[1], "temp_presto_thrift.thrift")
+    loader = Loader("temp_presto_thrift.thrift", logger=devnull)
+    everything = [{"comment": "// This file is generated DO NOT EDIT @" + "generated"}]
+    for module in loader.modules.values():
+        everything.extend(items(module))
+
+    print(json.dumps(everything))
+    os.remove("temp_presto_thrift.thrift")
+
 
-print(json.dumps(everything))
+if __name__ == "__main__":
+    sys.exit(main())