873 lines
41 KiB
Markdown
873 lines
41 KiB
Markdown
# DEVELOP.md — Developer Guide
|
|
|
|
This guide is for engineers working on the model platform codebase. It is the
|
|
authoritative source for architecture, code layout, configuration, conventions,
|
|
and common tasks. `ARCHITECTURE.md` now only records the simplification
|
|
history; `HANDOVER.md` covers recent commits and pending work.
|
|
|
|
## Architecture
|
|
|
|
The platform is a self-hosted Jupyter model development environment: interactive
|
|
workspaces, DAG scheduling, object-store artifacts, and per-workspace runtimes
|
|
— all exposed through a single Nginx gateway.
|
|
|
|
### Capability map
|
|
|
|
| Capability | Where it lives |
|
|
|---|---|
|
|
| Workspace notebook editing, row-level lock | `backend/api/jupyter.py` + `scripts.is_locked` |
|
|
| Jupyter auth routing (browser never holds runtime token) | `nginx/default.conf` + `auth_request` + `backend/api/jupyter.py` |
|
|
| Object storage for notebook / script / version / run_log (s3 / local) | `common/storage/` + `backend/services/storage.py` |
|
|
| DAG scheduling: nodes, edges, cron, manual trigger, retry, snapshot | `backend/api/schedules/` + `backend/api/schedules/runs.py` + `schedule/` (5 layers) |
|
|
| DAG execution via MySQL Outbox (no Redis, no in-process queue) | `schedule/application/orchestrator.py` + `schedule/execution/worker.py` |
|
|
| Per-workspace Jupyter subprocess pool, asyncio lock | `runtime/process.py` |
|
|
| Runtime rclone FUSE mount of workspace bucket (s3 mode only) | `runtime/mount.py` |
|
|
| 18 MySQL tables, soft delete, zero FK, async SQLAlchemy 2.0 | `common/db/models/` |
|
|
|
|
### Component diagram
|
|
|
|
```
|
|
┌────────────────────┐
|
|
│ Browser (SPA) │
|
|
└─────────┬──────────┘
|
|
│ HTTPS / WS
|
|
┌─────────▼──────────┐
|
|
│ Nginx (only :80) │ ← templates/default.conf
|
|
│ /api/ /jupyter/ /storage/
|
|
└────┬───────┬──────┘
|
|
│ │
|
|
┌──────────────┘ └─────────────┐
|
|
▼ ▼
|
|
┌──────────────────┐ ┌──────────────────────┐
|
|
│ FastAPI Backend │ │ Runtime (Jupyter) │
|
|
│ + /internal/v1 │ control │ - rclone FUSE mount │
|
|
│ /objects │ token-Auth │ │
|
|
│ (storage) ├──────────────►│ - subprocess pool │
|
|
│ - DAG CRUD │ │ (per workspace) │
|
|
│ - script CRUD │ └──────────┬───────────┘
|
|
│ - auth_request │ │ FUSE / shared vol
|
|
│ - /api/v1/... │ ▼
|
|
└────┬──────┬──────┘ ┌──────────────────────┐
|
|
│ │ │ Object storage │
|
|
│ └──────── HTTP ───────►│ (s3: S3 service / │
|
|
▼ │ local: shared vol) │
|
|
┌────────────┐ │ 4 buckets per usage │
|
|
│ MySQL │◄───────── poll ─────│ │
|
|
│ - 18 tbls │ └──────────────────────┘
|
|
│ - outbox │
|
|
│ - jobstore │
|
|
└────┬───────┘
|
|
▲
|
|
│ outbox poll
|
|
┌────┴──────────────────────────┐
|
|
│ Schedule Executor │
|
|
│ - CronScheduler (APScheduler) │
|
|
│ - DispatchOrchestrator │
|
|
│ - NodeExecutor (worker) │
|
|
│ - SchedulerService (facade) │
|
|
└───────────────────────────────┘
|
|
```
|
|
|
|
### Services (docker-compose)
|
|
|
|
The architecture intentionally exposes only one host port (the gateway); all
|
|
other services are on the Docker internal network.
|
|
|
|
| Service | Image | Exposed | Purpose |
|
|
|---|---|---|---|
|
|
| `web` | `nginx:alpine` | host `:8888` → `:80` | SPA, `/api/` reverse proxy, `/jupyter/{ws}/` `auth_request` proxy, `/storage/` S3 passthrough (s3 mode only) |
|
|
| `backend` | `Dockerfile` | internal only | DAG CRUD, script CRUD, schedule trigger, `/api/v1/auth/jupyter`, `/internal/v1/objects` inter-service RPC (shared `INTERNAL_SERVICE_TOKEN`, see `§Auth`) |
|
|
| `runtime` | `Dockerfile` | internal only | Per-workspace Jupyter subprocess pool, rclone FUSE mount of `workspace` bucket (s3 mode) |
|
|
| `schedule` | `Dockerfile` | internal only | Cron tick + DAG execution (polls MySQL Outbox) |
|
|
|
|
The pre-2026 host-port mappings for backend/runtime (`8891:8000` / `8892:8000`)
|
|
were removed; no service is reachable from the host except Nginx anymore.
|
|
Backend ↔ schedule now talk via the `X-Internal-Service-Token` header on
|
|
`/internal/v1/*`. See `API.md §9`.
|
|
|
|
### Storage layout
|
|
|
|
The object store is selected at deploy time by `settings.storage_backend`
|
|
(`"s3"` default, `"local"` for dev / single-node / air-gapped). Either way
|
|
there are 4 purpose-named buckets, resolved in a single place
|
|
(`common/storage/factory.py:actual_bucket_name` + `USAGE_TYPE_TO_PURPOSE`).
|
|
|
|
| `usage_type` | Bucket | Env var | Default name (s3) |
|
|
|---|---|---|---|
|
|
| `working_copy`, `public_script`, `data_resource`, `snapshot` | `workspace` | `S3_WORKSPACE_BUCKET` | `workspace` |
|
|
| `version_artifact` | `version` | `S3_VERSION_BUCKET` | `version` |
|
|
| `run_log`, `run_result` | `run_log` | `S3_RUN_LOG_BUCKET` | `run-log` |
|
|
| (soft-delete target) | `trash` | `S3_TRASH_BUCKET` | `trash` |
|
|
|
|
`STORAGE_BACKEND=s3` → 4 separate S3 buckets.
|
|
`STORAGE_BACKEND=local` → 4 subdirectories under `LOCAL_STORAGE_BASE_DIR`
|
|
(default `/data`):
|
|
|
|
```
|
|
/data/
|
|
├── workspace/ # S3_WORKSPACE_BUCKET
|
|
├── version/ # S3_VERSION_BUCKET
|
|
├── run_log/ # S3_RUN_LOG_BUCKET
|
|
└── trash/ # S3_TRASH_BUCKET
|
|
```
|
|
|
|
A workspace's `Workspaces.artifact_bucket` column (when non-null) overrides
|
|
the default for that workspace, regardless of `usage_type` — useful for
|
|
isolating paid customers onto a dedicated bucket.
|
|
|
|
Object keys are a flat two-level path — `workspace_id` plus a server-issued
|
|
ULID — preserving the original file extension so Jupyter can pick its editor
|
|
from the suffix:
|
|
|
|
```
|
|
<bucket>/<workspace_id>/<ulid>{.<ext>}
|
|
```
|
|
|
|
File name, extension, MIME, and logical path all live on `StorageObjects` /
|
|
`Scripts` rows; reorganizing the bucket does not require rewriting the
|
|
database. Backend code never writes to the container local filesystem except
|
|
in `STORAGE_BACKEND=local` mode (where the shared `local-storage` volume IS
|
|
the canonical store). Schedule Executor stages node artifacts in
|
|
`tempfile.TemporaryDirectory()` (auto-cleanup). Only the `runtime` container
|
|
keeps a host volume — s3 mode needs it for rclone FUSE; local mode is a no-op
|
|
passthrough.
|
|
|
|
## Code layout
|
|
|
|
All Python packages use the `src/<pkg>/` layout; `uv` workspace glues them into
|
|
one `.venv`. Always invoke via `uv run [--package <pkg>] <cmd>` (see "Local
|
|
development" for the gotcha).
|
|
|
|
```
|
|
common/src/common/ Pure-Python shared library
|
|
config.py Settings (pydantic-settings, lru_cache singleton)
|
|
db/ SQLAlchemy 2.0 async engine, session_scope, Base
|
|
db/models/ 18 tables in 7 domain files (zero FK, zero relationship)
|
|
auth/ JWT / bcrypt / workspace membership helpers
|
|
scheduler/ APScheduler trigger helpers (delayed import)
|
|
storage/ AsyncStorageBackend abstraction + Pydantic schemas
|
|
base.py Abstract interface
|
|
factory.py create_storage + build_storage_config + PURPOSE_BUCKETS
|
|
schemas.py CreateUploadRequest / ServerObjectRequest
|
|
backends/local.py Local filesystem impl
|
|
backends/s3.py S3-compatible impl (boto3)
|
|
registry.py Bucket registry
|
|
eventing.py add_outbox_event / utcnow / event_time
|
|
service_app.py /health/ready TCP probe, /api/v1/health
|
|
logging.py loguru config (LOG_LEVEL)
|
|
schemas.py StrictModel base
|
|
ids.py ULID generation helpers
|
|
utils.py get_free_port, start_process
|
|
|
|
backend/src/backend/ Public FastAPI service
|
|
main.py lifespan + route registration
|
|
audit.py HTTP access log middleware (loguru sink)
|
|
api/ HTTP route handlers (one module per bounded context)
|
|
auth.py /api/v1/auth/* (login / me / jupyter)
|
|
jupyter.py /api/v1/auth/jupyter — the ONLY auth entry
|
|
dependencies.py request_context, database_session
|
|
platform.py /api/v1/platform/* (system admin)
|
|
admin.py /api/v1/admin/* (workspace-internal admin)
|
|
scripts.py /api/v1/scripts/* + /api/v1/workspace-directories
|
|
resources.py /api/v1/data-resources/*
|
|
schedules/schedules.py DAG CRUD
|
|
schedules/runs.py Run lifecycle
|
|
storage.py /internal/v1/objects — single token-guarded endpoint (P0-1)
|
|
services/ Pure-Python business logic (no HTTP / no DI)
|
|
scripts.py create_workspace_directory, visibility-filtered queries
|
|
resources.py owner-scoped resource listing helpers
|
|
jupyter.py jupyter_path / lock helpers
|
|
storage.py object store helpers
|
|
schedules.py DAG validation (cycle / orphan detection)
|
|
schemas/ Pydantic request / response models
|
|
auth.py / common.py / jupyter.py / platform.py / resources.py / schedules.py / scripts.py
|
|
clients/ Outbound HTTP / RPC clients
|
|
runtime.py Self-contained httpx wrapper for the runtime
|
|
scheduler.py Backend → Schedule HTTP client (callback / dispatch)
|
|
rclone.py rclone RC API client (FUSE cache invalidation)
|
|
|
|
schedule/src/schedule/ Schedule Executor (DAG worker)
|
|
main.py Lifespan + FastAPI app
|
|
notebook_runner.py Subprocess entry point (nbclient) — DO NOT RENAME
|
|
domain/ Pure-Python domain types
|
|
execution.py ExecutionResult (frozen dataclass) + state enums
|
|
context.py Constants + naive_utc
|
|
scheduling/ Time-based trigger
|
|
scheduler.py CronScheduler (APScheduler + 5s sync loop)
|
|
application/ Facades / orchestrators
|
|
service.py SchedulerService (composes the three)
|
|
orchestrator.py DispatchOrchestrator (Outbox poll + DAG advance)
|
|
execution/ DAG node execution
|
|
executor.py Dispatch helpers (script-type routing)
|
|
worker.py NodeExecutor + asyncio entry, schedule-spawned task boundary
|
|
runners/notebook.py nbclient subprocess path (6-line `notebook_runner` shim re-exports `main`)
|
|
infrastructure/ External-system adapters
|
|
storage/client.py SchedulerStorageClient — talks to backend /internal/v1/objects
|
|
|
|
runtime/src/runtime/ Jupyter Runtime
|
|
main.py FastAPI entry: jupyter action endpoints
|
|
process.py Per-workspace subprocess pool + asyncio locks
|
|
mount.py rclone FUSE mount lifecycle
|
|
|
|
frontend/ React Router SPA (vite build → nginx)
|
|
app/ features/ routes/ services/ components/
|
|
|
|
migrations/ Alembic schema versions
|
|
docker-compose.yml 5 services (migrate / web / backend / runtime / schedule)
|
|
default.conf Nginx template
|
|
scripts/nginx-entrypoint.sh
|
|
.env.example All 26 config.py keys documented
|
|
```
|
|
|
|
## Configuration system
|
|
|
|
All env vars go through one place: `common/src/common/config.py`.
|
|
|
|
```python
|
|
from common.config import settings
|
|
|
|
# Auth / runtime
|
|
settings.database_url # str — SQLAlchemy async URL (mysql+asyncmy, charset utf8mb4)
|
|
settings.jwt_secret # HS256 secret for the auth_request handler
|
|
settings.cookie_force_secure # bool — write Secure flag even on plain HTTP (TLS-terminating proxy)
|
|
settings.service_name # surfaced in /health
|
|
settings.schedule_event_namespace # Outbox event-type namespace prefix (NOT APScheduler JobStore)
|
|
settings.readiness_targets # CSV host:port list for /health/ready
|
|
|
|
# HTTP clients (intra-cluster URLs)
|
|
settings.runtime_api_url # backend → runtime HTTP base
|
|
settings.public_base_url # runtime public base URL (browser-facing /jupyter/)
|
|
settings.backend_api_url # schedule → backend HTTP base
|
|
settings.rclone_rc_url # backend → rclone RC control API
|
|
settings.internal_service_token # Backend ↔ Schedule shared secret (X-Internal-Service-Token)
|
|
|
|
# Logging / audit
|
|
settings.log_level # DEBUG / INFO / WARNING / ERROR / CRITICAL (lowercase → fallback INFO)
|
|
settings.audit_log_dir # dir for daily audit logs (relative to cwd; "" disables file sink)
|
|
settings.audit_log_retention_days # 0 disables cleanup
|
|
settings.audit_excluded_paths # list[str] — paths skipped from audit (health probes, etc.)
|
|
|
|
# Storage
|
|
settings.storage_backend # "s3" (default) or "local"
|
|
settings.local_storage_base_dir # root dir for storage data (default "/data")
|
|
settings.s3_endpoint # str (s3 mode only)
|
|
settings.s3_access_key # str (s3 mode only)
|
|
settings.s3_secret_key # str (s3 mode only)
|
|
settings.s3_workspace_bucket # str (s3 mode only)
|
|
settings.s3_version_bucket # str (s3 mode only)
|
|
settings.s3_run_log_bucket # str (s3 mode only)
|
|
settings.s3_trash_bucket # str (s3 mode only)
|
|
settings.s3_trash_retention_days # int (s3 mode only)
|
|
|
|
# Schedule
|
|
settings.schedule_execution_concurrency # int — max concurrent notebook subprocesses
|
|
```
|
|
|
|
### Encrypted env values
|
|
|
|
`Settings` runs a `model_validator` (`_decrypt_encrypted_fields`,
|
|
`common/src/common/config.py:168-180`) that scans every string field for the
|
|
prefix `ENC(...)` and decrypts the inner value with `APP_CONFIG_SECRET_KEY`
|
|
using Fernet. Use this for secrets that should not be stored in plain `.env`
|
|
files (e.g. third-party API tokens shipped via deployment config).
|
|
|
|
```bash
|
|
# .env
|
|
APP_CONFIG_SECRET_KEY=<base64 Fernet key — generate with `python -c "from cryptography.fernet import Fernet; print(Fernet.generate_key().decode())"`>
|
|
SOME_TOKEN=ENC(gAAAAABm...) # ciphertext produced by Fernet.encrypt(b"plaintext")
|
|
```
|
|
|
|
At process start, `SOME_TOKEN` resolves to the decrypted plaintext. If the
|
|
field is *not* encrypted (no `ENC(...)` prefix), it passes through unchanged,
|
|
so plain `.env` files keep working. The pre-2026 in-repo `encrypt_secret.py`
|
|
script was the CLI wrapper around the same Fernet key; if you have old
|
|
ciphertexts they round-trip with the new `APP_CONFIG_SECRET_KEY` value.
|
|
|
|
`Settings` reads from process env first, then from a `.env` file at CWD
|
|
if present. `pydantic-settings` auto-loads. `case_sensitive=False` so
|
|
`DATABASE_URL` / `database_url` both work. The full list of 26 fields
|
|
is in `common/src/common/config.py`.
|
|
|
|
### Adding a new env var
|
|
|
|
1. Add the field to `Settings`:
|
|
```python
|
|
new_var: str = Field(default="x", description="...")
|
|
```
|
|
2. Add the line to `.env.example` with a comment (keep it synced — every
|
|
field in config.py must have a matching `.env.example` entry).
|
|
3. Use `settings.new_var` at the call site.
|
|
|
|
Do **not** call `os.environ["NEW_VAR"]` or `os.getenv("NEW_VAR")` in
|
|
application code. The grep below should return zero hits:
|
|
|
|
```bash
|
|
grep -rnE 'os\.(environ\[?["\x27][A-Z_]+|getenv\(["\x27][A-Z_]+)' --include="*.py" \
|
|
backend/src/ schedule/src/ runtime/src/ common/src/
|
|
```
|
|
|
|
## Conventions
|
|
|
|
### Database / SQLAlchemy
|
|
|
|
- **No foreign keys, no `relationship`** — every join is explicit.
|
|
- Every table has `is_deleted TINYINT(1) NOT NULL DEFAULT 0` and
|
|
`deleted_at DATETIME(3) NULL`; queries must filter `is_deleted == 0`
|
|
(or `deleted_at.is_(None)`) to avoid logical-deleted rows.
|
|
- Domain files under `common/src/common/db/models/` are split by
|
|
bounded context: `events` (2 tables: `ConsumerInbox`,
|
|
`OutboxEvents`) / `identity` (4: `Users`, `Roles`, `RolePermissions`,
|
|
`Permissions`) / `schedules` (5: `Schedules`, `ScheduleRuns`,
|
|
`ScheduleNodes`, `ScheduleEdges`, `ScheduleNodeRuns`) / `scripts`
|
|
(2: `Scripts`, `Versions`) / `storage` (3: `StorageObjects`,
|
|
`UploadSessions`, `DataResources`) / `workspaces` (2: `Workspaces`,
|
|
`WorkspaceMembers`) — 18 tables across 6 model files.
|
|
- All models are `class X(Base)` SQLAlchemy 2.0 declarative-mapped.
|
|
- The full schema is in `migrations/versions/`. Apply with:
|
|
```bash
|
|
uv run --frozen --package backend alembic upgrade head
|
|
```
|
|
|
|
### Storage
|
|
|
|
- All object bytes go through `common.storage.AsyncStorageBackend`,
|
|
created by `create_storage(config)` from `common.storage.factory`.
|
|
- Two backends are registered: `local` (filesystem, local mode) and
|
|
`s3` (S3-compatible service, s3 mode). Selection is per-deployment
|
|
via `settings.storage_backend` (`"s3"` default, `"local"` for
|
|
dev / single-node / air-gapped).
|
|
- The factory helper `build_storage_config(bucket_name)` returns the
|
|
right `create_storage` kwargs for each of the 4 purpose buckets
|
|
(`workspace`, `version`, `run_log`, `trash`). Use it in lifespan code;
|
|
route handlers don't see the difference.
|
|
- Bucket resolution from `usage_type` lives in **one place**
|
|
(`common/src/common/storage/factory.py:actual_bucket_name` +
|
|
`USAGE_TYPE_TO_PURPOSE`). The route handlers in
|
|
`backend/src/backend/api/storage.py` only know about
|
|
`app.state.object_stores[bucket_name]` and never call
|
|
`settings.s3_*_bucket` directly.
|
|
- The runtime's view of the workspace bucket on disk is exposed by
|
|
`common.storage.workspaces_root()`:
|
|
- Both modes resolve to `${settings.local_storage_base_dir}/workspace`
|
|
(default `/data/workspace`); s3 mode uses it as the rclone FUSE
|
|
mount target, local mode uses it as a subdir of the shared
|
|
`local-storage` volume.
|
|
`settings.local_storage_base_dir` is the **only** path setting; the
|
|
helper handles the per-mode suffix. Don't read `settings.workspaces_root`
|
|
or any other path setting directly in runtime code — use this helper.
|
|
- The pre-2026 abstraction (`RustFSObjectStore` / `common.storage.client`
|
|
/ `StorageClient` HTTP wrapper) is gone. Don't reintroduce it.
|
|
|
|
### Auth
|
|
|
|
- Browser → `/jupyter/{workspace_id}/...` → Nginx `auth_request` →
|
|
`GET /api/v1/auth/jupyter` (Backend).
|
|
- The handler:
|
|
1. Parses cookie / Bearer JWT (HS256 + `settings.jwt_secret`).
|
|
2. Verifies `WorkspaceMembers` for the workspace.
|
|
3. Verifies `Scripts.is_locked` for the requested notebook path
|
|
(owner / unlocked → allow; otherwise 403).
|
|
4. Calls `RuntimeClient.get_workspace` / `start_workspace`.
|
|
5. Returns `x-upstream-addr` + `x-jupyter-internal-token` response
|
|
headers. **Browser never holds the runtime token.**
|
|
|
|
### Service-to-service auth (P0-1 fix)
|
|
|
|
- Schedule → Backend single endpoint ``POST /internal/v1/objects`` is
|
|
guarded by ``require_internal_service`` in ``backend.api.storage``.
|
|
- The token header is ``X-Internal-Service-Token`` (case-insensitive
|
|
on the wire because FastAPI ``Header`` lowercase-matches the name
|
|
``x-internal-service-token``); the secret value comes from
|
|
``settings.internal_service_token`` / env ``INTERNAL_SERVICE_TOKEN``.
|
|
- Comparison uses ``secrets.compare_digest`` — never equality.
|
|
- Backend and schedule must be configured with the same value; a
|
|
mismatch fails fast at the first notebook run (``401``) which is
|
|
intentional. ``.env.example`` ships a placeholder
|
|
``change-me-internal-service-token`` and the docker-compose
|
|
``${INTERNAL_SERVICE_TOKEN:?...}`` reference forces production
|
|
deployments to set a real value.
|
|
- Removing the legacy backend / runtime host-port mappings
|
|
(``8891:8000`` / ``8892:8000``) is part of the same fix — no service
|
|
is reachable from the host except Nginx anymore.
|
|
- Nginx captures the headers via `auth_request_set` and proxies to the
|
|
upstream sub-process with `Authorization: token $jupyter_token`.
|
|
|
|
### Permission gates
|
|
|
|
- For write operations on a script/notebook, call
|
|
`require_script_modify_access(script, user_id=..., is_admin=...)`
|
|
from `backend/src/backend/services/scripts.py`. It enforces:
|
|
- admin or owner → allow
|
|
- non-owner, `is_locked == 0` → allow
|
|
- non-owner, `is_locked == 1` → 403
|
|
- Read endpoints (`list_scripts`, `get_script`) intentionally do **not**
|
|
check `is_locked` — workspace members can see the script list.
|
|
|
|
### Owner-scoping + visibility (cross-owner browsing)
|
|
|
|
`GET /api/v1/scripts`, `GET /api/v1/data-resources`, and
|
|
`GET /api/v1/workspace-directories` all accept an optional
|
|
`owner_user_id` query param and follow the same model:
|
|
|
|
- `owner_user_id` 缺省 = 当前请求者本人(scope to `workspace/{me}/...`).
|
|
- 传值时 scope 到 `workspace/{owner_user_id}/...`,用于前端"点开其他成员
|
|
分组"的懒加载(见 §3.4 of `API.md`).
|
|
- `visibility` 过滤(非 admin):`owner_user_id == me OR visibility IN
|
|
(workspace, public)` — 自己可见自己全部(含 private),他人只见其
|
|
workspace/public,排除他人 private.
|
|
- 系统管理员跳过 visibility 过滤.
|
|
- 目录行默认 `visibility='public'`(由 `create_workspace_directory` 写入),
|
|
不施加 visibility 过滤,使跨 owner 目录树可见.
|
|
|
|
实现 helper 在 `backend/src/backend/services/scripts.py`
|
|
(`_build_list_scripts_owner_descendant_prefix`) 和
|
|
`backend/src/backend/api/resources.py` 中按相同模式分别构造 owner-scoped
|
|
LIKE 前缀。
|
|
|
|
### Eventing — Outbox + Inbox + business tables
|
|
|
|
The platform has **no** message broker. The only async fabric is three MySQL
|
|
tables, each on a different side of the same boundary:
|
|
|
|
| Table | Side | Job |
|
|
|---|---|---|
|
|
| business tables (`schedule_runs`, `schedule_node_runs`, etc.) | producer's local commit | The state change that *triggered* the event |
|
|
| `outbox_events` | producer side | Durable handoff: "this business state change produced this event" |
|
|
| `consumer_inbox` | consumer side | Idempotency: "this consumer has already processed this event" |
|
|
|
|
Together they implement the **Transactional Outbox** + **Consumer Inbox**
|
|
patterns: atomic publish on the producer side, idempotent side effects on
|
|
the consumer side. There is no Redis Stream, no Kafka, no in-process queue.
|
|
If MySQL is down, both sides are down — and that's intentional.
|
|
|
|
#### Why two tables and not one
|
|
|
|
A naïve "publish event by inserting a row, then have the consumer mark it
|
|
done" design breaks under two failure modes:
|
|
|
|
1. **Producer crash after business commit, before publish.** The state
|
|
change is real but the event was never sent. Consumers never see it.
|
|
2. **Consumer crashes mid-processing.** The event was received but the
|
|
side effect may have happened or not. Re-delivery causes double
|
|
execution.
|
|
|
|
`outbox_events` fixes (1): writing to the business table and the outbox
|
|
happens in **one transaction**, so either both are durable or neither is.
|
|
A separate dispatcher (`schedule/application/orchestrator.py`) polls
|
|
`outbox_events` and pushes to consumers. The dispatcher crashing is
|
|
harmless — the next poll picks up where it left off.
|
|
|
|
`consumer_inbox` fixes (2): before applying a side effect, the consumer
|
|
inserts a row keyed by `(consumer_name, event_id)`. Re-delivery of the same
|
|
`event_id` hits the existing row and short-circuits.
|
|
|
|
#### Producer side — `outbox_events`
|
|
|
|
Schema lives in `common/db/models/events.py` (model `OutboxEvents`) and the
|
|
baseline migration `migrations/versions/e1f2a3b4c5d6_rebuild_baseline.py`.
|
|
|
|
Key fields:
|
|
|
|
- `event_id` (ULID, PK) — globally unique; `consumer_inbox` references this
|
|
as the dedup key.
|
|
- `aggregate_type` + `aggregate_id` — the producer's domain object
|
|
(e.g. `("schedule_run", "<run_id>")`).
|
|
- `event_type` — namespaced via `schedule_event_type(...)` so multiple
|
|
deployments sharing one MySQL don't cross-consume (see below).
|
|
- `schema_version` (default 1) — consumer can branch on this when the
|
|
payload shape changes.
|
|
- `payload_json` — the event body; opaque to the outbox.
|
|
- `event_status` (`pending` → `published` | `failed`) + `available_at` +
|
|
`retry_count` + `last_error` — dispatcher state machine. Indexed on
|
|
`(event_status, available_at, created_at)` because that's the poll
|
|
hot path.
|
|
- `idempotency_key` + index — upstream dedup at the producer (e.g. two
|
|
requests from the user that should produce one event, not two).
|
|
- `trace_id` — links to the request that produced the event for log
|
|
correlation.
|
|
|
|
Write path — the **only** entry point is `common.eventing.add_outbox_event`:
|
|
|
|
```python
|
|
await add_outbox_event(
|
|
session,
|
|
event_type=schedule_event_type("schedule.run.requested"),
|
|
producer="backend.api.schedules.runs",
|
|
trace_id=request.state.trace_id,
|
|
aggregate_type="schedule_run",
|
|
aggregate_id=run_id,
|
|
idempotency_key=f"schedule.run.requested:{run_id}",
|
|
payload={"run_id": run_id, "schedule_id": schedule_id},
|
|
)
|
|
```
|
|
|
|
`session` is the **same** SQLAlchemy session as the business-table write;
|
|
the outbox row goes in via `session.add(...)` and commits together with
|
|
the business row. Never call `session.commit()` between the business
|
|
write and the outbox write — that defeats the whole pattern.
|
|
|
|
**Event type namespacing.** Every event type MUST go through
|
|
`common.eventing.schedule_event_type(raw)` before being passed to
|
|
`add_outbox_event`. The helper prefixes `settings.schedule_event_namespace`
|
|
(default `model-platform-develop`) so that two deployments sharing one
|
|
MySQL — common during development — don't accidentally consume each
|
|
other's events. The constants `SCHEDULE_RUN_REQUESTED_EVENT`,
|
|
`NODE_EXECUTE_EVENT`, `NODE_FINISHED_EVENT` in
|
|
`schedule/application/orchestrator.py` and `schedule/execution/worker.py`
|
|
are already namespaced; never hard-code the raw string.
|
|
|
|
Current event types in use:
|
|
|
|
| Event | Produced by | Consumed by |
|
|
|---|---|---|
|
|
| `schedule.run.requested` | `backend/api/schedules/runs.py`, cron post-back | `schedule/application/orchestrator.py` (`schedule-orchestrator`) |
|
|
| `job.node.execute` | `schedule/application/orchestrator.py` when a node is ready | `schedule/execution/worker.py` (`schedule-results` writes back via the orchestrator) |
|
|
| `job.node.finished` | `schedule/execution/worker.py` after a node run finishes | `schedule/application/orchestrator.py` (`schedule-results`) |
|
|
|
|
The dispatcher polls every `0.25s` when there's pending work, dropping to
|
|
`1s` when idle — see the loop in
|
|
`schedule/application/orchestrator.py` (the `asyncio.sleep(0.25)` /
|
|
`asyncio.sleep(1)` branches).
|
|
|
|
#### Consumer side — `consumer_inbox`
|
|
|
|
Model `ConsumerInbox` in `common/db/models/events.py`. Composite PK
|
|
`(consumer_name, event_id)` so multiple consumers can independently
|
|
process the same `event_id`.
|
|
|
|
Lifecycle (lives in `schedule/application/orchestrator.py:_start_inbox` /
|
|
`_finish_inbox`):
|
|
|
|
```
|
|
┌──────────────┐
|
|
│ processing │ ← INSERT or re-claim on re-delivery
|
|
└──────┬───────┘
|
|
│
|
|
success │ failure
|
|
▼
|
|
┌─────────────┐
|
|
│ succeeded │ (terminal — no re-process)
|
|
└─────────────┘
|
|
|
|
on failure, error_message is set and the row is left in
|
|
`failed` for inspection; the dispatcher does NOT auto-retry
|
|
consumer failures (only producer-side publish failures).
|
|
```
|
|
|
|
The consumer **must** call `_start_inbox` (or equivalent) at the top of
|
|
every event handler. The function returns `(inbox_row, should_process)`;
|
|
if `should_process` is `False`, the event was already handled and the
|
|
handler returns immediately. On success the handler calls `_finish_inbox`
|
|
to flip `process_status` to `succeeded`.
|
|
|
|
Currently registered `consumer_name` values (see `orchestrator.py:594`,
|
|
`:924`):
|
|
|
|
- `schedule-orchestrator` — consumes `schedule.run.requested`
|
|
- `schedule-results` — consumes `job.node.finished`
|
|
|
|
A new consumer = a new `consumer_name` string. Two consumers sharing a
|
|
name will collide on the PK — pick a stable, descriptive name and treat
|
|
it as a contract.
|
|
|
|
#### Failure modes the design covers
|
|
|
|
| Scenario | What happens |
|
|
|---|---|
|
|
| Backend crashes after business commit, before dispatcher polls | Outbox row exists; next poll picks it up. |
|
|
| Dispatcher crashes after poll, before HTTP push to executor | `event_status` still `pending`; next poll retries. |
|
|
| Executor crashes mid-handler | Inbox row stays `processing`; on redelivery the handler re-enters `_start_inbox`, sees `succeeded`? — no, sees `processing` and re-runs. **This is currently a known soft spot** — the executor's `_finish_inbox` must run, and a crash before that means a re-run. Don't perform side effects before `_finish_inbox` succeeds. |
|
|
| Same `event_id` delivered twice (e.g. HTTP retry after success) | Inbox short-circuits; the second delivery is a no-op. |
|
|
| Multiple development deployments share one MySQL | `schedule_event_namespace` prefixes keep them isolated; each deployment only sees its own events. |
|
|
|
|
#### Adding a new event
|
|
|
|
1. Pick an `aggregate_type` / `aggregate_id` pair that identifies the
|
|
producing domain object.
|
|
2. Pick a `consumer_name` for each consumer. Stable, descriptive,
|
|
never reused for a different purpose.
|
|
3. Define the event type constant:
|
|
```python
|
|
# in the producing module
|
|
MY_NEW_EVENT = schedule_event_type("schedule.my_new_event")
|
|
```
|
|
4. Write it via `add_outbox_event` in the same transaction as the
|
|
business mutation.
|
|
5. In the consumer, start with `_start_inbox` (or follow the pattern
|
|
in `orchestrator.py`) before doing any side effects, and call
|
|
`_finish_inbox` on success.
|
|
6. Update the table above.
|
|
|
|
Do **not** invent a new messaging fabric (Redis Stream, Kafka, in-process
|
|
queue). The whole point of this design is that MySQL is the single
|
|
authority — adding a second one doubles the failure surface.
|
|
|
|
### Async / sync signatures
|
|
|
|
- `SchedulerService.start()` and `SchedulerService.close()` are
|
|
`async def` (so the FastAPI lifespan can `await` them).
|
|
- `CronScheduler.start()`, `DispatchOrchestrator.start()` are
|
|
**`def` (sync)** — they only `create_task(...)` and return. Don't
|
|
`await` them.
|
|
- `cron.start()`, `orchestrator.start()`, `worker.handle_node_execute`
|
|
are wired together in `SchedulerService.__init__`; their lifetime
|
|
is owned by the facade.
|
|
|
|
## Local development
|
|
|
|
### One-time setup
|
|
|
|
```bash
|
|
# Python workspace (monorepo via uv workspaces)
|
|
uv sync --all-packages
|
|
|
|
# Frontend deps
|
|
cd frontend && pnpm install && cd ..
|
|
```
|
|
|
|
### Per-service dev
|
|
|
|
Always go through `uv run` so the workspace `.venv` is used — bare
|
|
`uvicorn` / `python` resolves to system Python and `from backend.X`
|
|
imports fail with ModuleNotFoundError.
|
|
|
|
```bash
|
|
# Backend (terminal 1)
|
|
export DATABASE_URL="mysql+asyncmy://model_platform:model_platform@127.0.0.1:3306/model_platform?charset=utf8mb4"
|
|
export STORAGE_BACKEND=s3
|
|
export S3_ACCESS_KEY=modelplatform
|
|
export S3_SECRET_KEY=modelplatformsecret
|
|
export S3_ENDPOINT=http://127.0.0.1:9000
|
|
# Or for local mode:
|
|
# export STORAGE_BACKEND=local
|
|
# export LOCAL_STORAGE_BASE_DIR=/data
|
|
uv run --package backend uvicorn backend.main:app --host 0.0.0.0 --port 8000 --reload
|
|
|
|
# Schedule Executor (terminal 2)
|
|
uv run --package schedule uvicorn schedule.main:app --host 0.0.0.0 --port 8001 --reload
|
|
|
|
# Runtime (terminal 3 — needs SYS_ADMIN, FUSE, devmode)
|
|
uv run --package runtime python -m runtime.main
|
|
```
|
|
|
|
### Frontend dev
|
|
|
|
```bash
|
|
cd frontend
|
|
pnpm dev # http://localhost:5173, proxies /api to backend
|
|
pnpm typecheck
|
|
pnpm build
|
|
```
|
|
|
|
### Static checks
|
|
|
|
```bash
|
|
# Python compile
|
|
uv run --frozen --package backend python -m compileall -q backend/src common/src
|
|
uv run --frozen --package schedule python -m compileall -q schedule/src
|
|
uv run --frozen --package runtime python -m compileall -q runtime/src
|
|
|
|
# Type check (frontend)
|
|
cd frontend && pnpm typecheck && cd ..
|
|
|
|
# Docker compose config
|
|
docker compose config --quiet
|
|
```
|
|
|
|
### Smoke test
|
|
|
|
```bash
|
|
# Run the full ORM import + Settings smoke test
|
|
PYTHONPATH="backend/src:common/src" uv run --frozen --package backend python -c "
|
|
from backend.main import app
|
|
from common.config import settings
|
|
print('backend:', len(app.routes), 'routes')
|
|
print('settings ok:', settings.s3_endpoint)
|
|
"
|
|
```
|
|
|
|
## Common tasks
|
|
|
|
### Add a new DAG endpoint
|
|
|
|
1. Add the route handler in `backend/src/backend/api/schedules/schedules.py`
|
|
(DAG template) or `backend/src/backend/api/schedules/runs.py` (run
|
|
lifecycle).
|
|
2. Validate request via Pydantic schemas in
|
|
`backend/src/backend/schemas/schedules.py`.
|
|
3. For mutations on nodes/edges/versions: route through
|
|
`create_script_record` / `get_script_row` and apply
|
|
`require_script_modify_access` if it touches a script.
|
|
4. If it produces an outbox event, use
|
|
`add_outbox_event(session, event_type=schedule_event_type("..."), producer="...", ...)`
|
|
in the same transaction as the business write.
|
|
|
|
### Add a new env var
|
|
|
|
See "Adding a new env var" above.
|
|
|
|
### Add a new MySQL table
|
|
|
|
1. Add a model class in `common/src/common/db/models/<domain>.py`.
|
|
Include `is_deleted TINYINT(1) NOT NULL DEFAULT 0` and
|
|
`deleted_at DATETIME(3) NULL`.
|
|
2. Export it from `common/src/common/db/models/__init__.py`.
|
|
3. Generate the migration:
|
|
```bash
|
|
uv run --frozen --package backend alembic revision --autogenerate -m "add <feature>"
|
|
```
|
|
4. Review the generated `migrations/versions/*.py` — Alembic may
|
|
miss comments / server defaults. Manually fix the migration.
|
|
5. Apply locally:
|
|
```bash
|
|
uv run --frozen --package backend alembic upgrade head
|
|
```
|
|
|
|
### Wire a new storage bucket
|
|
|
|
The current 4 buckets are wired in `common/src/common/storage/factory.py`.
|
|
`USAGE_TYPE_TO_PURPOSE` maps each `usage_type` (`working_copy`,
|
|
`public_script`, `data_resource`, `snapshot`, `version_artifact`,
|
|
`run_log`, `run_result`) to one of the 4 purpose buckets (`workspace`,
|
|
`version`, `run_log`, `trash`); `actual_bucket_name(purpose)` then
|
|
returns the s3-style identifier from the corresponding
|
|
`settings.s3_<purpose>_bucket`. `BUCKET_FOR_USAGE` is a dict
|
|
comprehension built from these two.
|
|
|
|
The constant `PURPOSE_BUCKETS = ("workspace", "version", "run_log", "trash")`
|
|
in `common/storage/factory.py` enumerates the four backends built in the
|
|
backend lifespan. To add a fifth bucket:
|
|
|
|
1. Add the env var to `Settings` (s3 mode only):
|
|
```python
|
|
s3_<feature>_bucket: str = Field(default="<feature>", description="...")
|
|
```
|
|
2. Add to `.env.example` with a one-line comment.
|
|
3. Append `"<feature>"` to the `PURPOSE_BUCKETS` tuple in
|
|
`common/storage/factory.py`. `build_storage_config("<feature>")`
|
|
will then automatically read `settings.s3_<feature>_bucket` (s3
|
|
mode) or use `<local_storage_base_dir>/<feature>` (local mode).
|
|
4. Extend the `Literal` in `common/storage/schemas.py` (in
|
|
`CreateUploadRequest.usage_type`, `ServerObjectRequest.usage_type`)
|
|
to include the new value.
|
|
5. Add an entry in `USAGE_TYPE_TO_PURPOSE` mapping the new `usage_type`
|
|
to the new purpose; `BUCKET_FOR_USAGE` is regenerated automatically.
|
|
6. Pre-create the bucket (s3 mode) or subdirectory (local mode) in the
|
|
deployment. The backend no longer auto-creates buckets.
|
|
|
|
A workspace's `Workspaces.artifact_bucket` column (when non-null)
|
|
overrides the default for that workspace, regardless of `usage_type`.
|
|
|
|
### Add a new schedule node type
|
|
|
|
`schedule/src/schedule/execution/runners/notebook.py:execute_artifact`
|
|
dispatches on `script_type`. Add a new branch + a new `_<type>` function
|
|
in that file. The NodeExecutor in `worker.py` does not need to change —
|
|
the dispatch happens inside `execute_artifact`.
|
|
|
|
## Tests
|
|
|
|
There is an in-tree test suite, mostly covering scripts / resources /
|
|
DAG validation / upload state transitions. It is **not** the formal
|
|
release-gate suite the project still owes (see `HANDOVER.md` §8 for
|
|
known gaps: trash reaper, cross-backend migration, ENC round-trip).
|
|
|
|
Current coverage:
|
|
|
|
| Area | Tests | Where |
|
|
|---|---|---|
|
|
| Scripts (CRUD, soft delete, parent path, same-name siblings, visibility) | 10 functions across `test_scripts.py`, `test_count_scripts.py`, `test_list_scripts_parent_path.py`, `test_storage_upload_status.py` | `backend/tests/` |
|
|
| Resources (visibility, ownership, idempotency) | ~5 in `test_resources.py` | `backend/tests/` |
|
|
| DAG validation (cycle / orphan detection) | `test_validate_dag.py` | `backend/tests/` |
|
|
| Audit log middleware | `test_audit_logging.py` | `backend/tests/` |
|
|
| Jupyter auth cache | `test_jupyter_auth_cache.py` | `backend/tests/` |
|
|
| Runtime client (directory listing, error mapping) | `test_runtime_client_directories.py` | `backend/tests/` |
|
|
| Schedule layer (worker, janitor, layering invariants) | 11 functions in `test_janitor.py`, `test_layering.py`, `test_worker.py` | `schedule/tests/` |
|
|
|
|
Run:
|
|
|
|
```bash
|
|
# Backend
|
|
uv run --package backend pytest backend/tests -q
|
|
|
|
# Schedule
|
|
uv run --package schedule pytest schedule/tests -q
|
|
```
|
|
|
|
Conventions:
|
|
|
|
- pytest + `pytest-asyncio` for `async def` handlers.
|
|
- MySQL is required for the ORM tests (not SQLite — CHAR(26) ULIDs and
|
|
`mysql.TINYINT(1)` quirks do not translate). Local docker-compose
|
|
MySQL is the typical target.
|
|
- For storage, `common.storage.factory` selects between s3 and local
|
|
via `settings.storage_backend`; tests that exercise both modes
|
|
monkeypatch that setting (see `common/tests/storage/test_factory.py`).
|
|
- For HTTP boundaries (httpx to backend / runtime), tests use `respx`
|
|
with `assert_all_called=False` so unused stubs don't fail the test
|
|
— see the engineering notes in `CLAUDE.md`.
|
|
|
|
## Troubleshooting
|
|
|
|
### "the greenlet library is required"
|
|
|
|
SQLAlchemy 2.0 needs `greenlet` for `engine.dispose()` in async
|
|
contexts. Add `greenlet>=3.0.0` to `common/pyproject.toml` and
|
|
`uv sync --all-packages`. (Already present in this repo.)
|
|
|
|
### "Can't connect to MySQL server"
|
|
|
|
Either MySQL isn't running, or the network namespace doesn't allow
|
|
`mysql:3306` resolution. Inside the Docker network, services reach
|
|
each other by service name (`mysql`, `backend`, `runtime`,
|
|
`schedule`, `s3`).
|
|
|
|
### Jupyter routing 401s
|
|
|
|
Inspect `docker compose logs backend` — `jupyter.py:check_notebook_is_locked`
|
|
or `load_active_membership` will return an explicit reason. Then
|
|
check the JWT (use `JWT_SECRET` from `.env`).
|
|
|
|
### Schedule run never advances
|
|
|
|
`schedule_runs.run_status` is stuck at `queued`. Two likely causes:
|
|
- `outbox_events` is empty (Backend's `add_outbox_event` failed —
|
|
check `add_outbox_event` in `schedule_runs.py`).
|
|
- The orchestrator's polling loop is dead. Check
|
|
`docker compose logs schedule` and look for "database event loop
|
|
failed" exceptions.
|
|
|
|
### "AttributeError: 'SchedulerService' object has no attribute 'worker'"
|
|
|
|
`worker` must be constructed before `orchestrator` in
|
|
`SchedulerService.__init__`, because orchestrator's dispatch table
|
|
captures `self.worker.handle_node_execute` at construction time.
|
|
See `schedule/src/schedule/application/service.py` — the order is
|
|
load-bearing. (This was the symptom during the flat → layered schedule
|
|
refactor; if you see it today, the most likely cause is a partial
|
|
rebase that left an old import path.)
|
|
|
|
## Style
|
|
|
|
- Type hints everywhere (this repo uses `from __future__ import
|
|
annotations`).
|
|
- 4-space indent, double quotes, no trailing whitespace.
|
|
- Comments are technical (explain *why*, not *what*).
|
|
- Module docstrings document non-obvious invariants. Don't add
|
|
docstrings to functions whose behavior is self-evident from the
|
|
name.
|
|
- 4 levels of indentation = "this function is doing too much; split
|
|
it". (Project convention; see e.g. `execute_artifact`.)
|
|
|
|
## See also
|
|
|
|
- `ARCHITECTURE.md` — kept as a thin redirect; the authoritative
|
|
architecture diagrams and capability map now live in
|
|
[§Architecture](#architecture) above.
|
|
- `HANDOVER.md` — current refactor state, recent commits, pending work.
|
|
- `CLAUDE.md` — agent-facing conventions for the repo.
|
|
- `common/src/common/db/models/__init__.py` — exhaustive list of all 18 tables.
|
|
- `common/src/common/config.py` — all env vars in one place.
|
|
- Per-package READMEs: `backend/README.md`, `common/README.md`,
|
|
`runtime/README.md`, `frontend/README.md`.
|