feat: 멀티테넌트 1단계 — 사용자별 KIS 계좌·포트폴리오·수동매매 (MULTI_TENANT_ENABLED 기본 OFF)
설계: docs/multi_tenant_design.md / 검수: docs/multi_tenant_phase1_review.md (High 0, MT=False 배포 가능)
데이터·인프라
- user_kis_credentials(Fernet 암호화, PBKDF2 키 파생, key_version), stock_bans_v2 신설,
portfolio/trade_history/active_trades/trade_journals/trading_sessions 등 user_id nullable 추가
- alembic 0009~0011(idempotent, legacy owner 백필). main.py 기동 시 portfolio.user_id 스키마 검사(H1)
- Jenkinsfile: DB Migration 을 Deploy 앞으로, alembic 이력 없으면 stamp 0008 후 upgrade(N1)
- bootstrap: users 비어 있으면 BOOTSTRAP_ADMIN_* 로 관리자 시드 + .env KIS 자격증명 이관 + 백필(SAVEPOINT)
백엔드
- core/crypto.py, core/tenant_auth.py(get_current_user — 비활성 사용자 401, API 키→legacy owner,
deny_non_owner_session_when_mt), core/dt_utils.py
- services/tenant: context, kis_pool, registry(UserContextRegistry, 락 최소화·복호화 to_thread), risk_factory
- KoreaInvestmentAPI.for_user(사용자별 인스턴스, 토큰캐시 users/{id}/, 시세는 공용 위임)
- trade_executor.safe_place_order_v2(사용자별 락/세마포어, legacy ctx 는 기존 키)
- mdd_manager/loss_tracker _v2 + ban_until tz-aware/naive 경계 정규화(레거시 경로 버그 포함 수정)
- db_worker: payload user_id 옵션 필터, upsert_stock_ban_v2, legacy ActiveTrade 조회 LIMIT 1
- API: /account/credentials(등록·마스킹조회·삭제·연결테스트·자동매매), /auth/admin/tenant/*,
/portfolio/v2(조회·요약·sync·buy·sell), /trades/v2(+pending). PortfolioResponse Optional 수정
프론트
- /account 내 계좌 설정, useTenantMode(세션=v2 / API키=legacy, 미확정 시 쿼리·버튼 비활성),
계좌 미등록 인라인 안내, v2 매수/매도/거래이력 분기
테스트: tests/integration(격리 Postgres, KIS 전면 mock) 20 passed / 1 xfailed(0012 이전 제약)
2단계(스케줄러 사용자화·legacy 라우터 격리) 전까지 MULTI_TENANT_ENABLED=True 전환 금지
Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
This commit is contained in:
@@ -17,6 +17,15 @@ Supported task types:
|
||||
- upsert_portfolio : dict → Portfolio INSERT/UPDATE (strategy_type 포함)
|
||||
- upsert_active_trade : dict → ActiveTrade UPSERT (strategy_name 포함)
|
||||
- close_active_trade : dict → ActiveTrade 상태를 'closed'로 변경
|
||||
- upsert_stock_ban_v2 : dict → StockBanV2 UPSERT (user_id, stock_code 스코프 — 멀티테넌트)
|
||||
|
||||
멀티테넌트(사용자 스코프) 확장 (docs/multi_tenant_design.md §3.4):
|
||||
- insert_trade_history / update_trade_history(_by_order_id) / upsert_portfolio /
|
||||
upsert_active_trade / close_active_trade / insert_trade_journal 는 payload
|
||||
data["user_id"] 가 있으면 조회 필터·생성 컬럼에 반영하고, 없으면(None) 기존과
|
||||
100% 동일하게 동작한다(하위 호환). 새 task 타입을 추가하지 않는다.
|
||||
- stock_bans 는 PK가 stock_code 단일이라 사용자 스코프로 확장 불가 → 신규 테이블
|
||||
stock_bans_v2(user_id, stock_code UNIQUE 쌍) + 신규 task upsert_stock_ban_v2 사용.
|
||||
"""
|
||||
import asyncio
|
||||
import json
|
||||
@@ -183,6 +192,25 @@ async def enqueue_stock_ban_upsert(stock_code: str, consecutive_losses: int, ban
|
||||
})
|
||||
|
||||
|
||||
async def enqueue_stock_ban_upsert_v2(
|
||||
user_id: int, stock_code: str, consecutive_losses: int, ban_until
|
||||
) -> None:
|
||||
"""[v2] StockBanV2 UPSERT 를 큐에 등록 (사용자 스코프 loss_tracker에서 호출).
|
||||
|
||||
stock_bans(PK=stock_code, 단일 계정 전제)와 달리 stock_bans_v2 는
|
||||
UNIQUE(user_id, stock_code) 이므로 동일 종목을 사용자별로 독립 추적한다.
|
||||
"""
|
||||
await db_task_queue.put({
|
||||
"type": "upsert_stock_ban_v2",
|
||||
"data": {
|
||||
"user_id": user_id,
|
||||
"stock_code": stock_code,
|
||||
"consecutive_losses": consecutive_losses,
|
||||
"ban_until": ban_until,
|
||||
}
|
||||
})
|
||||
|
||||
|
||||
async def enqueue_portfolio_upsert(data: dict) -> None:
|
||||
"""Portfolio INSERT/UPDATE 를 큐에 등록.
|
||||
|
||||
@@ -211,6 +239,20 @@ async def enqueue_close_active_trade(stock_code: str, strategy_name: str | None
|
||||
})
|
||||
|
||||
|
||||
async def enqueue_close_active_trade_v2(
|
||||
user_id: int, stock_code: str, strategy_name: str | None = None
|
||||
) -> None:
|
||||
"""[v2] 사용자 스코프 ActiveTrade 를 'closed' 상태로 변경하는 태스크를 큐에 등록.
|
||||
|
||||
close_active_trade 와 동일 task type("close_active_trade")을 재사용한다 —
|
||||
핸들러가 data.get("user_id")로 필터를 추가하므로 새 타입이 필요 없다.
|
||||
"""
|
||||
await db_task_queue.put({
|
||||
"type": "close_active_trade",
|
||||
"data": {"stock_code": stock_code, "strategy_name": strategy_name, "user_id": user_id},
|
||||
})
|
||||
|
||||
|
||||
async def enqueue_update_trade_history_by_order_id(order_id: str, updates: dict) -> None:
|
||||
"""KIS 주문번호(order_id)로 TradeHistory UPDATE 를 큐에 등록.
|
||||
|
||||
@@ -288,10 +330,12 @@ async def db_writer_worker() -> None:
|
||||
# ── 2. TradeHistory UPDATE (체결 확인 후 슬리피지/수량 반영) ──
|
||||
elif task_type == "update_trade_history":
|
||||
trade_id = data.get("id")
|
||||
user_id = data.get("user_id")
|
||||
if trade_id:
|
||||
row = await session.execute(
|
||||
select(TradeHistory).where(TradeHistory.id == trade_id)
|
||||
)
|
||||
query = select(TradeHistory).where(TradeHistory.id == trade_id)
|
||||
if user_id is not None:
|
||||
query = query.where(TradeHistory.user_id == user_id)
|
||||
row = await session.execute(query)
|
||||
trade = row.scalar_one_or_none()
|
||||
if trade:
|
||||
for key, val in data.items():
|
||||
@@ -307,10 +351,12 @@ async def db_writer_worker() -> None:
|
||||
# ── 2b. TradeHistory UPDATE by order_id (KIS 주문번호 기준) ──
|
||||
elif task_type == "update_trade_history_by_order_id":
|
||||
order_id = data.get("order_id")
|
||||
user_id = data.get("user_id")
|
||||
if order_id:
|
||||
row = await session.execute(
|
||||
select(TradeHistory).where(TradeHistory.order_id == order_id)
|
||||
)
|
||||
query = select(TradeHistory).where(TradeHistory.order_id == order_id)
|
||||
if user_id is not None:
|
||||
query = query.where(TradeHistory.user_id == user_id)
|
||||
row = await session.execute(query)
|
||||
trade = row.scalar_one_or_none()
|
||||
if trade:
|
||||
for key, val in data.items():
|
||||
@@ -350,10 +396,12 @@ async def db_writer_worker() -> None:
|
||||
# ── 4. StockBan UPSERT ────────────────────────────────────────
|
||||
elif task_type == "upsert_stock_ban":
|
||||
from app.models.models import StockBan
|
||||
from app.core.dt_utils import to_naive_utc
|
||||
|
||||
stock_code = data.get("stock_code")
|
||||
consecutive_losses = data.get("consecutive_losses")
|
||||
ban_until = data.get("ban_until")
|
||||
# stock_bans.ban_until은 tz-naive 컬럼 — tz-aware 값이 들어와도 안전하게 저장.
|
||||
ban_until = to_naive_utc(data.get("ban_until"))
|
||||
|
||||
result = await session.execute(select(StockBan).where(StockBan.stock_code == stock_code))
|
||||
record = result.scalar_one_or_none()
|
||||
@@ -377,6 +425,42 @@ async def db_writer_worker() -> None:
|
||||
ban_until.isoformat() if ban_until else "None",
|
||||
)
|
||||
|
||||
# ── 4v2. StockBanV2 UPSERT (user_id, stock_code 스코프) ────────
|
||||
elif task_type == "upsert_stock_ban_v2":
|
||||
from app.models.models import StockBanV2
|
||||
from sqlalchemy.dialects.postgresql import insert as pg_insert
|
||||
from app.core.dt_utils import to_naive_utc
|
||||
|
||||
user_id = data.get("user_id")
|
||||
stock_code = data.get("stock_code")
|
||||
consecutive_losses = data.get("consecutive_losses")
|
||||
# stock_bans_v2.ban_until도 tz-naive 컬럼 — tz-aware 값이 들어와도 안전하게 저장.
|
||||
ban_until = to_naive_utc(data.get("ban_until"))
|
||||
|
||||
# PostgreSQL INSERT ... ON CONFLICT DO UPDATE — 여러 워커/사용자
|
||||
# 사이클이 동시에 같은 (user_id, stock_code)를 갱신해도 경쟁 안전.
|
||||
stmt = pg_insert(StockBanV2).values(
|
||||
user_id=user_id,
|
||||
stock_code=stock_code,
|
||||
consecutive_losses=consecutive_losses,
|
||||
ban_until=ban_until,
|
||||
)
|
||||
stmt = stmt.on_conflict_do_update(
|
||||
constraint="uq_stock_bans_v2_user_stock",
|
||||
set_={
|
||||
"consecutive_losses": stmt.excluded.consecutive_losses,
|
||||
"ban_until": stmt.excluded.ban_until,
|
||||
"updated_at": datetime.utcnow(),
|
||||
},
|
||||
)
|
||||
await session.execute(stmt)
|
||||
await session.commit()
|
||||
logger.debug(
|
||||
"✅ [Worker] StockBanV2 upsert: user=%s %s (losses=%s, ban_until=%s)",
|
||||
user_id, stock_code, consecutive_losses,
|
||||
ban_until.isoformat() if ban_until else "None",
|
||||
)
|
||||
|
||||
# ── 4b. News INSERT (중복 URL 무시) ────────────────────────────
|
||||
elif task_type == "insert_news":
|
||||
from app.models.models import News
|
||||
@@ -417,9 +501,11 @@ async def db_writer_worker() -> None:
|
||||
from app.models.models import Portfolio
|
||||
|
||||
stock_code = data.get("stock_code")
|
||||
result = await session.execute(
|
||||
select(Portfolio).where(Portfolio.stock_code == stock_code)
|
||||
)
|
||||
user_id = data.get("user_id")
|
||||
query = select(Portfolio).where(Portfolio.stock_code == stock_code)
|
||||
if user_id is not None:
|
||||
query = query.where(Portfolio.user_id == user_id)
|
||||
result = await session.execute(query)
|
||||
record = result.scalar_one_or_none()
|
||||
|
||||
if record is None:
|
||||
@@ -443,6 +529,7 @@ async def db_writer_worker() -> None:
|
||||
|
||||
stock_code = data.get("stock_code")
|
||||
strategy_name = data.get("strategy_name")
|
||||
user_id = data.get("user_id")
|
||||
|
||||
query = select(ActiveTrade).where(
|
||||
ActiveTrade.stock_code == stock_code,
|
||||
@@ -450,9 +537,17 @@ async def db_writer_worker() -> None:
|
||||
)
|
||||
if strategy_name:
|
||||
query = query.where(ActiveTrade.strategy_name == strategy_name)
|
||||
|
||||
result = await session.execute(query)
|
||||
record = result.scalar_one_or_none()
|
||||
if user_id is not None:
|
||||
query = query.where(ActiveTrade.user_id == user_id)
|
||||
result = await session.execute(query)
|
||||
record = result.scalar_one_or_none()
|
||||
else:
|
||||
# legacy(user_id=None) 경로: 사용자 다중화로 동일 종목 active
|
||||
# 행이 2건 이상 존재할 수 있어 scalar_one_or_none() 이
|
||||
# MultipleResultsFound 를 던질 수 있음 → 가장 최근 행 우선.
|
||||
query = query.order_by(ActiveTrade.id.desc()).limit(1)
|
||||
result = await session.execute(query)
|
||||
record = result.scalars().first()
|
||||
|
||||
if record is None:
|
||||
# remaining_pct_multiply는 신규 생성 시 무시
|
||||
@@ -486,6 +581,7 @@ async def db_writer_worker() -> None:
|
||||
|
||||
stock_code = data.get("stock_code")
|
||||
strategy_name = data.get("strategy_name")
|
||||
user_id = data.get("user_id")
|
||||
|
||||
query = select(ActiveTrade).where(
|
||||
ActiveTrade.stock_code == stock_code,
|
||||
@@ -493,9 +589,17 @@ async def db_writer_worker() -> None:
|
||||
)
|
||||
if strategy_name:
|
||||
query = query.where(ActiveTrade.strategy_name == strategy_name)
|
||||
|
||||
result = await session.execute(query)
|
||||
record = result.scalar_one_or_none()
|
||||
if user_id is not None:
|
||||
query = query.where(ActiveTrade.user_id == user_id)
|
||||
result = await session.execute(query)
|
||||
record = result.scalar_one_or_none()
|
||||
else:
|
||||
# legacy(user_id=None) 경로: 사용자 다중화로 동일 종목 active
|
||||
# 행이 2건 이상 존재할 수 있어 scalar_one_or_none() 이
|
||||
# MultipleResultsFound 를 던질 수 있음 → 가장 최근 행 우선.
|
||||
query = query.order_by(ActiveTrade.id.desc()).limit(1)
|
||||
result = await session.execute(query)
|
||||
record = result.scalars().first()
|
||||
|
||||
if record:
|
||||
record.status = "closed"
|
||||
@@ -543,6 +647,7 @@ async def db_writer_worker() -> None:
|
||||
"upsert_stock_analysis",
|
||||
"bulk_upsert_stock_analysis",
|
||||
"upsert_stock_ban",
|
||||
"upsert_stock_ban_v2",
|
||||
"upsert_portfolio",
|
||||
"upsert_active_trade",
|
||||
"close_active_trade",
|
||||
|
||||
Reference in New Issue
Block a user