cancan-microstack 0.0.1__py3-none-any.whl

cancan_microstack/services/opsbffsrv/infrastructure/db/model/service_info_tbl.py

@@ -0,0 +1,51 @@
+"""
+服务信息表 ORM 模型
+"""
+from sqlalchemy import (
+    Column,
+    BigInteger,
+    String,
+    Text,
+    SmallInteger,
+    TIMESTAMP,
+)
+from sqlalchemy.dialects.postgresql import JSONB
+from sqlalchemy.sql import func
+
+from linglong_web import TableBase
+
+from cancan_microstack.public.const.caddy_consts import InternalRequestPath
+
+
+class ServiceInfoTbl(TableBase):
+    """服务信息表"""
+    __tablename__ = 'service_info_tbl'
+
+    # 主键
+    id = Column(BigInteger, primary_key=True, autoincrement=True)
+
+    # 业务字段
+    service_name = Column(String(100), nullable=False, unique=True)
+    description = Column(Text)
+    service_type = Column(String(50), default='business')
+    health_check_path = Column(String(255), default=InternalRequestPath.HEALTH_CHECK.value)
+    service_metadata = Column(JSONB, default=dict)
+    expected_status = Column(String(20), default='running')
+    desired_replicas = Column(SmallInteger, default=1)
+    actual_replicas = Column(SmallInteger, default=0)
+    last_scale_at = Column(TIMESTAMP(timezone=True))
+    scale_policy = Column(JSONB, default=dict)
+    registered_time = Column(TIMESTAMP(timezone=True), default=func.current_timestamp())
+    last_registered_time = Column(TIMESTAMP(timezone=True), default=func.current_timestamp())
+
+    # 标准字段
+    flag = Column(SmallInteger, default=0)
+    created_time = Column(TIMESTAMP(timezone=True), default=func.current_timestamp())
+    update_time = Column(TIMESTAMP(timezone=True), default=func.current_timestamp(), onupdate=func.current_timestamp())
+
+    __table_args__ = (
+        {'extend_existing': True},
+    )
+
+    def __repr__(self):
+        return f"ServiceInfoTbl(id={self.id}, service_name={self.service_name}, type={self.service_type})"

cancan_microstack/services/opsbffsrv/infrastructure/db/model/service_instance_tbl.py

@@ -0,0 +1,68 @@
+"""Service instance ORM for opsbff read-only queries."""
+from sqlalchemy import (
+    Column,
+    BigInteger,
+    String,
+    Integer,
+    Text,
+    SmallInteger,
+    TIMESTAMP,
+    Index,
+)
+from sqlalchemy.dialects.postgresql import JSONB
+from sqlalchemy.sql import func
+
+from linglong_web import TableBase
+
+
+class ServiceInstanceTbl(TableBase):
+    """服务实例表 ORM (opsbff 只读)
+
+    Mirrors the structure defined in infra to allow analytics on instance state.
+    """
+
+    __tablename__ = 'service_instance_tbl'
+
+    id = Column(BigInteger, primary_key=True, autoincrement=True)
+    service_name = Column(String(100), nullable=False)
+    instance_id = Column(String(64), nullable=False, unique=True)
+    container_name = Column(String(100), nullable=True)
+    compose_service_name = Column(String(100), nullable=True)
+    host = Column(String(100), nullable=False)
+    port = Column(Integer, nullable=False)
+    internal_port = Column(Integer, nullable=False, default=8080)
+    status = Column(String(20), nullable=False, default='UP')
+    expected_status = Column(String(20), default='UP')
+    health_check_url = Column(String(255))
+    health_status = Column(String(20), default='unknown')
+    last_health_check = Column(TIMESTAMP(timezone=True))
+    last_heartbeat = Column(TIMESTAMP(timezone=True))
+    consecutive_failures = Column(SmallInteger, default=0)
+    last_health_error = Column(Text)
+    started_at = Column(TIMESTAMP(timezone=True))
+    stopped_at = Column(TIMESTAMP(timezone=True))
+    restart_count = Column(Integer, default=0)
+    cpu_limit = Column(String(20))
+    memory_limit = Column(String(20))
+    instance_metadata = Column(JSONB, default=dict)
+    flag = Column(SmallInteger, default=0)
+    created_time = Column(TIMESTAMP(timezone=True), default=func.current_timestamp())
+    update_time = Column(
+        TIMESTAMP(timezone=True),
+        default=func.current_timestamp(),
+        onupdate=func.current_timestamp(),
+    )
+
+    __table_args__ = (
+        Index('idx_service_instance_tbl_service_instance', 'service_name', 'instance_id'),
+        Index('idx_service_instance_tbl_status', 'service_name', 'status'),
+        Index('idx_service_instance_tbl_health_status', 'service_name', 'health_status'),
+        Index('idx_service_instance_tbl_last_heartbeat', 'service_name', 'last_heartbeat'),
+        {'extend_existing': True},
+    )
+
+    def __repr__(self) -> str:
+        return (
+            f"ServiceInstanceTbl(id={self.id}, service={self.service_name}, "
+            f"instance={self.instance_id}, status={self.status})"
+        )

cancan_microstack/services/opsbffsrv/infrastructure/db/operate/admin_user_operate.py

@@ -0,0 +1,59 @@
+"""admin_user_tbl 数据库操作 / DB operations for admin_user_tbl."""
+from typing import Optional
+
+from sqlalchemy import select, update
+
+from linglong_web import Rmanager
+from cancan_microstack.services.opsbffsrv.infrastructure.db.model.admin_user_tbl import AdminUserTbl
+
+
+async def get_admin_user(username: str) -> Optional[AdminUserTbl]:
+    """按用户名查询 admin 用户 / Get admin user by username."""
+    async with Rmanager.pg_session() as session:
+        async with session.begin():
+            stmt = select(AdminUserTbl).where(
+                AdminUserTbl.username == username,
+                AdminUserTbl.flag == 0,
+            )
+            return (await session.execute(stmt)).scalar_one_or_none()
+
+
+async def get_admin_user_by_id(user_id: int) -> Optional[AdminUserTbl]:
+    """按 ID 查询 admin 用户 / Get admin user by ID."""
+    async with Rmanager.pg_session() as session:
+        async with session.begin():
+            stmt = select(AdminUserTbl).where(
+                AdminUserTbl.id == user_id,
+                AdminUserTbl.flag == 0,
+            )
+            return (await session.execute(stmt)).scalar_one_or_none()
+
+
+async def create_admin_user(username: str, password_hash: str) -> AdminUserTbl:
+    """创建 admin 用户 / Create a new admin user."""
+    async with Rmanager.pg_session() as session:
+        async with session.begin():
+            user = AdminUserTbl(username=username, password_hash=password_hash)
+            session.add(user)
+            await session.flush()
+            return user
+
+
+async def update_totp_secret(user_id: int, encrypted_secret: str) -> None:
+    """更新 TOTP secret / Update encrypted TOTP secret for a user."""
+    async with Rmanager.pg_session() as session:
+        async with session.begin():
+            stmt = update(AdminUserTbl).where(
+                AdminUserTbl.id == user_id,
+            ).values(totp_secret_encrypted=encrypted_secret)
+            await session.execute(stmt)
+
+
+async def mark_totp_bound(user_id: int) -> None:
+    """标记 TOTP 已绑定 / Mark TOTP as bound for a user."""
+    async with Rmanager.pg_session() as session:
+        async with session.begin():
+            stmt = update(AdminUserTbl).where(
+                AdminUserTbl.id == user_id,
+            ).values(totp_bound=True)
+            await session.execute(stmt)