add operation to TrackedRequest (#796)

* add operation to TrackedRequest

* add sqlalachemy tracked_request operation set and tests

* add additional tests for new tracked_request property

* bump minor version

* stop setting tracked_request operation for sqlalchemy

* Update changelog

* bump macos runner version

Author: quinnmil

.github/workflows/release.yml

@@ -53,7 +53,7 @@ jobs:
 
   build_macos_wheels:
     name: Build macos wheels (cross-compiles arm64)
-    runs-on: macos-11
+    runs-on: macos-13
     steps:
       - uses: actions/checkout@v3
 

CHANGELOG.md

@@ -2,6 +2,10 @@
 
 ## Pending
 
+## [3.2.0] 2024-09-12
+### Added
+- "Operation" attribute added to TrackedRequest class to better support development of [scout_apm_python_logging](https://github.com/scoutapp/scout_apm_python_logging)
+
 ## [3.1.0] 2023-12-18
 ### Added
 - Updates Core Agent to v1.5.0

setup.py

@@ -33,7 +33,7 @@
 
 setup(
     name="scout_apm",
-    version="3.1.0",
+    version="3.2.0",
     description="Scout Application Performance Monitoring Agent",
     long_description=long_description,
     long_description_content_type="text/markdown",

src/scout_apm/async_/starlette.py

@@ -40,6 +40,7 @@ def grab_extra_data():
                     endpoint.__module__,
                     endpoint.__qualname__,
                 )
+                tracked_request.operation = controller_span.operation
             else:
                 # Mark the request as not real
                 tracked_request.is_real_request = False
@@ -90,11 +91,11 @@ async def wrapped_background_call(wrapped, instance, args, kwargs):
         tracked_request = TrackedRequest.instance()
         tracked_request.is_real_request = True
 
-        with tracked_request.span(
-            operation="Job/{}.{}".format(
-                instance.func.__module__, instance.func.__qualname__
-            )
-        ):
+        operation = "Job/{}.{}".format(
+            instance.func.__module__, instance.func.__qualname__
+        )
+        tracked_request.operation = operation
+        with tracked_request.span(operation=operation):
             return await wrapped(*args, **kwargs)
 
     BackgroundTask.__call__ = wrapped_background_call(BackgroundTask.__call__)

src/scout_apm/bottle.py

@@ -71,10 +71,10 @@ def wrap_callback(wrapped, instance, args, kwargs):
         "x-request-start", ""
     )
     track_request_queue_time(queue_time, tracked_request)
+    operation = "Controller{}".format(controller_name)
 
-    with tracked_request.span(
-        operation="Controller{}".format(controller_name), should_capture_backtrace=False
-    ):
+    with tracked_request.span(operation=operation):
+        tracked_request.operation = operation
         try:
             value = wrapped(*args, **kwargs)
         except Exception:

src/scout_apm/core/tracked_request.py

@@ -35,6 +35,7 @@ class TrackedRequest(object):
         "n_plus_one_tracker",
         "hit_max",
         "sent",
+        "operation",
     )
 
     # Stop adding new spans at this point, to avoid exhausting memory
@@ -58,6 +59,7 @@ def __init__(self):
         self.n_plus_one_tracker = NPlusOneTracker()
         self.hit_max = False
         self.sent = False
+        self.operation = None
         logger.debug("Starting request: %s", self.request_id)
 
     def __repr__(self):

src/scout_apm/django/middleware.py

@@ -122,6 +122,7 @@ def process_view(self, request, view_func, view_args, view_kwargs):
         span = tracked_request.current_span()
         if span is not None:
             span.operation = get_controller_name(request)
+            tracked_request.operation = span.operation
 
     def process_exception(self, request, exception):
         """

src/scout_apm/falcon.py

@@ -106,6 +106,7 @@ def process_resource(self, req, resp, resource, params):
         span = tracked_request.start_span(
             operation=operation, should_capture_backtrace=False
         )
+        tracked_request.operation = operation
         req.context.scout_resource_span = span
 
     def _name_operation(self, req, responder, resource):

src/scout_apm/flask/__init__.py

@@ -47,6 +47,7 @@ def wrapped_full_dispatch_request(self, wrapped, instance, args, kwargs):
 
         tracked_request = TrackedRequest.instance()
         tracked_request.is_real_request = True
+        tracked_request.operation = operation
         request._scout_tracked_request = tracked_request
 
         werkzeug_track_request_data(request, tracked_request)

tests/integration/test_bottle.py

@@ -32,7 +32,8 @@ def app_with_scout(config=None, catchall=False):
     # Disable running the agent.
     config["scout.core_agent_launch"] = False
 
-    app = Bottle(catchall=catchall)
+    app = Bottle()
+    app.config["catchall"] = catchall
 
     @app.route("/")
     def home():
@@ -169,6 +170,7 @@ def test_hello(tracked_requests):
     assert len(tracked_request.complete_spans) == 1
     span = tracked_request.complete_spans[0]
     assert span.operation == "Controller/hello/"
+    assert tracked_request.operation == "Controller/hello/"
 
 
 def test_not_found(tracked_requests):
@@ -190,6 +192,7 @@ def test_server_error(tracked_requests):
     assert len(tracked_request.complete_spans) == 1
     span = tracked_requests[0].complete_spans[0]
     assert span.operation == "Controller/crash/"
+    assert tracked_request.operation == "Controller/crash/"
 
 
 def test_return_error(tracked_requests):
@@ -203,6 +206,7 @@ def test_return_error(tracked_requests):
     assert len(tracked_request.complete_spans) == 1
     span = tracked_requests[0].complete_spans[0]
     assert span.operation == "Controller/return-error/"
+    assert tracked_request.operation == "Controller/return-error/"
 
 
 def test_named(tracked_requests):