Skip to content

Commit 296026e

Browse files
fregataaclaude
andcommitted
refactor: rename RouteTargetStatuses.traffic_status to traffic
The dataclass field shares its name with ``RoutingRow.traffic_status`` (the SQL column) and ``RouteBatchUpdaterSpec.traffic_status`` (a writer field) — both unrelated. Shorten to ``traffic`` so the call site reads as a target filter axis next to ``lifecycle`` and ``health`` instead of echoing the column name. Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
1 parent 23e8df2 commit 296026e

5 files changed

Lines changed: 10 additions & 10 deletions

File tree

src/ai/backend/manager/data/deployment/types.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -252,13 +252,13 @@ class DeploymentTargetStatuses:
252252
class RouteTargetStatuses:
253253
"""Target statuses for route handler filtering (lifecycle x health x traffic).
254254
255-
``traffic_status=None`` skips the filter; otherwise the row's
256-
``traffic_status`` must match.
255+
``traffic=None`` skips the filter; otherwise the row's
256+
``traffic_status`` column must match.
257257
"""
258258

259259
lifecycle: list[RouteStatus]
260260
health: list[RouteHealthStatus]
261-
traffic_status: RouteTrafficStatus | None = None
261+
traffic: RouteTrafficStatus | None = None
262262

263263

264264
@dataclass(frozen=True)

src/ai/backend/manager/repositories/deployment/db_source/db_source.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1619,7 +1619,7 @@ async def get_routes_by_statuses(
16191619
self,
16201620
target: RouteTargetStatuses,
16211621
) -> list[RouteData]:
1622-
"""Routes matching ``(lifecycle, health, traffic_status)``.
1622+
"""Routes matching ``(lifecycle, health, traffic)``.
16231623
16241624
``model_definition`` is selected so the resolved
16251625
``ModelHealthCheck`` (or ``None``) can be attached to each
@@ -1643,8 +1643,8 @@ async def get_routes_by_statuses(
16431643
RoutingRow.health_status.in_(target.health),
16441644
)
16451645
)
1646-
if target.traffic_status is not None:
1647-
query = query.where(RoutingRow.traffic_status == target.traffic_status)
1646+
if target.traffic is not None:
1647+
query = query.where(RoutingRow.traffic_status == target.traffic)
16481648
result = await db_sess.execute(query)
16491649
return [
16501650
RouteData(

src/ai/backend/manager/repositories/deployment/repository.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -632,7 +632,7 @@ async def get_routes_by_statuses(
632632
self,
633633
target: RouteTargetStatuses,
634634
) -> list[RouteData]:
635-
"""Routes matching ``(lifecycle, health, traffic_status)``."""
635+
"""Routes matching ``(lifecycle, health, traffic)``."""
636636
return await self._db_source.get_routes_by_statuses(target)
637637

638638
@deployment_repository_resilience.apply()

src/ai/backend/manager/sokovan/deployment/route/handlers/appproxy_sync.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ def target_statuses(cls) -> RouteTargetStatuses:
6363
return RouteTargetStatuses(
6464
lifecycle=[RouteStatus.RUNNING],
6565
health=[RouteHealthStatus.HEALTHY, RouteHealthStatus.NOT_CHECKED],
66-
traffic_status=RouteTrafficStatus.ACTIVE,
66+
traffic=RouteTrafficStatus.ACTIVE,
6767
)
6868

6969
@classmethod

tests/unit/manager/repositories/deployment/test_deployment_repository.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3955,14 +3955,14 @@ async def test_traffic_status_filter(
39553955
deployment_repository: DeploymentRepository,
39563956
populated_routes: dict[str, uuid.UUID],
39573957
) -> None:
3958-
"""``traffic_status=INACTIVE`` excludes ACTIVE rows; omitting it (``None``)
3958+
"""``traffic=INACTIVE`` excludes ACTIVE rows; omitting it (``None``)
39593959
keeps both ACTIVE and INACTIVE rows in the result set.
39603960
"""
39613961
only_inactive = await deployment_repository.get_routes_by_statuses(
39623962
RouteTargetStatuses(
39633963
lifecycle=[RouteStatus.RUNNING],
39643964
health=[RouteHealthStatus.HEALTHY],
3965-
traffic_status=RouteTrafficStatus.INACTIVE,
3965+
traffic=RouteTrafficStatus.INACTIVE,
39663966
),
39673967
)
39683968
both = await deployment_repository.get_routes_by_statuses(

0 commit comments

Comments
 (0)