From 1d708752eea2c390f0f998a687a2382752e9f0ad Mon Sep 17 00:00:00 2001
From: Divyank Jain <divyank.jain@temporal.io>
Date: Fri, 5 Jul 2024 13:30:24 -0400
Subject: [PATCH] skip files that are already mapped for enhanced stack traces

---
 temporalio/worker/_workflow_instance.py | 39 +++++++++++++++----------
 1 file changed, 23 insertions(+), 16 deletions(-)

diff --git a/temporalio/worker/_workflow_instance.py b/temporalio/worker/_workflow_instance.py
index 004faa70..378cfc01 100644
--- a/temporalio/worker/_workflow_instance.py
+++ b/temporalio/worker/_workflow_instance.py
@@ -1870,28 +1870,36 @@ def _enhanced_stack_trace(self) -> temporalio.api.sdk.v1.EnhancedStackTrace:
 
         # this is to use `open`
         with temporalio.workflow.unsafe.sandbox_unrestricted():
-            sources = dict()
-            stacks = []
+            sources: Dict[str, temporalio.api.sdk.v1.StackTraceFileSlice] = dict()
+            stacks: List[temporalio.api.sdk.v1.StackTrace] = []
+
+            # future TODO
+            # site package filter list -- we want to filter out traces from Python's internals and our sdk's internals. This is what `internal_code` is for, but right now it's just set to false.
 
             for task in list(self._tasks):
-                locations = []
+                locations: List[temporalio.api.sdk.v1.StackTraceFileLocation] = []
+
                 for frame in task.get_stack():
                     filename = frame.f_code.co_filename
                     line_number = frame.f_lineno
                     func_name = frame.f_code.co_name
 
-                    try:
-                        with open(filename, "r") as f:
-                            code = f.read()
-                    except OSError as ose:
-                        code = f"Cannot access code.\n---\n{ose.strerror}"
-                        # TODO possibly include sentinel/property for success of src scrape? work out with ui
-                    except Exception:
-                        code = f"Generic Error.\n\n{traceback.format_exc()}"
-
-                    file_slice = temporalio.api.sdk.v1.StackTraceFileSlice(
-                        line_offset=0, content=code
-                    )
+                    if filename not in sources.keys():
+                        try:
+                            with open(filename, "r") as f:
+                                code = f.read()
+                        except OSError as ose:
+                            code = f"Cannot access code.\n---\n{ose.strerror}"
+                            # TODO possibly include sentinel/property for success of src scrape? work out with ui
+                        except Exception:
+                            code = f"Generic Error.\n\n{traceback.format_exc()}"
+
+                        file_slice = temporalio.api.sdk.v1.StackTraceFileSlice(
+                            line_offset=0, content=code
+                        )
+
+                        sources[filename] = file_slice
+
                     file_location = temporalio.api.sdk.v1.StackTraceFileLocation(
                         file_path=filename,
                         line=line_number,
@@ -1900,7 +1908,6 @@ def _enhanced_stack_trace(self) -> temporalio.api.sdk.v1.EnhancedStackTrace:
                         internal_code=False,
                     )
 
-                    sources[filename] = file_slice
                     locations.append(file_location)
 
                 stacks.append(temporalio.api.sdk.v1.StackTrace(locations=locations))