Nexus Deploy
4ba45abf38
- 密码/SSH密钥预设增加用户名；快速添加(IP)轮询全部预设尝试 SSH 登录 - 成功入 servers 列表；失败写入 pending_servers 并支持重试/删除 - 新增 credential_poller、try_ssh_login 与 add-by-ip/pending API
82 lines
2.5 KiB
Python
82 lines
2.5 KiB
Python
"""Nexus — PendingServer Repository (SSH poll failures queue)"""
|
|
|
|
from datetime import datetime, timezone
|
|
from typing import Optional, List
|
|
|
|
from sqlalchemy import select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from server.domain.models import PendingServer
|
|
|
|
|
|
class PendingServerRepositoryImpl:
|
|
def __init__(self, session: AsyncSession):
|
|
self.session = session
|
|
|
|
async def get_by_id(self, id: int) -> Optional[PendingServer]:
|
|
result = await self.session.execute(select(PendingServer).where(PendingServer.id == id))
|
|
return result.scalar_one_or_none()
|
|
|
|
async def get_all(self) -> List[PendingServer]:
|
|
result = await self.session.execute(
|
|
select(PendingServer).order_by(PendingServer.last_attempt_at.desc(), PendingServer.id.desc())
|
|
)
|
|
return list(result.scalars().all())
|
|
|
|
async def create(self, pending: PendingServer) -> PendingServer:
|
|
self.session.add(pending)
|
|
await self.session.commit()
|
|
await self.session.refresh(pending)
|
|
return pending
|
|
|
|
async def update(self, pending: PendingServer) -> PendingServer:
|
|
await self.session.commit()
|
|
await self.session.refresh(pending)
|
|
return pending
|
|
|
|
async def delete(self, id: int) -> bool:
|
|
row = await self.get_by_id(id)
|
|
if not row:
|
|
return False
|
|
await self.session.delete(row)
|
|
await self.session.commit()
|
|
return True
|
|
|
|
async def record_failure(
|
|
self,
|
|
*,
|
|
name: str,
|
|
domain: str,
|
|
port: int,
|
|
agent_port: int,
|
|
target_path: str | None,
|
|
category: str | None,
|
|
platform_id: int | None,
|
|
node_id: int | None,
|
|
last_error: str,
|
|
existing_id: int | None = None,
|
|
) -> PendingServer:
|
|
now = datetime.now(timezone.utc)
|
|
if existing_id:
|
|
row = await self.get_by_id(existing_id)
|
|
if row:
|
|
row.attempts = (row.attempts or 0) + 1
|
|
row.last_error = last_error[:4000]
|
|
row.last_attempt_at = now
|
|
return await self.update(row)
|
|
|
|
pending = PendingServer(
|
|
name=name,
|
|
domain=domain,
|
|
port=port,
|
|
agent_port=agent_port,
|
|
target_path=target_path,
|
|
category=category,
|
|
platform_id=platform_id,
|
|
node_id=node_id,
|
|
attempts=1,
|
|
last_error=last_error[:4000],
|
|
last_attempt_at=now,
|
|
)
|
|
return await self.create(pending)
|