Add default to _after_request_fn (#6)

opentracing-contrib · Jun 8, 2018 · c003984 · c003984
1 parent fcf20ad
commit c003984
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/flask_opentracing/tracer.py b/flask_opentracing/tracer.py
@@ -92,6 +92,7 @@ def _before_request_fn(self, attributes):
 
     def _after_request_fn(self):
         request = stack.top.request
-        span = self._current_spans.pop(request)
+        # the pop call can fail if the request is interrupted by a `before_request` method so we need a default
+        span = self._current_spans.pop(request, None)
         if span is not None:
             span.finish()