Merge pull request #3068 from cloudflare/hoodmane/more-jaeger-spans

Add more jaeger spans in Python setup
cloudflare · Nov 14, 2024 · 2115004 · 2115004
2 parents 24b9a56 + d9309dd
commit 2115004
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/pyodide/internal/python.ts b/src/pyodide/internal/python.ts
@@ -62,10 +62,12 @@ async function prepareWasmLinearMemory(Module: Module): Promise<void> {
   mountSitePackages(Module, SITE_PACKAGES.rootInfo);
   entropyMountFiles(Module);
   Module.noInitialRun = !SHOULD_RESTORE_SNAPSHOT;
-  preloadDynamicLibs(Module);
-  Module.removeRunDependency('dynlibs');
+  enterJaegerSpan('preload_dynamic_libs', () => preloadDynamicLibs(Module));
+  enterJaegerSpan('remove_run_dependency', () =>
+    Module.removeRunDependency('dynlibs')
+  );
   if (SHOULD_RESTORE_SNAPSHOT) {
-    restoreSnapshot(Module);
+    enterJaegerSpan('restore_snapshot', () => restoreSnapshot(Module));
     // Invalidate caches if we have a snapshot because the contents of site-packages
     // may have changed.
     simpleRunPython(