Fix report builder

gaogaotiantian · Nov 19, 2024 · 0d1796a · 0d1796a
1 parent f470246
commit 0d1796a
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/viztracer/report_builder.py b/src/viztracer/report_builder.py
@@ -132,6 +132,11 @@ def combine_json(self) -> None:
             if one.get("viztracer_metadata", {}).get("baseTimeNanoseconds") is not None:
                 self.combined_json["viztracer_metadata"]["baseTimeNanoseconds"] = \
                     one["viztracer_metadata"]["baseTimeNanoseconds"]
+            if "file_info" in one:
+                if "file_info" not in self.combined_json:
+                    self.combined_json["file_info"] = {"files": {}, "functions": {}}
+                self.combined_json["file_info"]["files"].update(one["file_info"]["files"])
+                self.combined_json["file_info"]["functions"].update(one["file_info"]["functions"])
 
     def align_events(self, original_events: list[dict[str, Any]]) -> list[dict[str, Any]]:
         """
@@ -166,7 +171,8 @@ def prepare_json(self, file_info: bool = True, display_time_unit: Optional[str]
             self.combined_json["viztracer_metadata"]["baseTimeNanoseconds"] = self.base_time
 
         if file_info:
-            self.combined_json["file_info"] = {"files": {}, "functions": {}}
+            if "file_info" not in self.combined_json:
+                self.combined_json["file_info"] = {"files": {}, "functions": {}}
             pattern = re.compile(r".*\((.*):([0-9]*)\)")
             file_dict = self.combined_json["file_info"]["files"]
             func_dict = self.combined_json["file_info"]["functions"]