Skip to content

Commit

Permalink
pause all service on startup
Browse files Browse the repository at this point in the history
  • Loading branch information
solarw committed Jun 18, 2024
1 parent 49b034e commit ab5347d
Show file tree
Hide file tree
Showing 2 changed files with 19 additions and 2 deletions.
17 changes: 17 additions & 0 deletions operate/cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -179,6 +179,23 @@ def cancel_funding_job(service: str) -> None:
if not status:
logger.info(f"Funding job cancellation for {service} failed")

def pause_all_services_on_startup():
logger.info(f"stopping services on startup")
services = [i["hash"] for i in operate.service_manager().json]

for service in services:
if not operate.service_manager().exists(service=service):
continue
logger.info(f"stopping service {service}")
deployment = operate.service_manager().create_or_load(service).deployment
deployment.stop(force=True)
logger.info(f"Cancelling funding job for {service}")
cancel_funding_job(service=service)
logger.info(f"stopping services on startup: done")

# on backend app started we assume there are now started agents, so we force to pause all
pause_all_services_on_startup()

app = FastAPI()

app.add_middleware(
Expand Down
4 changes: 2 additions & 2 deletions operate/services/service.py
Original file line number Diff line number Diff line change
Expand Up @@ -822,9 +822,9 @@ def start(self, use_docker: bool = False) -> None:
self.status = DeploymentStatus.DEPLOYED
self.store()

def stop(self, use_docker: bool = False) -> None:
def stop(self, use_docker: bool = False, force: bool=False) -> None:
"""Stop the deployment."""
if self.status != DeploymentStatus.DEPLOYED:
if self.status != DeploymentStatus.DEPLOYED and not force:
return

self.status = DeploymentStatus.STOPPING
Expand Down

0 comments on commit ab5347d

Please sign in to comment.