perf: 优化 MySQL 查询并接入 pytest
This commit is contained in:
@@ -5,7 +5,7 @@ import threading
|
||||
from datetime import datetime
|
||||
from fastapi import APIRouter, Depends, HTTPException, WebSocket, WebSocketDisconnect
|
||||
from sqlalchemy import func
|
||||
from sqlalchemy.orm import Session, defer
|
||||
from sqlalchemy.orm import Session, defer, load_only
|
||||
|
||||
from ..database import get_db, SessionLocal
|
||||
from ..models import User, Account, LoginTask, ProxyConfig as ProxyConfigModel
|
||||
@@ -120,7 +120,12 @@ def list_tasks(
|
||||
account_ids = [t.account_id for t in rows]
|
||||
accounts_map = {}
|
||||
if account_ids:
|
||||
accs = db.query(Account).filter(Account.id.in_(account_ids)).all()
|
||||
accs = (
|
||||
db.query(Account)
|
||||
.options(load_only(Account.id, Account.username))
|
||||
.filter(Account.id.in_(account_ids))
|
||||
.all()
|
||||
)
|
||||
accounts_map = {a.id: a.username for a in accs}
|
||||
|
||||
result = []
|
||||
|
||||
Reference in New Issue
Block a user