Browse Source

HUE-8936 [tracing] Add span for additional trace in execute

Romain 6 năm trước cách đây
mục cha
commit
047e1782f2
1 tập tin đã thay đổi với 7 bổ sung5 xóa
  1. 7 5
      desktop/libs/notebook/src/notebook/api.py

+ 7 - 5
desktop/libs/notebook/src/notebook/api.py

@@ -130,7 +130,8 @@ def _execute_notebook(request, notebook, snippet):
       if snippet.get('interface') == 'sqlalchemy':
         interpreter.options['session'] = session
 
-      response['handle'] = interpreter.execute(notebook, snippet)
+      with opentracing.tracer.start_span('execute') as site_span:
+        response['handle'] = interpreter.execute(notebook, snippet)
 
       # Retrieve and remove the result from the handle
       if response['handle'].get('sync'):
@@ -181,10 +182,11 @@ def execute(request, engine=None):
 
   response = _execute_notebook(request, notebook, snippet)
 
-  opentracing.tracer.active_span.set_tag(
-    'query-id',
-    response['handle']['guid'] if response.get('handle') and response['handle'].get('guid') else None
-  )
+  with opentracing.tracer.start_span('execute') as span:
+    span.set_tag(
+      'query-id',
+      response['handle']['guid'] if response.get('handle') and response['handle'].get('guid') else None
+    )
 
   return JsonResponse(response)