Skip to content

Commit 4b0ae75

Browse files
authored
fix: Task processor overload when handling high volume SDK traffic (#5358)
1 parent 3545566 commit 4b0ae75

4 files changed

Lines changed: 4 additions & 4 deletions

File tree

api/app_analytics/cache.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ def __init__(self) -> None:
2020

2121
def _flush(self) -> None:
2222
for key, value in self._cache.items():
23-
track_request.delay(
23+
track_request.run_in_thread(
2424
kwargs={
2525
"resource": key[0].value,
2626
"host": key[1],

api/app_analytics/services.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ def track_usage_by_resource_host_and_environment(
2020
environment_key=environment_key,
2121
)
2222
else:
23-
track_request.delay(
23+
track_request.run_in_thread(
2424
kwargs={
2525
"resource": resource.value,
2626
"host": host,

api/tests/unit/app_analytics/test_middleware.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ def test_api_usage_middleware__no_cache__calls_expected(
7878
middleware(request)
7979

8080
# Then
81-
mocked_track_request.delay.assert_called_once_with(
81+
mocked_track_request.run_in_thread.assert_called_once_with(
8282
kwargs={
8383
"resource": Resource.get_from_name(resource_name),
8484
"environment_key": environment_key,

api/tests/unit/app_analytics/test_unit_app_analytics_cache.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@ def test_api_usage_cache(
7272
}
7373
)
7474
)
75-
mocked_track_request_task.delay.assert_has_calls(expected_calls)
75+
mocked_track_request_task.run_in_thread.assert_has_calls(expected_calls)
7676

7777
# Next, let's reset the mock
7878
mocked_track_request_task.reset_mock()

0 commit comments

Comments
 (0)