Mina Choi
479edf9403
로그인한 사장님이 자기 사이트를 볼 화면이 없었다. 사이트는 place_id 로 한 건씩만 읽혀서 (site_crud.get_site_by_place) 사업장 목록으로 그리면 줄마다 사이트를 다시 물어 N+1 이 된다. - site_crud.list_company_sites: places LEFT JOIN sites LEFT JOIN site_versions 한 번. 사이트가 아직 없는 사업장(위저드만 걸어온 것)도 내려간다 — 빠지면 만들다 만 것을 찾을 길이 없다 - protocol.MySiteData: 한 줄 = 사업장 + 사이트. render(정적 파일 존재)는 넣지 않았다 — 보고서 파일을 읽는 값이라 줄 수만큼 파일 IO 가 된다. 단건(Res_Site)이 계속 소유한다 - site_service.list_my_sites: 회사 스코프. needs_rebuild 는 단건과 같은 규칙으로 판정한다 - GET /v1/site/list 는 라우터 객체를 따로 둔다 — 기존 라우터는 접두어에 place_id 가 박혀 있다 테스트 5건 추가(비어 있는 사업장·조인·회사 격리·재빌드 일치·비로그인), 539 passed (기존 실패 4건은 이 변경 전에도 같다 — build_publish 3 · snapshot 1)
233 lines
9.9 KiB
Python
233 lines
9.9 KiB
Python
from abc import ABC, abstractmethod
|
|
from typing import Tuple
|
|
|
|
from sqlalchemy import and_, func, select, update
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from common.database.db_session_manager import DB_SESSION_MNG
|
|
from common.database.model.models import places, publish_logs, site_versions, sites
|
|
from common.enums import BuildStatus, ErrorType
|
|
from common.logger import LOG
|
|
from common.utils.gtime import GTime
|
|
|
|
|
|
# 사이트/버전/발행로그 CRUD. 항상 place_id 또는 site_id 로 스코프한다.
|
|
class ISiteCRUD(ABC):
|
|
@abstractmethod
|
|
async def get_site_by_place(self, cdb: AsyncSession, place_id) -> Tuple[ErrorType, sites]:
|
|
pass
|
|
|
|
@abstractmethod
|
|
async def get_site_by_domain(self, cdb: AsyncSession, domain: str) -> Tuple[ErrorType, sites]:
|
|
pass
|
|
|
|
@abstractmethod
|
|
async def list_company_sites(self, cdb: AsyncSession, company_id, skip, limit) -> Tuple[ErrorType, list, int]:
|
|
pass
|
|
|
|
@abstractmethod
|
|
async def taken_domains(self, cdb: AsyncSession, domains: list) -> Tuple[ErrorType, set]:
|
|
pass
|
|
|
|
@abstractmethod
|
|
async def add_site(self, cdb: AsyncSession, site: sites) -> ErrorType:
|
|
pass
|
|
|
|
@abstractmethod
|
|
async def next_version_no(self, cdb: AsyncSession, site_id) -> Tuple[ErrorType, int]:
|
|
pass
|
|
|
|
@abstractmethod
|
|
async def add_version(self, cdb: AsyncSession, version: site_versions) -> ErrorType:
|
|
pass
|
|
|
|
@abstractmethod
|
|
async def get_version(self, cdb: AsyncSession, site_id, site_version_id) -> Tuple[ErrorType, site_versions]:
|
|
pass
|
|
|
|
@abstractmethod
|
|
async def finish_version(self, cdb: AsyncSession, site_version_id, data: dict) -> Tuple[ErrorType, int]:
|
|
pass
|
|
|
|
@abstractmethod
|
|
async def update_site(self, cdb: AsyncSession, site_id, data: dict) -> Tuple[ErrorType, int]:
|
|
pass
|
|
|
|
@abstractmethod
|
|
async def add_log(self, cdb: AsyncSession, log: publish_logs) -> ErrorType:
|
|
pass
|
|
|
|
@abstractmethod
|
|
async def list_logs(self, cdb: AsyncSession, site_id, limit: int) -> Tuple[ErrorType, list]:
|
|
pass
|
|
|
|
|
|
class SiteCRUD(ISiteCRUD):
|
|
async def get_site_by_place(self, cdb: AsyncSession, place_id) -> Tuple[ErrorType, sites]:
|
|
try:
|
|
query = select(sites).where(sites.place_id == place_id, sites.deleted == False).limit(1) # noqa: E712
|
|
err_type, rows = await DB_SESSION_MNG.execute(cdb, query)
|
|
if err_type != ErrorType.SUCCESS:
|
|
return err_type, None
|
|
return ErrorType.SUCCESS, (rows[0] if rows else None)
|
|
except Exception as ex:
|
|
LOG.e_no_callstack(ex)
|
|
return ErrorType.DB_RUN_FAILED, None
|
|
|
|
async def get_site_by_domain(self, cdb: AsyncSession, domain: str) -> Tuple[ErrorType, sites]:
|
|
"""주소(도메인 라벨)의 주인을 찾는다. 없으면 (SUCCESS, None).
|
|
|
|
uq_sites_domain(deleted=false AND domain IS NOT NULL)과 같은 조건으로 본다 —
|
|
인덱스가 막는 것과 조회가 막는 것이 다르면 "확인은 통과, 저장은 실패"가 난다."""
|
|
try:
|
|
query = select(sites).where(sites.domain == domain, sites.deleted == False).limit(1) # noqa: E712
|
|
err_type, rows = await DB_SESSION_MNG.execute(cdb, query)
|
|
if err_type != ErrorType.SUCCESS:
|
|
return err_type, None
|
|
return ErrorType.SUCCESS, (rows[0] if rows else None)
|
|
except Exception as ex:
|
|
LOG.e_no_callstack(ex)
|
|
return ErrorType.DB_RUN_FAILED, None
|
|
|
|
async def list_company_sites(self, cdb: AsyncSession, company_id, skip: int, limit: int) -> Tuple[ErrorType, list, int]:
|
|
"""회사의 사업장 + 사이트 + 마지막 빌드 시각. (ErrorType, [(place, site, built_at)], 총건수).
|
|
|
|
따로 읽으면 줄마다 사이트를 다시 물어 N+1 이다. LEFT JOIN 이라 사이트가 없는 사업장
|
|
(위저드만 걸어온 것)도 내려간다 — 빠지면 만들다 만 것을 찾을 길이 없다."""
|
|
try:
|
|
where = and_(places.deleted == False, places.company_id == company_id) # noqa: E712
|
|
|
|
cnt_err, cnt_rows = await DB_SESSION_MNG.execute(cdb, select(func.count()).select_from(places).where(where))
|
|
if cnt_err != ErrorType.SUCCESS:
|
|
return cnt_err, [], 0
|
|
total = int(cnt_rows[0] or 0) if cnt_rows else 0
|
|
|
|
query = (
|
|
select(places, sites, site_versions.built_at)
|
|
.outerjoin(sites, and_(sites.place_id == places.place_id, sites.deleted == False)) # noqa: E712
|
|
.outerjoin(site_versions, site_versions.site_version_id == sites.current_version_id)
|
|
.where(where)
|
|
.order_by(places.created_at.desc())
|
|
.offset(skip)
|
|
.limit(limit)
|
|
)
|
|
list_err, rows = await DB_SESSION_MNG.execute(cdb, query)
|
|
if list_err != ErrorType.SUCCESS:
|
|
return list_err, [], 0
|
|
return ErrorType.SUCCESS, list(rows), total
|
|
except Exception as ex:
|
|
LOG.e_no_callstack(ex)
|
|
return ErrorType.DB_RUN_FAILED, [], 0
|
|
|
|
async def taken_domains(self, cdb: AsyncSession, domains: list) -> Tuple[ErrorType, set]:
|
|
"""후보 주소들 중 이미 쓰이는 것만 추린다. 대안 제안이 후보마다 왕복하지 않게 한 번에 본다."""
|
|
try:
|
|
if not domains:
|
|
return ErrorType.SUCCESS, set()
|
|
query = select(sites.domain).where(
|
|
sites.domain.in_(domains), sites.deleted == False # noqa: E712
|
|
)
|
|
err_type, rows = await DB_SESSION_MNG.execute(cdb, query)
|
|
if err_type != ErrorType.SUCCESS:
|
|
return err_type, set()
|
|
return ErrorType.SUCCESS, {r for r in rows if r}
|
|
except Exception as ex:
|
|
LOG.e_no_callstack(ex)
|
|
return ErrorType.DB_RUN_FAILED, set()
|
|
|
|
async def add_site(self, cdb: AsyncSession, site: sites) -> ErrorType:
|
|
try:
|
|
return await DB_SESSION_MNG.insert(cdb, site)
|
|
except Exception as ex:
|
|
LOG.e_no_callstack(ex)
|
|
return ErrorType.DB_RUN_FAILED
|
|
|
|
async def next_version_no(self, cdb: AsyncSession, site_id) -> Tuple[ErrorType, int]:
|
|
"""다음 버전 번호. 1부터 시작한다."""
|
|
try:
|
|
query = select(func.max(site_versions.version)).where(
|
|
site_versions.site_id == site_id, site_versions.deleted == False # noqa: E712
|
|
)
|
|
err_type, rows = await DB_SESSION_MNG.execute(cdb, query)
|
|
if err_type != ErrorType.SUCCESS:
|
|
return err_type, 0
|
|
current = rows[0] if rows else None
|
|
return ErrorType.SUCCESS, int(current or 0) + 1
|
|
except Exception as ex:
|
|
LOG.e_no_callstack(ex)
|
|
return ErrorType.DB_RUN_FAILED, 0
|
|
|
|
async def add_version(self, cdb: AsyncSession, version: site_versions) -> ErrorType:
|
|
try:
|
|
return await DB_SESSION_MNG.insert(cdb, version)
|
|
except Exception as ex:
|
|
LOG.e_no_callstack(ex)
|
|
return ErrorType.DB_RUN_FAILED
|
|
|
|
async def get_version(self, cdb: AsyncSession, site_id, site_version_id) -> Tuple[ErrorType, site_versions]:
|
|
try:
|
|
query = (
|
|
select(site_versions)
|
|
.where(
|
|
site_versions.site_version_id == site_version_id,
|
|
site_versions.site_id == site_id,
|
|
site_versions.deleted == False, # noqa: E712
|
|
)
|
|
.limit(1)
|
|
)
|
|
err_type, rows = await DB_SESSION_MNG.execute(cdb, query)
|
|
if err_type != ErrorType.SUCCESS:
|
|
return err_type, None
|
|
if len(rows) != 1:
|
|
return ErrorType.DB_INVALID_KEY, None
|
|
return ErrorType.SUCCESS, rows[0]
|
|
except Exception as ex:
|
|
LOG.e_no_callstack(ex)
|
|
return ErrorType.DB_RUN_FAILED, None
|
|
|
|
async def finish_version(self, cdb: AsyncSession, site_version_id, data: dict) -> Tuple[ErrorType, int]:
|
|
"""빌드 결과를 버전에 기록한다(BUILT 또는 FAILED)."""
|
|
try:
|
|
query = (
|
|
update(site_versions)
|
|
.where(site_versions.site_version_id == site_version_id, site_versions.deleted == False) # noqa: E712
|
|
.values(**data, updated_at=GTime.UTC())
|
|
)
|
|
return await DB_SESSION_MNG.add_with_rowcount(cdb, query)
|
|
except Exception as ex:
|
|
LOG.e_no_callstack(ex)
|
|
return ErrorType.DB_RUN_FAILED, 0
|
|
|
|
async def update_site(self, cdb: AsyncSession, site_id, data: dict) -> Tuple[ErrorType, int]:
|
|
try:
|
|
query = (
|
|
update(sites)
|
|
.where(sites.site_id == site_id, sites.deleted == False) # noqa: E712
|
|
.values(**data, updated_at=GTime.UTC())
|
|
)
|
|
return await DB_SESSION_MNG.add_with_rowcount(cdb, query)
|
|
except Exception as ex:
|
|
LOG.e_no_callstack(ex)
|
|
return ErrorType.DB_RUN_FAILED, 0
|
|
|
|
async def add_log(self, cdb: AsyncSession, log: publish_logs) -> ErrorType:
|
|
try:
|
|
return await DB_SESSION_MNG.insert(cdb, log)
|
|
except Exception as ex:
|
|
LOG.e_no_callstack(ex)
|
|
return ErrorType.DB_RUN_FAILED
|
|
|
|
async def list_logs(self, cdb: AsyncSession, site_id, limit: int = 50) -> Tuple[ErrorType, list]:
|
|
try:
|
|
query = (
|
|
select(publish_logs)
|
|
.where(publish_logs.site_id == site_id, publish_logs.deleted == False) # noqa: E712
|
|
.order_by(publish_logs.created_at.desc())
|
|
.limit(limit)
|
|
)
|
|
err_type, rows = await DB_SESSION_MNG.execute(cdb, query)
|
|
return (err_type, list(rows) if err_type == ErrorType.SUCCESS else [])
|
|
except Exception as ex:
|
|
LOG.e_no_callstack(ex)
|
|
return ErrorType.DB_RUN_FAILED, []
|