Release v0.1.4

Amp-Thread-ID: https://ampcode.com/threads/T-019e70e6-2c85-756c-8cfb-ac16708fb9ca
Co-authored-by: Amp <amp@ampcode.com>

Author: marcleblanc2

AGENTS.md

@@ -64,7 +64,7 @@ uv run python -m unittest discover -s tests
 ```sh
 set -euo pipefail
 
-VERSION=0.1.2
+VERSION=0.1.4
 BRANCH="release-v${VERSION}"
 
 git fetch origin --tags --prune
@@ -116,7 +116,7 @@ rm -rf /tmp/src-py-lib-release-check
 ```sh
 set -euo pipefail
 
-VERSION=0.1.2
+VERSION=0.1.4
 BRANCH="release-v${VERSION}"
 GH_REPO="sourcegraph/src-py-lib"
 
@@ -140,7 +140,7 @@ gh pr merge "${BRANCH}" --repo "${GH_REPO}" --squash --delete-branch
 ```sh
 set -euo pipefail
 
-VERSION=0.1.2
+VERSION=0.1.4
 
 git fetch origin --tags --prune
 git switch main
@@ -154,7 +154,7 @@ git push origin "v${VERSION}"
 ```sh
 set -euo pipefail
 
-VERSION=0.1.2
+VERSION=0.1.4
 GH_REPO="sourcegraph/src-py-lib"
 
 RUN_ID="$(

pyproject.toml

@@ -10,7 +10,7 @@ dev = [
 
 [project]
 name = "src-py-lib"
-version = "0.1.3"
+version = "0.1.4"
 description = "Reusable libraries for Sourcegraph projects"
 readme = "README.md"
 requires-python = ">=3.11"

src/src_py_lib/__init__.py

@@ -176,6 +176,7 @@ def _script_name() -> str:
     "load_json_cache",
     "load_json_subset",
     "logging",
+    "logging_context",
     "logging_settings_from_config",
     "log",
     "log_context",

src/src_py_lib/clients/sourcegraph.py

@@ -8,6 +8,7 @@
 import queue
 import time
 from collections.abc import Iterable, Iterator, Mapping, Sequence
+from concurrent.futures import ThreadPoolExecutor, as_completed
 from dataclasses import dataclass, field
 from typing import Final, cast
 from urllib.parse import urlsplit
@@ -19,6 +20,7 @@
 from src_py_lib.utils.logging import (
     current_trace_context,
     new_trace_context,
+    submit_with_log_context,
     trace_context_from_traceparent,
     traceparent_header,
 )
@@ -244,13 +246,27 @@ def stream_jaeger_trace_summaries(
         traces: Iterable[SourcegraphTrace] | None = None,
         *,
         retry_delays_seconds: Sequence[float] = JAEGER_TRACE_RETRY_DELAYS_SECONDS,
+        parallelism: int = 8,
     ) -> Iterator[SourcegraphJaegerTraceSummary]:
         """Yield compact Jaeger/debug summaries for traced Sourcegraph requests."""
-        for trace in self.drain_traces() if traces is None else traces:
-            yield self.fetch_jaeger_trace_summary(
-                trace,
-                retry_delays_seconds=retry_delays_seconds,
-            )
+        if parallelism < 1:
+            raise ValueError("parallelism must be at least 1")
+        pending_traces = list(self.drain_traces() if traces is None else traces)
+        with ThreadPoolExecutor(
+            max_workers=parallelism,
+            thread_name_prefix="SourcegraphJaegerTrace",
+        ) as executor:
+            futures = [
+                submit_with_log_context(
+                    executor,
+                    self.fetch_jaeger_trace_summary,
+                    trace,
+                    retry_delays_seconds=retry_delays_seconds,
+                )
+                for trace in pending_traces
+            ]
+            for future in as_completed(futures):
+                yield future.result()
 
     def fetch_jaeger_trace_summary(
         self,

tests/test_logging_http_clients.py

@@ -8,6 +8,7 @@
 import logging
 import subprocess
 import tempfile
+import threading
 import unittest
 from collections.abc import Mapping
 from contextlib import redirect_stderr, redirect_stdout
@@ -36,6 +37,7 @@
 from src_py_lib.clients.sourcegraph import (
     SourcegraphClient,
     SourcegraphClientConfig,
+    SourcegraphTrace,
     decode_external_service_id,
     decode_repository_id,
     encode_repository_id,
@@ -1535,6 +1537,50 @@ def handler(request: httpx.Request) -> httpx.Response:
         self.assertEqual(summaries[0].graphql_operations[0]["operation"], "Viewer")
         self.assertEqual(summaries[0].errored_spans[0]["description"], "boom")
 
+    def test_sourcegraph_streams_jaeger_summaries_in_parallel(self) -> None:
+        trace_ids = ("1" * 32, "2" * 32, "3" * 32)
+        requested_trace_ids: list[str] = []
+        first_batch_barrier = threading.Barrier(2, timeout=1)
+
+        def handler(request: httpx.Request) -> httpx.Response:
+            trace_id = request.url.path.rsplit("/", 1)[-1]
+            requested_trace_ids.append(trace_id)
+            if trace_id in trace_ids[:2]:
+                first_batch_barrier.wait()
+            return httpx.Response(
+                200,
+                json={
+                    "data": [
+                        {
+                            "spans": [
+                                {
+                                    "operationName": f"trace {trace_id[0]}",
+                                    "duration": 1_000,
+                                    "tags": [],
+                                }
+                            ]
+                        }
+                    ]
+                },
+            )
+
+        client = SourcegraphClient(
+            "https://sourcegraph.example.com/",
+            "token",
+            http=HTTPClient(max_attempts=1, transport=httpx.MockTransport(handler)),
+        )
+
+        summaries = list(
+            client.stream_jaeger_trace_summaries(
+                [SourcegraphTrace(trace_id) for trace_id in trace_ids],
+                retry_delays_seconds=(0,),
+                parallelism=2,
+            )
+        )
+
+        self.assertCountEqual(requested_trace_ids, trace_ids)
+        self.assertCountEqual([summary.trace.trace_id for summary in summaries], trace_ids)
+
     def test_graphql_client_paginates_cursor_results(self) -> None:
         http = RecordingHTTP(
             [