Add more jaeger spans in Python setup

cloudflare · Nov 6, 2024 · 6a27b4e · 6a27b4e
1 parent 3b9fc88
commit 6a27b4e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/pyodide/internal/python.ts b/src/pyodide/internal/python.ts
@@ -62,10 +62,10 @@ 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('preloadDynamicLibs', () => preloadDynamicLibs(Module));
+  enterJaegerSpan('removeRunDependency', () => Module.removeRunDependency('dynlibs'));
   if (SHOULD_RESTORE_SNAPSHOT) {
-    restoreSnapshot(Module);
+    enterJaegerSpan("restoreSnapshot", () => restoreSnapshot(Module));
     // Invalidate caches if we have a snapshot because the contents of site-packages
     // may have changed.
     simpleRunPython(