Skip to content

Commit

Permalink
Start new job API
Browse files Browse the repository at this point in the history
  • Loading branch information
Sebastien Deschambault committed Aug 4, 2023
1 parent dcb0b3e commit 567dfba
Show file tree
Hide file tree
Showing 2 changed files with 27 additions and 0 deletions.
7 changes: 7 additions & 0 deletions src/saturn_engine/core/api.py
Original file line number Diff line number Diff line change
Expand Up @@ -177,6 +177,13 @@ class JobInput:
error: Optional[str] = None


@dataclasses.dataclass
class StartJobInput:
name: JobId
queue_name: str
job_definition_name: str


@dataclasses.dataclass
class JobsResponse(ListResponse[JobItem]):
items: list[JobItem]
Expand Down
20 changes: 20 additions & 0 deletions src/saturn_engine/worker_manager/api/jobs.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,9 +8,11 @@
from saturn_engine.core.api import JobsStatesSyncInput
from saturn_engine.core.api import JobsStatesSyncResponse
from saturn_engine.core.api import JobsSyncResponse
from saturn_engine.core.api import StartJobInput
from saturn_engine.core.api import UpdateResponse
from saturn_engine.database import session_scope
from saturn_engine.stores import jobs_store
from saturn_engine.utils import utcnow
from saturn_engine.utils.flask import Json
from saturn_engine.utils.flask import abort
from saturn_engine.utils.flask import check_found
Expand Down Expand Up @@ -93,3 +95,21 @@ def post_fetch_states() -> Json[FetchCursorsStatesResponse]:
session=session,
)
return jsonify(FetchCursorsStatesResponse(cursors=cursors))


@bp.route("/_start", methods=("POST",))
def post_start_job() -> Json[JobResponse]:
start_input = marshall_request(StartJobInput)
with session_scope() as session:
job = jobs_store.get_job(session=session, name=start_input.name)
if job:
jobs_store.update_job(
session=session, name=start_input.name, completed_at=utcnow()
)
new_job = jobs_store.create_job(
session=session,
name=start_input.name,
queue_name=start_input.queue_name,
job_definition_name=start_input.job_definition_name,
)
return jsonify(JobResponse(data=new_job.as_core_item()))

0 comments on commit 567dfba

Please sign in to comment.