prefect 3.1.16.dev2__tar.gz → 3.1.16.dev4__tar.gz

{prefect-3.1.16.dev2/src/prefect.egg-info → prefect-3.1.16.dev4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: prefect
-Version: 3.1.16.dev2
+Version: 3.1.16.dev4
 Summary: Workflow orchestration and management.
 Home-page: https://www.prefect.io
 Author: Prefect Technologies, Inc.
@@ -34,7 +34,7 @@ Requires-Dist: graphviz>=0.20.1
 Requires-Dist: jinja2<4.0.0,>=3.0.0
 Requires-Dist: jinja2-humanize-extension>=0.4.0
 Requires-Dist: humanize<5.0.0,>=4.9.0
-Requires-Dist: pytz<2025,>=2021.1
+Requires-Dist: pytz<2026,>=2021.1
 Requires-Dist: readchar<5.0.0,>=4.0.0
 Requires-Dist: sqlalchemy[asyncio]<3.0.0,>=2.0
 Requires-Dist: typer!=0.12.2,<0.16.0,>=0.12.0
@@ -104,8 +104,8 @@ Requires-Dist: vale; extra == "dev"
 Requires-Dist: virtualenv; extra == "dev"
 Requires-Dist: watchfiles; extra == "dev"
 Requires-Dist: respx; extra == "dev"
-Requires-Dist: codespell==2.2.6; extra == "dev"
-Requires-Dist: ruff==0.9.2; extra == "dev"
+Requires-Dist: codespell==2.4.1; extra == "dev"
+Requires-Dist: ruff==0.9.4; extra == "dev"
 Requires-Dist: mypy==1.14.1; extra == "dev"
 Requires-Dist: vermin==1.6.0; extra == "dev"
 Requires-Dist: types-cachetools; extra == "dev"

{prefect-3.1.16.dev2 → prefect-3.1.16.dev4}/requirements-dev.txt

@@ -27,8 +27,8 @@ respx
 
 # Linters and dev tools that are also in .pre-commit-config.yaml, which
 # should usually be updated together.
-codespell==2.2.6
-ruff==0.9.2
+codespell==2.4.1
+ruff==0.9.4
 mypy==1.14.1
 vermin==1.6.0
 

{prefect-3.1.16.dev2 → prefect-3.1.16.dev4}/requirements.txt

@@ -12,7 +12,7 @@ graphviz >= 0.20.1
 jinja2 >= 3.0.0, < 4.0.0
 jinja2-humanize-extension >= 0.4.0
 humanize >= 4.9.0, < 5.0.0
-pytz >= 2021.1, < 2025
+pytz >= 2021.1, < 2026
 readchar >= 4.0.0, < 5.0.0
 sqlalchemy[asyncio] >= 2.0, < 3.0.0
 typer >= 0.12.0, != 0.12.2, < 0.16.0

{prefect-3.1.16.dev2 → prefect-3.1.16.dev4}/src/prefect/_version.py

@@ -8,11 +8,11 @@ import json
 
 version_json = '''
 {
- "date": "2025-01-31T16:51:35-0600",
+ "date": "2025-02-03T16:32:08-0800",
  "dirty": false,
  "error": null,
- "full-revisionid": "26ae72909896078d4436e4b7e90075d586347f53",
- "version": "3.1.16.dev2"
+ "full-revisionid": "7622daf8f43694aa2aad726e7b93aad4203a8c14",
+ "version": "3.1.16.dev4"
 }
 '''  # END VERSION_JSON
 

{prefect-3.1.16.dev2 → prefect-3.1.16.dev4}/src/prefect/cli/profile.py

@@ -67,7 +67,7 @@ def create(
     Create a new profile.
     """
 
-    profiles = prefect.settings.load_profiles()
+    profiles = prefect.settings.load_profiles(include_defaults=False)
     if name in profiles:
         app.console.print(
             textwrap.dedent(
@@ -211,7 +211,7 @@ def rename(name: str, new_name: str):
     """
     Change the name of a profile.
     """
-    profiles = prefect.settings.load_profiles()
+    profiles = prefect.settings.load_profiles(include_defaults=False)
     if name not in profiles:
         exit_with_error(f"Profile {name!r} not found.")
 

{prefect-3.1.16.dev2 → prefect-3.1.16.dev4}/src/prefect/cli/server.py

@@ -523,28 +523,6 @@ async def stamp(revision: str):
     exit_with_success("Stamping database with revision succeeded!")
 
 
-async def _run_services(
-    service_classes: list[type[Service]],
-):
-    """Run the given service classes until cancelled."""
-    services = [cls() for cls in service_classes]
-    tasks: list[tuple[asyncio.Task[None], type[Service]]] = []
-
-    for service in services:
-        task = asyncio.create_task(service.start())
-        tasks.append((task, service))
-        logger.debug(f"Started service: {service.name}")
-
-    try:
-        await asyncio.gather(*(t for t, _ in tasks))
-    except asyncio.CancelledError:
-        logger.info("Received cancellation, stopping services...")
-        for task, service in tasks:
-            task.cancel()
-            logger.debug(f"Stopped service: {service.name}")
-        await asyncio.gather(*(t for t, _ in tasks), return_exceptions=True)
-
-
 def _is_process_running(pid: int) -> bool:
     """Check if a process is running by attempting to send signal 0."""
     try:
@@ -586,13 +564,13 @@ def run_manager_process():
 
     We do everything in sync so that the child won't exit until the user kills it.
     """
-    if not (enabled_services := Service.enabled_services()):
+    if not Service.enabled_services():
         logger.error("No services are enabled! Exiting manager.")
         sys.exit(1)
 
     logger.debug("Manager process started. Starting services...")
     try:
-        asyncio.run(_run_services(enabled_services))
+        asyncio.run(Service.run_services())
     except KeyboardInterrupt:
         pass
     finally:
@@ -644,22 +622,19 @@ def start_services(
             # Stale or invalid file
             _cleanup_pid_file(SERVICES_PID_FILE)
 
-    if not (enabled_services := Service.enabled_services()):
+    if not Service.enabled_services():
         app.console.print("[red]No services are enabled![/]")
         raise typer.Exit(code=1)
 
     if not background:
         app.console.print("\n[blue]Starting services... Press CTRL+C to stop[/]\n")
         try:
-            asyncio.run(_run_services(enabled_services))
+            asyncio.run(Service.run_services())
         except KeyboardInterrupt:
             pass
         app.console.print("\n[green]All services stopped.[/]")
         return
 
-    for service in enabled_services:
-        app.console.print(f"Starting service: [yellow]{service.__name__}[/]")
-
     process = subprocess.Popen(
         [
             "prefect",

{prefect-3.1.16.dev2 → prefect-3.1.16.dev4}/src/prefect/client/schemas/schedules.py

@@ -8,7 +8,6 @@ from typing import TYPE_CHECKING, Annotated, Any, ClassVar, Optional, Union
 import dateutil
 import dateutil.rrule
 import dateutil.tz
-import pendulum
 from pydantic import AfterValidator, ConfigDict, Field, field_validator, model_validator
 from typing_extensions import TypeAlias, TypeGuard
 
@@ -19,14 +18,7 @@ from prefect._internal.schemas.validators import (
     validate_cron_string,
     validate_rrule_string,
 )
-
-if TYPE_CHECKING:
-    # type checkers have difficulty accepting that
-    # pydantic_extra_types.pendulum_dt and pendulum.DateTime can be used
-    # together.
-    DateTime = pendulum.DateTime
-else:
-    from prefect.types import DateTime
+from prefect.types._datetime import Date, DateTime
 
 MAX_ITERATIONS = 1000
 # approx. 1 years worth of RDATEs + buffer
@@ -83,7 +75,7 @@ class IntervalSchedule(PrefectBaseModel):
 
     interval: datetime.timedelta = Field(gt=datetime.timedelta(0))
     anchor_date: Annotated[DateTime, AfterValidator(default_anchor_date)] = Field(
-        default_factory=lambda: pendulum.now("UTC"),
+        default_factory=lambda: DateTime.now("UTC"),
         examples=["2020-01-01T00:00:00Z"],
     )
     timezone: Optional[str] = Field(default=None, examples=["America/New_York"])
@@ -99,9 +91,7 @@ class IntervalSchedule(PrefectBaseModel):
             self,
             /,
             interval: datetime.timedelta,
-            anchor_date: Optional[
-                Union[pendulum.DateTime, datetime.datetime, str]
-            ] = None,
+            anchor_date: Optional[Union[DateTime, datetime.datetime, str]] = None,
             timezone: Optional[str] = None,
         ) -> None: ...
 
@@ -153,7 +143,7 @@ class CronSchedule(PrefectBaseModel):
         return validate_cron_string(v)
 
 
-DEFAULT_ANCHOR_DATE = pendulum.date(2020, 1, 1)
+DEFAULT_ANCHOR_DATE = Date(2020, 1, 1)
 
 
 def _rrule_dt(
@@ -219,7 +209,7 @@ class RRuleSchedule(PrefectBaseModel):
             return RRuleSchedule(rrule=str(rrule), timezone=timezone)
         rrules = _rrule(rrule)
         dtstarts = [dts for rr in rrules if (dts := _rrule_dt(rr)) is not None]
-        unique_dstarts = set(pendulum.instance(d).in_tz("UTC") for d in dtstarts)
+        unique_dstarts = set(DateTime.instance(d).in_tz("UTC") for d in dtstarts)
         unique_timezones = set(d.tzinfo for d in dtstarts if d.tzinfo is not None)
 
         if len(unique_timezones) > 1:
@@ -380,7 +370,7 @@ def construct_schedule(
         if isinstance(interval, (int, float)):
             interval = datetime.timedelta(seconds=interval)
         if not anchor_date:
-            anchor_date = pendulum.DateTime.now()
+            anchor_date = DateTime.now()
         schedule = IntervalSchedule(
             interval=interval, anchor_date=anchor_date, timezone=timezone
         )

{prefect-3.1.16.dev2 → prefect-3.1.16.dev4}/src/prefect/server/api/server.py

@@ -19,9 +19,9 @@ import subprocess
 import sys
 import time
 from contextlib import asynccontextmanager
-from functools import partial, wraps
+from functools import wraps
 from hashlib import sha256
-from typing import TYPE_CHECKING, Any, Awaitable, Callable, Optional
+from typing import TYPE_CHECKING, Any, AsyncGenerator, Awaitable, Callable, Optional
 
 import anyio
 import asyncpg
@@ -42,18 +42,12 @@ from typing_extensions import Self
 
 import prefect
 import prefect.server.api as api
-import prefect.server.services as services
 import prefect.settings
 from prefect.client.constants import SERVER_API_VERSION
 from prefect.logging import get_logger
 from prefect.server.api.dependencies import EnforceMinimumAPIVersion
-from prefect.server.events import stream
-from prefect.server.events.services.actions import Actions
-from prefect.server.events.services.event_persister import EventPersister
-from prefect.server.events.services.triggers import ProactiveTriggers, ReactiveTriggers
 from prefect.server.exceptions import ObjectNotFoundError
-from prefect.server.services.base import Service
-from prefect.server.services.task_run_recorder import TaskRunRecorder
+from prefect.server.services.base import RunInAllServers, Service
 from prefect.server.utilities.database import get_dialect
 from prefect.settings import (
     PREFECT_API_DATABASE_CONNECTION_URL,
@@ -606,88 +600,19 @@ def create_app(
         async with session:
             await run_block_auto_registration(session=session)
 
-    async def start_services():
-        """Start additional services when the Prefect REST API starts up."""
-
-        service_instances: list[Service] = []
-
-        # these services are for events and are not implemented as loop services right now
-        if prefect.settings.PREFECT_API_SERVICES_TASK_RUN_RECORDER_ENABLED:
-            service_instances.append(TaskRunRecorder())
-
-        if prefect.settings.PREFECT_API_SERVICES_EVENT_PERSISTER_ENABLED:
-            service_instances.append(EventPersister())
-
-        if prefect.settings.PREFECT_API_EVENTS_STREAM_OUT_ENABLED:
-            service_instances.append(stream.Distributor())
-
-        if prefect.settings.PREFECT_API_SERVICES_TRIGGERS_ENABLED.value():
-            service_instances.append(ReactiveTriggers())
-            service_instances.append(ProactiveTriggers())
-            service_instances.append(Actions())
-
-        if (
-            not webserver_only
-            and prefect.settings.PREFECT_SERVER_ANALYTICS_ENABLED.value()
-        ):
-            service_instances.append(services.telemetry.Telemetry())
-
-        # don't run services in ephemeral mode
-        if not ephemeral and not webserver_only:
-            if prefect.settings.PREFECT_API_SERVICES_SCHEDULER_ENABLED.value():
-                service_instances.append(services.scheduler.Scheduler())
-                service_instances.append(
-                    services.scheduler.RecentDeploymentsScheduler()
-                )
-
-            if prefect.settings.PREFECT_API_SERVICES_LATE_RUNS_ENABLED.value():
-                service_instances.append(services.late_runs.MarkLateRuns())
-
-            if prefect.settings.PREFECT_API_SERVICES_PAUSE_EXPIRATIONS_ENABLED.value():
-                service_instances.append(services.pause_expirations.FailExpiredPauses())
-
-            if prefect.settings.PREFECT_API_SERVICES_CANCELLATION_CLEANUP_ENABLED.value():
-                service_instances.append(
-                    services.cancellation_cleanup.CancellationCleanup()
-                )
-
-            if prefect.settings.PREFECT_API_SERVICES_FLOW_RUN_NOTIFICATIONS_ENABLED.value():
-                service_instances.append(
-                    services.flow_run_notifications.FlowRunNotifications()
-                )
-
-            if prefect.settings.PREFECT_API_SERVICES_FOREMAN_ENABLED.value():
-                service_instances.append(services.foreman.Foreman())
-
-        loop = asyncio.get_running_loop()
-
-        service_tasks: dict[Service, asyncio.Task[None]] = {}
-        for service in service_instances:
-            service_tasks[service] = loop.create_task(service.start())
-        app.state.services = service_tasks
-
-        for service, task in app.state.services.items():
-            logger.info(f"{service.name} service scheduled to start in-app")
-            task.add_done_callback(partial(on_service_exit, service))
-
-    async def stop_services():
-        """Ensure services are stopped before the Prefect REST API shuts down."""
-        if hasattr(app.state, "services") and app.state.services:
-            service_tasks: dict[Service, asyncio.Task[None]] = app.state.services
-            await asyncio.gather(*[service.stop() for service in service_tasks])
-            await asyncio.gather(*service_tasks.values(), return_exceptions=True)
-
     @asynccontextmanager
-    async def lifespan(app: FastAPI):
+    async def lifespan(app: FastAPI) -> AsyncGenerator[None, None]:
         if app not in LIFESPAN_RAN_FOR_APP:
-            try:
-                await run_migrations()
-                await add_block_types()
-                await start_services()
+            await run_migrations()
+            await add_block_types()
+
+            Services: type[Service] = Service
+            if ephemeral or webserver_only:
+                Services = RunInAllServers
+
+            async with Services.running():
                 LIFESPAN_RAN_FOR_APP.add(app)
                 yield
-            finally:
-                await stop_services()
         else:
             yield
 

{prefect-3.1.16.dev2 → prefect-3.1.16.dev4}/src/prefect/server/database/_migrations/MIGRATION-NOTES.md

@@ -8,6 +8,10 @@ Each time a database migration is written, an entry is included here with:
 
 This gives us a history of changes and will create merge conflicts if two migrations are made at once, flagging situations where a branch needs to be updated before merging.
 
+# Add parameters column to deployment schedules
+SQLite: `67f886da208e`
+Postgres: `c163acd7e8e3`
+
 # Bring ORM models and migrations back in sync
 SQLite: `a49711513ad4`
 Postgres: `5d03c01be85e`

prefect-3.1.16.dev4/src/prefect/server/database/_migrations/versions/postgresql/2025_02_03_125146_c163acd7e8e3_add_parameters_column_to_deployment_.py

@@ -0,0 +1,34 @@
+"""Add parameters column to deployment schedules
+
+Revision ID: c163acd7e8e3
+Revises: 5d03c01be85e
+Create Date: 2025-02-03 12:51:46.641928
+
+"""
+
+import sqlalchemy as sa
+from alembic import op
+
+import prefect
+
+# revision identifiers, used by Alembic.
+revision = "c163acd7e8e3"
+down_revision = "5d03c01be85e"
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    op.add_column(
+        "deployment_schedule",
+        sa.Column(
+            "parameters",
+            prefect.server.utilities.database.JSON,
+            server_default="{}",
+            nullable=False,
+        ),
+    )
+
+
+def downgrade():
+    op.drop_column("deployment_schedule", "parameters")

prefect-3.1.16.dev4/src/prefect/server/database/_migrations/versions/sqlite/2025_02_03_125228_67f886da208e_add_parameters_column_to_deployment_.py

@@ -0,0 +1,34 @@
+"""Add parameters column to deployment schedules
+
+Revision ID: 67f886da208e
+Revises: a49711513ad4
+Create Date: 2025-02-03 12:52:28.370958
+
+"""
+
+import sqlalchemy as sa
+from alembic import op
+
+import prefect
+
+# revision identifiers, used by Alembic.
+revision = "67f886da208e"
+down_revision = "a49711513ad4"
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    op.add_column(
+        "deployment_schedule",
+        sa.Column(
+            "parameters",
+            prefect.server.utilities.database.JSON,
+            server_default="{}",
+            nullable=False,
+        ),
+    )
+
+
+def downgrade():
+    op.drop_column("deployment_schedule", "parameters")

{prefect-3.1.16.dev2 → prefect-3.1.16.dev4}/src/prefect/server/database/orm_models.py

@@ -784,6 +784,9 @@ class DeploymentSchedule(Base):
     )
     active: Mapped[bool] = mapped_column(default=True)
     max_scheduled_runs: Mapped[Optional[int]]
+    parameters: Mapped[dict[str, Any]] = mapped_column(
+        JSON, server_default="{}", default=dict, nullable=False
+    )
 
 
 class Deployment(Base):

{prefect-3.1.16.dev2 → prefect-3.1.16.dev4}/src/prefect/server/events/services/actions.py

@@ -5,7 +5,7 @@ from typing import TYPE_CHECKING, NoReturn
 
 from prefect.logging import get_logger
 from prefect.server.events import actions
-from prefect.server.services.base import Service
+from prefect.server.services.base import RunInAllServers, Service
 from prefect.server.utilities.messaging import Consumer, create_consumer
 from prefect.settings.context import get_current_settings
 from prefect.settings.models.server.services import ServicesBaseSetting
@@ -16,7 +16,7 @@ if TYPE_CHECKING:
 logger: "logging.Logger" = get_logger(__name__)
 
 
-class Actions(Service):
+class Actions(RunInAllServers, Service):
     """Runs the actions triggered by automations"""
 
     consumer_task: asyncio.Task[None] | None = None

{prefect-3.1.16.dev2 → prefect-3.1.16.dev4}/src/prefect/server/events/services/event_logger.py

@@ -8,7 +8,7 @@ import rich
 
 from prefect.logging import get_logger
 from prefect.server.events.schemas.events import ReceivedEvent
-from prefect.server.services.base import Service
+from prefect.server.services.base import RunInAllServers, Service
 from prefect.server.utilities.messaging import Consumer, Message, create_consumer
 from prefect.settings.context import get_current_settings
 from prefect.settings.models.server.services import ServicesBaseSetting
@@ -19,7 +19,7 @@ if TYPE_CHECKING:
 logger: "logging.Logger" = get_logger(__name__)
 
 
-class EventLogger(Service):
+class EventLogger(RunInAllServers, Service):
     """A debugging service that logs events to the console as they arrive."""
 
     consumer_task: asyncio.Task[None] | None = None

{prefect-3.1.16.dev2 → prefect-3.1.16.dev4}/src/prefect/server/events/services/event_persister.py

@@ -17,7 +17,7 @@ from prefect.logging import get_logger
 from prefect.server.database import provide_database_interface
 from prefect.server.events.schemas.events import ReceivedEvent
 from prefect.server.events.storage.database import write_events
-from prefect.server.services.base import Service
+from prefect.server.services.base import RunInAllServers, Service
 from prefect.server.utilities.messaging import (
     Consumer,
     Message,
@@ -38,7 +38,7 @@ if TYPE_CHECKING:
 logger: "logging.Logger" = get_logger(__name__)
 
 
-class EventPersister(Service):
+class EventPersister(RunInAllServers, Service):
     """A service that persists events to the database as they arrive."""
 
     consumer_task: asyncio.Task[None] | None = None

{prefect-3.1.16.dev2 → prefect-3.1.16.dev4}/src/prefect/server/events/services/triggers.py

@@ -5,7 +5,7 @@ from typing import TYPE_CHECKING, Any, NoReturn, Optional
 
 from prefect.logging import get_logger
 from prefect.server.events import triggers
-from prefect.server.services.base import LoopService, Service
+from prefect.server.services.base import LoopService, RunInAllServers, Service
 from prefect.server.utilities.messaging import Consumer, create_consumer
 from prefect.settings import PREFECT_EVENTS_PROACTIVE_GRANULARITY
 from prefect.settings.context import get_current_settings
@@ -18,7 +18,7 @@ if TYPE_CHECKING:
 logger: "logging.Logger" = get_logger(__name__)
 
 
-class ReactiveTriggers(Service):
+class ReactiveTriggers(RunInAllServers, Service):
     """Evaluates reactive automation triggers"""
 
     consumer_task: asyncio.Task[None] | None = None
@@ -52,7 +52,7 @@ class ReactiveTriggers(Service):
         logger.debug("Reactive triggers stopped")
 
 
-class ProactiveTriggers(LoopService):
+class ProactiveTriggers(RunInAllServers, LoopService):
     """Evaluates proactive automation triggers"""
 
     @classmethod

{prefect-3.1.16.dev2 → prefect-3.1.16.dev4}/src/prefect/server/events/stream.py

@@ -16,7 +16,10 @@ from typing import (
 from prefect.logging import get_logger
 from prefect.server.events.filters import EventFilter
 from prefect.server.events.schemas.events import ReceivedEvent
+from prefect.server.services.base import RunInAllServers, Service
 from prefect.server.utilities import messaging
+from prefect.settings.context import get_current_settings
+from prefect.settings.models.server.services import ServicesBaseSetting
 
 if TYPE_CHECKING:
     import logging
@@ -133,9 +136,21 @@ async def stop_distributor() -> None:
         pass
 
 
-class Distributor:
+class Distributor(RunInAllServers, Service):
     name: str = "Distributor"
 
+    @classmethod
+    def service_settings(cls) -> ServicesBaseSetting:
+        raise NotImplementedError("Distributor does not have settings")
+
+    @classmethod
+    def environment_variable_name(cls) -> dict[str, str]:
+        return "PREFECT_API_EVENTS_STREAM_OUT_ENABLED"
+
+    @classmethod
+    def enabled(cls) -> bool:
+        return get_current_settings().server.events.stream_out_enabled
+
     async def start(self) -> None:
         await start_distributor()
         try:

{prefect-3.1.16.dev2 → prefect-3.1.16.dev4}/src/prefect/server/services/base.py

@@ -5,10 +5,11 @@ import asyncio
 import inspect
 import signal
 from abc import ABC, abstractmethod
+from contextlib import asynccontextmanager
 from logging import Logger
 from operator import methodcaller
 from types import ModuleType
-from typing import Any, List, NoReturn, Optional, Sequence, overload
+from typing import Any, AsyncGenerator, List, NoReturn, Optional, Sequence, overload
 
 import anyio
 from typing_extensions import Self
@@ -22,9 +23,12 @@ from prefect.utilities.processutils import (
     _register_signal,  # type: ignore[reportPrivateUsage]
 )
 
+logger: Logger = get_logger(__name__)
+
 
 def _known_service_modules() -> list[ModuleType]:
     """Get list of Prefect server modules containing Service subclasses"""
+    from prefect.server.events import stream
     from prefect.server.events.services import actions, triggers
     from prefect.server.services import (
         cancellation_cleanup,
@@ -48,6 +52,7 @@ def _known_service_modules() -> list[ModuleType]:
         telemetry,
         triggers,
         actions,
+        stream,
     ]
 
 
@@ -89,6 +94,32 @@ class Service(ABC):
         """Get list of enabled service classes"""
         return [svc for svc in cls.all_services() if svc.enabled()]
 
+    @classmethod
+    @asynccontextmanager
+    async def running(cls) -> AsyncGenerator[None, None]:
+        """A context manager that runs enabled services on entry and stops them on
+        exit."""
+        service_tasks: dict[Service, asyncio.Task[None]] = {}
+        for service_class in cls.enabled_services():
+            service = service_class()
+            service_tasks[service] = asyncio.create_task(service.start())
+
+        try:
+            yield
+        finally:
+            await asyncio.gather(*[service.stop() for service in service_tasks])
+            await asyncio.gather(*service_tasks.values(), return_exceptions=True)
+
+    @classmethod
+    async def run_services(cls) -> NoReturn:
+        """Run enabled services until cancelled."""
+        async with cls.running():
+            heat_death_of_the_universe = asyncio.get_running_loop().create_future()
+            try:
+                await heat_death_of_the_universe
+            except asyncio.CancelledError:
+                logger.info("Received cancellation, stopping services...")
+
     @abstractmethod
     async def start(self) -> NoReturn:
         """Start running the service, which may run indefinitely"""
@@ -104,6 +135,15 @@ class Service(ABC):
         self.logger = get_logger(f"server.services.{self.name.lower()}")
 
 
+class RunInAllServers(Service, abc.ABC):
+    """
+    A marker class for services that should run in all server processes, both the
+    web server and the standalone services process.
+    """
+
+    pass
+
+
 class LoopService(Service, abc.ABC):
     """
     Loop services are relatively lightweight maintenance routines that need to run

{prefect-3.1.16.dev2 → prefect-3.1.16.dev4}/src/prefect/server/services/task_run_recorder.py

@@ -19,7 +19,7 @@ from prefect.server.events.ordering import CausalOrdering, EventArrivedEarly
 from prefect.server.events.schemas.events import ReceivedEvent
 from prefect.server.schemas.core import TaskRun
 from prefect.server.schemas.states import State
-from prefect.server.services.base import Service
+from prefect.server.services.base import RunInAllServers, Service
 from prefect.server.utilities.messaging import (
     Consumer,
     Message,
@@ -217,7 +217,7 @@ async def consumer() -> AsyncGenerator[MessageHandler, None]:
     yield message_handler
 
 
-class TaskRunRecorder(Service):
+class TaskRunRecorder(RunInAllServers, Service):
     """Constructs task runs and states from client-emitted events"""
 
     consumer_task: asyncio.Task[None] | None = None