Implement REQ-005~009 (dcy-002 slice 1) as an onion stack on top of the
BAT-001 pure domain:
- application: IntentService (admission + single-writer transition +
intent_id idempotency via durable intent_log)
- adapters/persistence: PostgreSQLSlotRepository + PostgreSQLIntentStore +
trex-core-owned migrations (daily_slot per Appendix A, daily_symbol_entries,
slot_transition_log, intent_log) + in-memory intent store
- adapters/api: FastAPI /v1 (healthz, intents/buy|sell|{id}, slots, capacity)
- runtime: trex-core-daemon entrypoint with lifespan migrate+preallocate
Verification (against local PostgreSQL 18): 74 passed, ruff clean, domain
import sweep shows zero web/db deps (NFR-1), 20-thread concurrency admits
exactly the idle count (NFR-2), idempotent resubmit causes no double
transition (NFR-3), EXPECTED_SLOT_FIELDS<->information_schema drift test
(realizes dcy-001 OQ-2's deferred DB-side check), and a daemon E2E smoke.
Align all four baseline docs to the new daemon/API/persistence layers.
BAT-002 -> release-candidate; REQ-005~009 -> Implemented. Full async outbox,
broker execution, safety floor, and strategy daemons remain deferred to a
follow-up Discovery.
Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
115 lines
4.2 KiB
Python
115 lines
4.2 KiB
Python
"""FastAPI app factory for the slot-core daemon (REQ-005/007/008).
|
|
|
|
`create_app` wires the HTTP /v1 surface to injected services so the same app
|
|
runs against the in-memory adapters (tests) or PostgreSQL adapters (daemon).
|
|
All slot writes go through IntentService → SlotService (single writer).
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
from datetime import date as date_cls
|
|
from typing import Optional
|
|
|
|
from fastapi import FastAPI, HTTPException, Query
|
|
|
|
from trex_core.adapters.api.schemas import (
|
|
BuyIntentRequest,
|
|
CapacityResponse,
|
|
HealthResponse,
|
|
IntentResponse,
|
|
SellIntentRequest,
|
|
SlotSnapshot,
|
|
)
|
|
from trex_core.application.intents import BuyIntent, IntentService, SellIntent
|
|
from trex_core.domain.slot import SlotRepository, SlotService
|
|
|
|
|
|
def create_app(
|
|
*,
|
|
slot_service: SlotService,
|
|
slot_repo: SlotRepository,
|
|
intent_service: IntentService,
|
|
default_total_slots: int,
|
|
) -> FastAPI:
|
|
app = FastAPI(title="trex-core slot daemon", version="0.1.0")
|
|
|
|
@app.get("/v1/healthz", response_model=HealthResponse)
|
|
def healthz() -> HealthResponse:
|
|
return HealthResponse()
|
|
|
|
@app.post("/v1/intents/buy", response_model=IntentResponse)
|
|
def submit_buy(req: BuyIntentRequest) -> IntentResponse:
|
|
d = intent_service.submit_buy(
|
|
BuyIntent(
|
|
intent_id=req.intent_id,
|
|
trading_date=req.trading_date,
|
|
symbol=req.symbol,
|
|
qty=req.qty,
|
|
allocated_capital=req.allocated_capital,
|
|
entry_price=req.entry_price,
|
|
strategy_id=req.strategy_id,
|
|
sell_strategy_id=req.sell_strategy_id,
|
|
portfolio_id=req.portfolio_id,
|
|
forced_close_at=req.forced_close_at,
|
|
)
|
|
)
|
|
return IntentResponse(intent_id=d.intent_id, status=d.status, slot_id=d.slot_id, reason=d.reason)
|
|
|
|
@app.post("/v1/intents/sell", response_model=IntentResponse)
|
|
def submit_sell(req: SellIntentRequest) -> IntentResponse:
|
|
d = intent_service.submit_sell(
|
|
SellIntent(
|
|
intent_id=req.intent_id,
|
|
slot_id=req.slot_id,
|
|
reason=req.reason,
|
|
sell_strategy_id=req.sell_strategy_id,
|
|
)
|
|
)
|
|
return IntentResponse(intent_id=d.intent_id, status=d.status, slot_id=d.slot_id, reason=d.reason)
|
|
|
|
@app.get("/v1/intents/{intent_id}", response_model=IntentResponse)
|
|
def get_intent(intent_id: str) -> IntentResponse:
|
|
d = intent_service.lookup(intent_id)
|
|
if d is None:
|
|
raise HTTPException(status_code=404, detail="intent not found")
|
|
return IntentResponse(intent_id=d.intent_id, status=d.status, slot_id=d.slot_id, reason=d.reason)
|
|
|
|
@app.get("/v1/slots", response_model=list[SlotSnapshot])
|
|
def list_slots(
|
|
status: Optional[str] = Query(default=None),
|
|
trading_date: Optional[date_cls] = Query(default=None),
|
|
) -> list[SlotSnapshot]:
|
|
day = trading_date or _today()
|
|
slots = slot_repo.list_slots(day)
|
|
if status is not None:
|
|
slots = [s for s in slots if s.status.value == status]
|
|
return [
|
|
SlotSnapshot(
|
|
slot_id=s.slot_id, trading_date=s.trading_date, status=s.status.value,
|
|
symbol=s.symbol, qty=s.qty, avg_price=s.avg_price, entry_price=s.entry_price,
|
|
allocated_capital=s.allocated_capital, strategy_id=s.strategy_id,
|
|
sell_strategy_id=s.sell_strategy_id, forced_close_at=s.forced_close_at,
|
|
)
|
|
for s in slots
|
|
]
|
|
|
|
@app.get("/v1/capacity", response_model=CapacityResponse)
|
|
def capacity(
|
|
trading_date: Optional[date_cls] = Query(default=None),
|
|
total_slots: Optional[int] = Query(default=None),
|
|
) -> CapacityResponse:
|
|
day = trading_date or _today()
|
|
cap = slot_service.compute_slot_capacity(day, total_slots or default_total_slots)
|
|
return CapacityResponse(
|
|
idle_count=cap.idle_count, holding_count=cap.holding_count,
|
|
pending_count=cap.pending_count, total_slots=cap.total_slots,
|
|
scarcity_ratio=cap.scarcity_ratio,
|
|
)
|
|
|
|
return app
|
|
|
|
|
|
def _today() -> date_cls:
|
|
from datetime import datetime, timezone
|
|
|
|
return datetime.now(tz=timezone.utc).date()
|