chore(plan): add job-registry + consistency check
purpose is to design a job registry with a scheduler thi aim to drive in the same way any services requiring execution of periodic background tasks plugins could benefeciate it purpose is mostly to add a normative way to implement consistency check per services this is to edutcate implementors adding any new services goal is to ensure data quality with oxicloud and resumable jobs by default
This commit is contained in:
@@ -0,0 +1,798 @@
|
||||
# Plan — Job engines (periodic + recoverable) + admin surface
|
||||
|
||||
## Context
|
||||
|
||||
OxiCloud runs several fire-and-forget background daemons today, each
|
||||
spawned by a service factory in `src/common/di.rs` at startup:
|
||||
|
||||
| Service | Cadence | Shape |
|
||||
|---|---|---|
|
||||
| `TrashCleanupService` | every 24 h | Fixed interval, no per-run state |
|
||||
| `StorageUsageService::start_reconciliation_job` | every 600 s | Fixed interval, no per-run state |
|
||||
| `db_pool_monitor` | every N s | Fixed interval, no per-run state |
|
||||
| `dedup_service` GC | on demand + inline | Fixed interval, no per-run state |
|
||||
| `GrantCleanupService` | every 24 h | Fixed interval, no per-run state |
|
||||
| `tree_etag_flush_job` | every ~500 ms | Fixed interval, no per-run state |
|
||||
| `content_index` worker | continuous | Fixed interval, no per-run state |
|
||||
| Blob storage backend migration | admin-triggered | Long-running, cursor, resumable, in-memory state today |
|
||||
| `admin/audio/metadata/reextract` | admin-triggered | Long-running, blocks HTTP request today |
|
||||
| `admin/photos/metadata/reextract` | admin-triggered | Long-running, blocks HTTP request today |
|
||||
| `ConsistencyCheck` runs (see `docs/plan/consistency-check.md`) | admin-triggered v1 | Long-running, cursor, resumable, needs DB state |
|
||||
|
||||
Two shapes bleed together in the current codebase but shouldn't. Each
|
||||
daemon reinvents its own env var pattern, admin trigger endpoint,
|
||||
logging schema, and (for the long-running ones) its own in-memory
|
||||
progress state that vanishes on restart.
|
||||
|
||||
## Two engines, one file
|
||||
|
||||
This plan is intentionally two plans in one file (Ed 2026-07-27),
|
||||
because the two engines share an admin URL prefix, a config-var
|
||||
convention, and a logging target — but nothing else:
|
||||
|
||||
- **Part 1 — Periodic Scheduler.** In-memory registration + tokio
|
||||
interval loop. Serves fixed-interval jobs an operator might trigger
|
||||
manually. No DB tables, no cursor, no per-run persistence.
|
||||
- **Part 2 — Recoverable-Run Engine.** DB-backed cursor persistence +
|
||||
exclusivity + crash recovery. Serves the four long-running tenants
|
||||
(storage-migration, reextract-audio, reextract-image, consistency
|
||||
check runs) and any future work that iterates over a large space
|
||||
with restart tolerance.
|
||||
|
||||
A recoverable job CAN optionally be periodically-triggered (register
|
||||
once in each engine; Part 1's tick calls Part 2's `run_or_resume`
|
||||
instead of a bare handler). Most Layer B tenants are admin-triggered
|
||||
only.
|
||||
|
||||
Cross-cutting concerns (admin URL taxonomy, env vars, logging target,
|
||||
plugin future) live in a shared section at the bottom so we're not
|
||||
duplicating them between parts.
|
||||
|
||||
## Migration criterion — the trigger question
|
||||
|
||||
Not every background loop belongs in JobRegistry. The single question
|
||||
that decides:
|
||||
|
||||
> **"Would an operator plausibly `POST /trigger-job/{name}` to make
|
||||
> it run right now?"**
|
||||
|
||||
**Yes → migrate.** The whole payoff of JobRegistry is a uniform
|
||||
*operator surface* — list, trigger, last-outcome, log line, config
|
||||
knobs. If nobody would ever manually trigger the job, the surface
|
||||
delivers no value; you're paying framework overhead for nothing.
|
||||
Anything an operator would manually trigger is by definition
|
||||
periodic + discrete + meaningful.
|
||||
|
||||
**No → leave it as its own loop.** Continuous drains and
|
||||
event-reactive workers ("core workers") fail this test — "trigger
|
||||
the content-index worker" makes no sense; it's already running.
|
||||
Standardise their env var naming and log target as a light
|
||||
convention (see [Cross-cutting](#cross-cutting) below) but do NOT
|
||||
wedge them into the scheduler.
|
||||
|
||||
Secondary confirmation questions — if the primary is yes and any of
|
||||
these is no, migrate anyway but flag the mismatch:
|
||||
|
||||
1. Does each invocation report a meaningful `count` (rows swept,
|
||||
blobs GC'd, bytes reclaimed)? Continuous workers don't have
|
||||
discrete invocations to count.
|
||||
2. Does the operator tune it via env vars beyond enable/disable?
|
||||
3. Would an operator want a "did this run within the last N?" health
|
||||
signal? Periodic jobs benefit from `last_outcome`; always-on
|
||||
workers need liveness signals of a different shape.
|
||||
|
||||
**Cadence is NOT the trigger** — it's a symptom. Sub-second jobs
|
||||
almost always fail the primary question (nobody manually triggers
|
||||
something that fires 2× per second), but a hypothetical 1 s periodic
|
||||
job that operators do want to kick still belongs in JobRegistry.
|
||||
Cadence tells you "probably no"; the operator-trigger question is
|
||||
what decides.
|
||||
|
||||
### Applied to the current daemons
|
||||
|
||||
| Service | Operator-trigger? | Destination |
|
||||
|---|---|---|
|
||||
| `TrashCleanupService` | Yes — "purge expired trash now" | Part 1 |
|
||||
| `StorageUsageService::start_reconciliation_job` | Yes — "recompute quotas now" | Part 1 |
|
||||
| `dedup_service` GC | Yes — already has `trigger-gc` | Part 1 |
|
||||
| `GrantCleanupService` | Yes — already has `trigger-grant-cleanup` | Part 1 |
|
||||
| `tree_etag_flush_job` | No — a "flush now" is meaningless (queue drains itself) | Core worker, unchanged |
|
||||
| `content_index` worker | No — continuous drain, no discrete invocation | Core worker, unchanged |
|
||||
| `db_pool_monitor` | No — "log stats now" is either grep-existing-logs or attach-a-debugger, not a scheduled job trigger | Core worker, unchanged |
|
||||
| Blob storage backend migration | Yes — already admin-triggered | Part 2 |
|
||||
| `admin/audio/metadata/reextract` | Yes — currently admin-triggered (synchronously) | Part 2 |
|
||||
| `admin/photos/metadata/reextract` | Yes — currently admin-triggered (synchronously) | Part 2 |
|
||||
| `ConsistencyCheck` runs | Yes — needs a trigger endpoint | Part 2 |
|
||||
|
||||
The `db_pool_monitor` case is illustrative: cadence-wise it *could*
|
||||
fit Part 1 (10-30 s periodic, bounded work), but the operator-trigger
|
||||
question kills it. Nobody manually triggers a stats-log because logs
|
||||
are already there. Keeping it as its own loop is right.
|
||||
|
||||
## Implementation order
|
||||
|
||||
1. **Part 1 lands first** — small, self-contained, unblocks migration
|
||||
of trash-cleanup + storage-usage + db_pool_monitor + dedup GC +
|
||||
grant-cleanup + tree-etag flush + content-index. High mechanical
|
||||
payoff, zero new schema, minimal review surface.
|
||||
2. **Part 2 lands next** — introduces `admin.background_runs` schema
|
||||
+ `RecoverableJob` trait + `JobStore` port + `run_or_resume`. On
|
||||
its own PR (schema change deserves independent review).
|
||||
3. **Consistency-check framework (`docs/plan/consistency-check.md`)**
|
||||
lands third, consuming Part 2 as its runtime.
|
||||
4. **Storage-migration and reextract-* migrated to Part 2** as
|
||||
follow-ups.
|
||||
|
||||
---
|
||||
|
||||
## Part 1 — Periodic Scheduler
|
||||
|
||||
### Contract — `JobHandler` trait
|
||||
|
||||
The implementor-facing surface for a fixed-interval job:
|
||||
|
||||
```rust
|
||||
#[async_trait]
|
||||
pub trait JobHandler: Send + Sync {
|
||||
/// Stable snake_case identifier. Must be unique across the process.
|
||||
/// Log lines, admin listing, env vars, and trigger URLs all key on
|
||||
/// this name.
|
||||
fn name(&self) -> &str;
|
||||
|
||||
/// One execution. Called by the supervisor at the registered
|
||||
/// interval and (optionally) on admin trigger. Return `Ok { count,
|
||||
/// extra }` on success — the count is the primary scalar the job
|
||||
/// reports (rows swept, ETags flushed, blobs GC'd). Return
|
||||
/// `Err(msg)` on failure; the supervisor logs it and continues.
|
||||
async fn run(&self) -> JobOutcome;
|
||||
}
|
||||
```
|
||||
|
||||
Native services implement this trait on an existing service type (no
|
||||
new wrapper) and register a single `Arc<dyn JobHandler>` with the
|
||||
scheduler.
|
||||
|
||||
### `JobOutcome`
|
||||
|
||||
```rust
|
||||
pub enum JobOutcome {
|
||||
Ok { count: u64, extra: serde_json::Value },
|
||||
Err(String),
|
||||
}
|
||||
```
|
||||
|
||||
Two variants only. Every reason a run can fail (handler returned an
|
||||
error, wall-clock timeout, panic caught by the supervisor) collapses
|
||||
to `Err(String)`, with the *cause* encoded in the message AND in a
|
||||
`cause` tracing field the supervisor sets when it emits the log line:
|
||||
|
||||
- Handler returned `Err(msg)` → `cause = "handler"`, message = `msg`.
|
||||
- `tokio::time::timeout` tripped → `cause = "timeout"`.
|
||||
- `catch_unwind` caught a panic → `cause = "panicked"`, message = the
|
||||
payload as a string.
|
||||
|
||||
Handlers never construct the cause themselves; they either return
|
||||
`Ok { count, extra }` or `Err(String)`. Keeping the enum to two
|
||||
variants prevents every consumer of `match outcome` from having to
|
||||
distinguish diagnostic sub-cases that behave identically for logging,
|
||||
persistence, retry, and admin display.
|
||||
|
||||
### Runtime model
|
||||
|
||||
- **One `tokio::spawn`** at startup runs the scheduler main loop.
|
||||
Sleeps until the earliest due job, dispatches, sleeps again.
|
||||
- Per-run **panic catching** via `tokio::spawn` inside the dispatch
|
||||
(or `AssertUnwindSafe` + `catch_unwind`). A bad handler crashes
|
||||
its own run, not the scheduler.
|
||||
- **Sequential dispatch within a tick** by default. Two jobs due at
|
||||
the same instant run one after the other. Parallel dispatch can
|
||||
layer on later as a per-job toggle if a real need appears — most
|
||||
handlers touch the DB and don't benefit from concurrency.
|
||||
- **`ScheduledJob.timeout: Option<Duration>`** is applied by the
|
||||
supervisor via `tokio::time::timeout` when set. Optional; use it
|
||||
when the handler has a real wall-clock bound. None means "let it
|
||||
run to completion."
|
||||
|
||||
Single supervisor is chosen for **operational** clarity, not runtime
|
||||
cost: one place to observe, one panic-containment boundary, one
|
||||
config surface, one plugin-registration hook when plugins land.
|
||||
|
||||
### Exclusivity — one in-flight run per `job_name`
|
||||
|
||||
Mirrors Part 2's exclusivity invariant, enforced in-memory since
|
||||
Part 1 has no DB row:
|
||||
|
||||
- Each `RegisteredJob` carries an `is_running` flag (an
|
||||
`AtomicBool` or single-permit `Semaphore`).
|
||||
- Before dispatching a tick, the supervisor tries to acquire the
|
||||
flag. If it's already held (the previous run is still executing),
|
||||
the tick is **skipped, not queued**:
|
||||
|
||||
```rust
|
||||
tracing::warn!(
|
||||
target: "oxicloud::scheduler",
|
||||
event = "job.tick_skipped",
|
||||
job = %name,
|
||||
interval_ms = interval.as_millis(),
|
||||
running_for_ms = current_run_start.elapsed().as_millis(),
|
||||
"{name} still running past its interval — tick skipped"
|
||||
);
|
||||
```
|
||||
|
||||
`next_run_at` advances by one interval so the schedule stays on
|
||||
its cadence rather than queueing backlog.
|
||||
- On completion (or panic caught by the supervisor), the flag is
|
||||
released. The next tick is free to fire.
|
||||
- **Diagnostic value.** A `job.tick_skipped` line on every interval
|
||||
is the operator signal that either the job is chronically slower
|
||||
than its cadence (retune the interval) or hung (attach a debugger
|
||||
/ set a timeout / kill the process). Without this warning a slow
|
||||
or hung handler would silently starve.
|
||||
- **Interaction with timeout.** If a job has a `timeout` configured
|
||||
and it trips, the supervisor kills the run and releases the flag.
|
||||
Timeouts prevent hangs from permanently silencing a job.
|
||||
Handlers without a timeout can, in principle, hang forever — the
|
||||
repeated `tick_skipped` warning is the only signal.
|
||||
|
||||
Cross-job concurrency is unchanged — different `job_name`s can run
|
||||
sequentially per tick as described above. Exclusivity is per
|
||||
job_name, not global.
|
||||
|
||||
### `JobRegistry`
|
||||
|
||||
```rust
|
||||
pub struct JobRegistry {
|
||||
jobs: RwLock<HashMap<String, RegisteredJob>>,
|
||||
}
|
||||
|
||||
struct RegisteredJob {
|
||||
handler: Arc<dyn JobHandler>,
|
||||
interval: Duration,
|
||||
timeout: Option<Duration>,
|
||||
/// Single-permit gate that enforces the "one in-flight run per
|
||||
/// `job_name`" invariant (see Exclusivity above). A tick that
|
||||
/// finds the permit taken emits `job.tick_skipped` and does not
|
||||
/// spawn.
|
||||
in_flight: Arc<tokio::sync::Semaphore>, // capacity = 1
|
||||
/// Set when a run starts, cleared when it ends. Used to include
|
||||
/// `running_for_ms` in the skip warning.
|
||||
current_run_start: Arc<parking_lot::Mutex<Option<Instant>>>,
|
||||
last_outcome: Option<(chrono::DateTime<Utc>, JobOutcome)>,
|
||||
next_run_at: chrono::DateTime<Utc>,
|
||||
}
|
||||
```
|
||||
|
||||
`Arc<JobRegistry>` lives on `AppState`. Native services register
|
||||
themselves during DI:
|
||||
|
||||
```rust
|
||||
registry.register(
|
||||
Arc::clone(&trash_cleanup) as Arc<dyn JobHandler>,
|
||||
Duration::from_secs(interval_hours * 3600),
|
||||
None, // no timeout
|
||||
);
|
||||
```
|
||||
|
||||
### Engine loop
|
||||
|
||||
```rust
|
||||
async fn run(registry: Arc<JobRegistry>) {
|
||||
loop {
|
||||
let next = registry.pick_next().await; // earliest next_run_at
|
||||
let sleep = next.deadline().saturating_duration_since(Instant::now());
|
||||
tokio::time::sleep(sleep).await;
|
||||
|
||||
let outcome = registry.dispatch(&next.name).await;
|
||||
registry.record_outcome(&next.name, outcome).await;
|
||||
}
|
||||
}
|
||||
```
|
||||
|
||||
`dispatch` grabs the handler under a read lock, spawns a task, applies
|
||||
the timeout, catches panics, and returns the `JobOutcome`. Sequential
|
||||
dispatch is intentional; two jobs due at the same instant run
|
||||
one-after-the-other.
|
||||
|
||||
### Native tenants and migration order
|
||||
|
||||
Four services satisfy the operator-trigger criterion above and migrate:
|
||||
|
||||
1. **trash-cleanup** — simplest self-contained loop; reference for the
|
||||
migration shape. Ships with Part 1's landing PR.
|
||||
2. **storage-usage reconciliation** — same shape, different service.
|
||||
3. **dedup GC** — already has `trigger-gc`; the shim forwards to the
|
||||
new registry-backed trigger.
|
||||
4. **grant-cleanup** — already has `trigger-grant-cleanup`; same shim
|
||||
pattern.
|
||||
|
||||
Three services are **core workers** and STAY on their own loops
|
||||
(fail the operator-trigger question — see the criterion table above):
|
||||
|
||||
- `tree_etag_flush_job` — 500 ms queue-drain, coalescing semantics.
|
||||
- `content_index` worker — continuous channel drain, event-reactive.
|
||||
- `db_pool_monitor` — periodic stats-log with no discrete-invocation
|
||||
count and no operator use for manual trigger.
|
||||
|
||||
Standardise their env var naming (`OXICLOUD_JOB_<NAME>_*`) and
|
||||
tracing target for uniform operator ergonomics, but do NOT wedge them
|
||||
into the scheduler.
|
||||
|
||||
### Verification (Part 1)
|
||||
|
||||
1. **Compile**: `cargo check --all-features --all-targets` +
|
||||
`cargo clippy -- -D warnings` clean.
|
||||
2. **Boot**: start server; expect `scheduler started, N job(s) registered`.
|
||||
3. **Admin listing**:
|
||||
```
|
||||
curl -s http://localhost:8086/api/admin/internal/jobs -H "Authorization: Bearer $TOKEN"
|
||||
```
|
||||
returns a JSON array with each registered job, its `interval_ms`,
|
||||
`next_run_at`, and `last_outcome` (null until first tick).
|
||||
4. **Trigger**: `POST /api/admin/internal/trigger-job/trash_cleanup`
|
||||
invokes the handler immediately, records the outcome.
|
||||
5. **Panic containment**: unit test a handler that panics; `last_outcome`
|
||||
records `Err(...)` with `cause = "panicked"` in the log; the scheduler
|
||||
is still alive (verified by triggering another job); the in-flight
|
||||
permit is released so the next tick can fire.
|
||||
6. **Timeout enforcement**: unit test a handler that blocks longer than
|
||||
its declared timeout; `last_outcome` records `Err(...)` with
|
||||
`cause = "timeout"`; the in-flight permit is released.
|
||||
7. **Overrun exclusivity**: unit test a handler with a 100 ms interval
|
||||
that sleeps 300 ms. Assert exactly ONE run is in flight at any moment
|
||||
(no parallel dispatch), and that two `job.tick_skipped` log events
|
||||
fire (one at each missed tick) with `running_for_ms` monotonically
|
||||
increasing.
|
||||
8. **Shim compatibility**: existing per-service trigger endpoints
|
||||
(`trigger-sweep`, `trigger-gc`, `trigger-grant-cleanup`) keep working
|
||||
as thin forwards. Existing api-test Hurl suites pass unchanged.
|
||||
|
||||
---
|
||||
|
||||
## Part 2 — Recoverable-Run Engine
|
||||
|
||||
### Contract — `RecoverableJob` trait
|
||||
|
||||
Sibling to `JobHandler`, NOT a subtrait. A stateless job that only
|
||||
implements `JobHandler` never needs to know Part 2 exists.
|
||||
|
||||
```rust
|
||||
#[async_trait]
|
||||
pub trait RecoverableJob: Send + Sync {
|
||||
/// Stable snake_case identifier — matches the `job_name` column
|
||||
/// in `admin.background_runs`.
|
||||
fn name(&self) -> &str;
|
||||
|
||||
/// Long-running, cooperative scan. The store is the job's ONLY
|
||||
/// side effect: cursor checkpointing, cancel polling, run-state
|
||||
/// updates all go through it.
|
||||
///
|
||||
/// Between batches the handler MUST poll `store.status()` — a
|
||||
/// `CancelRequested` return means the operator asked for a pause
|
||||
/// and the handler should return `Paused { cursor }` at the next
|
||||
/// safe boundary. A mid-batch `tokio::spawn` abort corrupts the
|
||||
/// cursor and MUST NEVER happen — that's why the supervisor does
|
||||
/// not apply `tokio::time::timeout` to recoverable jobs (Part 1's
|
||||
/// timeout policy does not apply here).
|
||||
async fn run_resumable(&self, store: &dyn JobStore) -> RunOutcome;
|
||||
}
|
||||
```
|
||||
|
||||
### `RunOutcome`
|
||||
|
||||
```rust
|
||||
pub enum RunOutcome {
|
||||
Completed,
|
||||
Paused { cursor: Vec<u8> },
|
||||
Failed { message: String },
|
||||
}
|
||||
```
|
||||
|
||||
- `Completed` — walked the whole space. Engine writes `status = Completed`.
|
||||
- `Paused { cursor }` — cooperative pause (cancel poll or graceful
|
||||
shutdown). Engine persists cursor + writes `status = Paused` so a
|
||||
future resume picks up here.
|
||||
- `Failed { message }` — irrecoverable error. Cursor NOT advanced;
|
||||
engine writes `status = Failed` and captures the message.
|
||||
|
||||
### `JobStore` trait
|
||||
|
||||
The port the engine passes to a recoverable job. Backed by
|
||||
`admin.background_runs` in production; can be mocked for unit tests.
|
||||
|
||||
```rust
|
||||
#[async_trait]
|
||||
pub trait JobStore: Send + Sync {
|
||||
/// The `run_id` this handler was invoked with. Uniquely identifies
|
||||
/// the row in `admin.background_runs`.
|
||||
fn run_id(&self) -> Uuid;
|
||||
|
||||
/// Fixed at run start; used by consistency checks (and any other
|
||||
/// job with a grace boundary) as the reference `NOW()` — NOT
|
||||
/// `chrono::Utc::now()`, which would drift across a multi-hour
|
||||
/// scan. See `docs/plan/consistency-check.md` trap #1.
|
||||
fn started_at(&self) -> chrono::DateTime<chrono::Utc>;
|
||||
|
||||
/// Read the current `status` from the row. Between batches the
|
||||
/// handler polls this; a return of `CancelRequested` means the
|
||||
/// operator asked for a pause.
|
||||
async fn status(&self) -> Result<RunStatus, DomainError>;
|
||||
|
||||
/// The last-persisted cursor (raw bytes, per-job schema), or
|
||||
/// `None` on a fresh run. The handler decodes into its own key
|
||||
/// type (blob hash, file_id UUID, ltree path, …).
|
||||
async fn load_cursor(&self) -> Result<Option<Vec<u8>>, DomainError>;
|
||||
|
||||
/// Advance cursor + stats, bump `last_progress_at`. Called between
|
||||
/// batches, typically every ~30 s OR every ~1 000 rows, whichever
|
||||
/// comes first. See `docs/plan/consistency-check.md` trap #6.
|
||||
async fn checkpoint(&self, cursor: Vec<u8>, delta_count: u64)
|
||||
-> Result<(), DomainError>;
|
||||
}
|
||||
```
|
||||
|
||||
Domain-specific extensions (consistency-check's finding sink, for
|
||||
instance) are separate traits the impl composes on top of `JobStore`.
|
||||
`JobStore` itself carries no findings/severity concept — those are
|
||||
Layer C in the consistency-check plan, not the engine's concern.
|
||||
|
||||
### Schema — `admin.background_runs`
|
||||
|
||||
```sql
|
||||
CREATE SCHEMA IF NOT EXISTS admin;
|
||||
|
||||
CREATE TABLE admin.background_runs (
|
||||
id UUID PRIMARY KEY,
|
||||
job_name TEXT NOT NULL,
|
||||
status TEXT NOT NULL, -- Running / Paused / CancelRequested / Completed / Failed
|
||||
started_at TIMESTAMPTZ NOT NULL, -- fixed at run start
|
||||
last_progress_at TIMESTAMPTZ NOT NULL, -- heartbeat + last-checkpoint marker
|
||||
completed_at TIMESTAMPTZ,
|
||||
cursor BYTEA, -- opaque, per-job resume key (NULL = fresh)
|
||||
stats JSONB NOT NULL DEFAULT '{}'::jsonb, -- job-specific counters
|
||||
params JSONB NOT NULL DEFAULT '{}'::jsonb, -- job-specific params
|
||||
error_message TEXT
|
||||
);
|
||||
|
||||
CREATE UNIQUE INDEX one_active_run_per_job
|
||||
ON admin.background_runs (job_name)
|
||||
WHERE status IN ('Running', 'Paused', 'CancelRequested');
|
||||
|
||||
CREATE INDEX ON admin.background_runs (last_progress_at)
|
||||
WHERE status = 'Running';
|
||||
```
|
||||
|
||||
**The partial unique index is load-bearing.** It enforces the "at
|
||||
most one non-terminal run per `job_name`" invariant at the DB layer
|
||||
so it survives concurrent triggers, admin-vs-scheduler races, and
|
||||
transaction interleavings. The `CancelRequested` inclusion prevents
|
||||
a second trigger during cancel from spawning a parallel run.
|
||||
|
||||
`admin.*` is a NEW schema — kept distinct from `auth.*` / `storage.*`
|
||||
so operational tables don't pollute domain schemas. Consistency
|
||||
checks own their own `admin.consistency_findings` in the same
|
||||
schema.
|
||||
|
||||
Cursor is `BYTEA`, not JSONB, because per-job cursors are fixed-shape
|
||||
opaque keys (32-byte BLAKE3, 16-byte UUID, ltree bytes) — JSONB adds
|
||||
encoding overhead and a keying convention every impl has to agree on.
|
||||
`stats` and `params` ARE JSONB because they carry human-readable
|
||||
key/value pairs read by observability code, not compared inside SQL.
|
||||
|
||||
### Cursor semantics
|
||||
|
||||
- **`NULL` cursor** = fresh run, no rows processed yet. Handler
|
||||
interprets as "start from the beginning." Every keyset-pagination
|
||||
helper handles this as `WHERE ($1::bytea IS NULL OR key > $1)`.
|
||||
- **Non-NULL cursor** = last-processed key. On resume, `key > cursor`
|
||||
in the ORDER BY key ASC iteration.
|
||||
- **Advance rule** = handler updates its in-memory cursor to the LAST
|
||||
row it successfully processed at the end of each batch, checkpoints
|
||||
periodically. On crash: at most one batch of work replays. Idempotent
|
||||
processing (e.g. `UNIQUE (run_id, kind, resource_id)` on findings)
|
||||
makes replay a no-op for anything already recorded.
|
||||
|
||||
### Checkpoint mechanics
|
||||
|
||||
One `UPDATE` per checkpoint. Cheap, no row-lock contention (this
|
||||
process owns the row):
|
||||
|
||||
```sql
|
||||
UPDATE admin.background_runs
|
||||
SET cursor = $2,
|
||||
stats = jsonb_set(
|
||||
stats,
|
||||
'{scanned_count}',
|
||||
((COALESCE(stats->>'scanned_count','0')::bigint + $3)::text)::jsonb
|
||||
),
|
||||
last_progress_at = NOW()
|
||||
WHERE id = $1;
|
||||
```
|
||||
|
||||
- `cursor` advances to the last row we processed.
|
||||
- `stats.scanned_count` accumulates the delta — not overwritten. Each
|
||||
job's handler picks its own key names inside `stats`. There's ONE
|
||||
convention: a top-level `count` field mirroring the value carried
|
||||
in `JobOutcome::Ok.count` (see next section) — everything else is
|
||||
free-form.
|
||||
- `last_progress_at` doubles as heartbeat. Boot recovery uses it to
|
||||
spot stale-Running rows.
|
||||
|
||||
### `RunOutcome` → `JobOutcome` bridge
|
||||
|
||||
The supervisor translates so a periodic-triggered recoverable job
|
||||
records the same `JobOutcome` shape as any other tick:
|
||||
|
||||
- `Completed` → `Ok { count, extra: json!({"completed": true}) }`
|
||||
- `Paused { cursor }` → `Ok { count, extra: json!({"paused": true, "cursor_hex": …}) }`
|
||||
- `Failed { message }` → `Err(message)`
|
||||
|
||||
Paused is deliberately NOT an error — the run cooperatively yielded,
|
||||
that's a success. Log lines stay meaningful (`outcome=ok`,
|
||||
`extra.paused=true` distinguishes from full completion). Only
|
||||
`Failed` alerts an operator.
|
||||
|
||||
### `run_or_resume` helper
|
||||
|
||||
The engine module exposes:
|
||||
|
||||
```rust
|
||||
pub async fn run_or_resume<J: RecoverableJob + ?Sized>(
|
||||
job: Arc<J>,
|
||||
store_factory: &dyn JobStoreFactory,
|
||||
) -> JobOutcome
|
||||
```
|
||||
|
||||
Body:
|
||||
|
||||
1. Look up the latest row for `job.name()`.
|
||||
2. If `Completed`/`Failed` or nothing → `INSERT` a new `Running` row
|
||||
with `started_at = NOW()`, cursor NULL. On unique-index conflict
|
||||
(rare race), read the winning row and continue from step 3.
|
||||
3. If `Paused` → `UPDATE ... SET status='Running'` on that row.
|
||||
4. If `Running`/`CancelRequested` → short-circuit
|
||||
`Ok { count: 0, extra: {"skipped": "already_running"} }`.
|
||||
5. Build a `JobStore` bound to the row's `run_id` and pass it to
|
||||
`job.run_resumable(store).await`.
|
||||
6. Translate the returned `RunOutcome`, write the terminal status
|
||||
(`Completed` / `Paused` / `Failed`) with the final cursor/stats
|
||||
snapshot, return the `JobOutcome`.
|
||||
|
||||
### Concurrency policy — exclusive-by-default
|
||||
|
||||
**At most one non-terminal run per `job_name` may exist at any time.**
|
||||
Non-terminal = `status IN ('Running', 'Paused', 'CancelRequested')`.
|
||||
This is the default, not opt-in — a job runs to completion, gets
|
||||
manually paused, or fails; a second trigger while one is active
|
||||
never spawns a parallel run.
|
||||
|
||||
- A storage-migration cannot run twice at once. Neither can a
|
||||
reextract-audio, a reextract-image, or a consistency-check.
|
||||
- The registry's trigger endpoint is idempotent: called while a run
|
||||
is active it returns the existing `run_id` + status; called while
|
||||
the latest run is `Paused` it resumes it (same cursor, same stats
|
||||
accumulator); called when no non-terminal run exists it starts
|
||||
fresh.
|
||||
- The DB-level partial unique index makes the invariant impossible
|
||||
to violate even under concurrent triggers or scheduler-vs-operator
|
||||
races.
|
||||
- The scheduler's periodic tick honours the same rule — if the
|
||||
latest row for a job is non-terminal, the tick does not spawn
|
||||
another. For long-running jobs "interval" effectively means "check
|
||||
every N whether a run needs starting", not "start every N."
|
||||
- Cross-job concurrency is unchanged — different `job_name`s can
|
||||
run in parallel subject to Part 1's sequential-dispatch default.
|
||||
Exclusivity is per job_name, not global.
|
||||
|
||||
### Boot-time crashed-run recovery
|
||||
|
||||
At `AppServiceFactory` init, after DB pool is up:
|
||||
|
||||
```rust
|
||||
sqlx::query!(
|
||||
"UPDATE admin.background_runs
|
||||
SET status = 'Paused',
|
||||
error_message = COALESCE(error_message, 'server restart mid-run')
|
||||
WHERE status IN ('Running', 'CancelRequested')"
|
||||
).execute(&pool).await?;
|
||||
```
|
||||
|
||||
Do NOT auto-resume — the bug that killed the last run may still be
|
||||
present. Operators decide. The next scheduler tick (or an explicit
|
||||
trigger) resumes any `Paused` row per the normal flow.
|
||||
|
||||
Consistency-check.md's existing consistency-scoped sweep collapses
|
||||
into this general one.
|
||||
|
||||
### Admin surface (recoverable runs)
|
||||
|
||||
Same URL taxonomy as Part 1, extended for run identity:
|
||||
|
||||
```
|
||||
POST /api/admin/internal/trigger-job/{name}
|
||||
→ { run_id, status } # starts or resumes; idempotent
|
||||
POST /api/admin/internal/trigger-job/{name}/cancel
|
||||
→ { run_id, status: "CancelRequested" }
|
||||
GET /api/admin/internal/jobs/{name}/runs
|
||||
→ [{ run_id, status, started_at, last_progress_at, stats, ... }]
|
||||
GET /api/admin/internal/jobs/{name}/runs/{id}
|
||||
→ { run_id, status, cursor_hex, stats, params, error_message, ... }
|
||||
```
|
||||
|
||||
### Native tenants (Part 2)
|
||||
|
||||
- **Blob storage backend migration.** `migration_job.rs` becomes a
|
||||
`RecoverableJob` impl. Cursor = last processed blob hash. Retires
|
||||
the `Arc<RwLock<MigrationState>>` in-memory struct.
|
||||
- **Reextract audio metadata.** Currently synchronous inside the
|
||||
admin HTTP request. Becomes a `RecoverableJob` iterating audio
|
||||
files by `file_id`.
|
||||
- **Reextract image/video capture dates.** Same as above.
|
||||
- **Consistency-check runs.** Every `ConsistencyCheck` impl gets
|
||||
wrapped by a `RecoverableJob` adapter; the wrapper writes to
|
||||
`admin.background_runs` via `JobStore`, and separately writes
|
||||
findings to `admin.consistency_findings` via a check-specific
|
||||
extension trait. See `docs/plan/consistency-check.md`.
|
||||
|
||||
### Verification (Part 2)
|
||||
|
||||
1. **Compile + schema-migration idempotence.**
|
||||
2. **Fresh run:** `POST /trigger-job/storage_migration` → new row with
|
||||
`status='Running'`, `cursor=NULL`.
|
||||
3. **Concurrent trigger:** second `POST` while the first is running
|
||||
returns the SAME `run_id` (idempotent, DB unique index enforces).
|
||||
4. **Cancel + resume round-trip:** `trigger-job/…/cancel` flips to
|
||||
`CancelRequested`; handler polls, returns `Paused { cursor }`;
|
||||
engine writes `Paused`. `POST /trigger-job/…` again resumes; cursor
|
||||
picks up where left off; `stats.count` continues accumulating.
|
||||
5. **Crash recovery:** stop the server mid-run; restart; boot sweep
|
||||
flips the row to `Paused` with `error_message = 'server restart mid-run'`;
|
||||
admin triggers again and it resumes.
|
||||
6. **Idempotent replay:** for consistency-check specifically, verify
|
||||
that re-processing the last unpersisted batch does NOT double-record
|
||||
findings (`UNIQUE (run_id, kind, resource_id)` on
|
||||
`admin.consistency_findings`).
|
||||
7. **`RunOutcome` bridge log lines:** completed run logs
|
||||
`outcome=ok, extra.completed=true`; paused logs
|
||||
`outcome=ok, extra.paused=true`; failed logs `outcome=err, cause=handler`.
|
||||
|
||||
---
|
||||
|
||||
## Cross-cutting
|
||||
|
||||
### Admin URL taxonomy
|
||||
|
||||
All under `/api/admin/internal/*`, gated by the existing
|
||||
`OXICLOUD_ENABLE_ADMIN_INTERNAL_ENDPOINTS` env var — reuses the same
|
||||
admin-guard middleware and the same "disabled → 404" contract as
|
||||
today's per-service triggers.
|
||||
|
||||
**Existing per-service shims** (`trigger-sweep`, `trigger-gc`,
|
||||
`trigger-grant-cleanup`) stay as thin forwards to `trigger-job/{name}`
|
||||
during migration so the existing Hurl suites keep working.
|
||||
Deprecation surfaces via a `Deprecation: true` response header
|
||||
operators can grep for.
|
||||
|
||||
### Config surface — env vars
|
||||
|
||||
Canonical form for every job (Part 1 or Part 2 alike, AND for core
|
||||
workers even though they don't register with the scheduler):
|
||||
|
||||
```
|
||||
OXICLOUD_JOB_<NAME>_ENABLED
|
||||
OXICLOUD_JOB_<NAME>_INTERVAL_HOURS # or _INTERVAL_SECS for sub-hour cadences
|
||||
OXICLOUD_JOB_<NAME>_<CUSTOM>... # e.g. _GRACE_HOURS, _BATCH_SIZE
|
||||
```
|
||||
|
||||
Core workers reuse this naming purely for uniform operator ergonomics
|
||||
(e.g. `OXICLOUD_JOB_TREE_ETAG_FLUSH_INTERVAL_MS`) — the convention is
|
||||
what operators grep for; whether the loop is scheduler-driven or a
|
||||
dedicated `tokio::spawn` is an implementation detail they don't see.
|
||||
|
||||
Existing per-service env vars keep working as **aliases** during
|
||||
migration — `OXICLOUD_GRANT_CLEANUP_INTERVAL_HOURS` reads first, falls
|
||||
back to `OXICLOUD_JOB_GRANT_CLEANUP_INTERVAL_HOURS`. Deprecated aliases
|
||||
warn once on startup and stay recognised through one minor version.
|
||||
|
||||
### Logging schema
|
||||
|
||||
Uniform structured target across both engines:
|
||||
|
||||
```rust
|
||||
tracing::info!(
|
||||
target: "oxicloud::scheduler",
|
||||
event = "job.run",
|
||||
job = %name,
|
||||
outcome = %outcome_kind, // "ok" | "err"
|
||||
cause = %cause, // omitted on ok; "handler" | "timeout" | "panicked"
|
||||
count = ...,
|
||||
elapsed_ms = ...,
|
||||
// extras from the JobOutcome::Ok.extra map, flattened
|
||||
...,
|
||||
"job {name} ran"
|
||||
);
|
||||
```
|
||||
|
||||
Security-relevant jobs (grant cleanup, authz cache invalidation) still
|
||||
double-log to `target: "audit"` — the scheduler channel is for
|
||||
observability; the audit channel is for compliance.
|
||||
|
||||
For Part 2 handlers, the same log line fires at run completion. The
|
||||
`extra` map surfaces `completed`/`paused`/`cursor_hex` per the
|
||||
`RunOutcome` bridge above.
|
||||
|
||||
### Composability
|
||||
|
||||
A recoverable job CAN also be periodically-triggered — register with
|
||||
both engines. Part 1's tick calls Part 2's `run_or_resume(job, store_factory).await`
|
||||
as its handler. The exclusivity index in Part 2 makes this safe even
|
||||
if the interval is short enough that a tick fires while a previous
|
||||
run is still going: the second tick's `run_or_resume` short-circuits
|
||||
to "already running."
|
||||
|
||||
### Ordering and dependencies (deferred)
|
||||
|
||||
Cross-job dependencies (e.g. "trash cleanup runs before dedup GC")
|
||||
are not modelled. Every job runs independently. If a real ordering
|
||||
constraint appears, we add a `depends_on: Vec<String>` field and
|
||||
topological scheduling then.
|
||||
|
||||
### Shutdown coordination (deferred)
|
||||
|
||||
Matches the existing daemons: no cancellation channel. The scheduler
|
||||
task dies with the runtime. Recoverable jobs surviving a hard shutdown
|
||||
land as `Paused` on the next boot via the sweep. If graceful shutdown
|
||||
lands elsewhere in the codebase, the scheduler and all jobs migrate
|
||||
together.
|
||||
|
||||
### Future extension — plugins
|
||||
|
||||
Once these engines exist they become the natural place for Extism
|
||||
plugins to declare scheduled work — manifest `[[jobs]]` entries,
|
||||
registered on `on_plugin_loaded`, unregistered on unload. Deliberately
|
||||
deferred: no plugin needs it today, and adding
|
||||
`JobOwner { Native | Plugin { id } }` + `unregister_by_owner` is a
|
||||
small type extension the day one does. Nothing in the v1 design
|
||||
precludes it.
|
||||
|
||||
### Job-history observability
|
||||
|
||||
`admin.background_runs` already carries the latest run per Part 2 job
|
||||
— "last run time + status" is a `SELECT DISTINCT ON (job_name) …`
|
||||
query. Deeper history (retention window, per-run drill-down UI) is
|
||||
deferred; the log stream is the source of truth for older runs.
|
||||
|
||||
Part 1's periodic jobs only carry the last outcome IN MEMORY — no
|
||||
DB row. If a periodic-only job needs persisted last-run visibility,
|
||||
either promote it to a "trivial" recoverable job (immediate
|
||||
`Completed`) or add a small `admin.periodic_runs_last` table later.
|
||||
No such need today.
|
||||
|
||||
## Out of scope
|
||||
|
||||
- **Cross-job dependencies.** Register-time ordering only, not runtime
|
||||
graph.
|
||||
- **Retention pruning of terminal `background_runs` rows.** Deferred
|
||||
until the volume warrants a policy.
|
||||
- **Prometheus / OpenMetrics export.** Log-only for now.
|
||||
- **Distributed scheduling.** Single-process. If OxiCloud ever runs
|
||||
multi-node, `SELECT … FOR UPDATE SKIP LOCKED` on the runs table is
|
||||
the pattern; not now.
|
||||
- **Backfill on startup.** If the process is down when a Part 1 job's
|
||||
tick was due, we do NOT catch up — the job runs at its next
|
||||
interval. Matches every existing daemon's behaviour today.
|
||||
- **Cron expressions.** Fixed intervals only.
|
||||
- **Rate limiting the admin trigger endpoint.** It's already
|
||||
admin-gated.
|
||||
|
||||
## Related memory notes
|
||||
|
||||
- `feedback_no_abbreviated_env_vars` — full-word env var names
|
||||
(`OXICLOUD_JOB_TRASH_CLEANUP_INTERVAL_HOURS`, not
|
||||
`OXICLOUD_JOB_TC_INTERVAL_H`).
|
||||
- The grant-cleanup implementation is the closest reference for the
|
||||
Part 1 daemon → tenant migration shape: three env vars, one impl of
|
||||
an authz trait method, one daemon service, one admin trigger.
|
||||
- `project_consistency_check_trait` — the consistency framework
|
||||
described in `docs/plan/consistency-check.md` is a *consumer* of
|
||||
Part 2 (the recoverable-run engine), not a peer. It ships after
|
||||
Part 2 lands.
|
||||
Reference in New Issue
Block a user