refactor(cloud-store): adopt SQLAlchemy adapter

2026-07-12 16:51:03 +08:00
parent ae29477f6a
commit 05a5f0bfa7
6 changed files with 408 additions and 405 deletions
@@ -0,0 +1,50 @@
+from __future__ import annotations
+
+from sqlalchemy import Integer, String, Text
+from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column
+
+
+class Base(DeclarativeBase):
+    pass
+
+
+class HostRow(Base):
+    __tablename__ = "host_registrations"
+
+    host_id: Mapped[str] = mapped_column(String, primary_key=True)
+    address: Mapped[str | None] = mapped_column(String, nullable=True)
+    last_seen_at: Mapped[str] = mapped_column(String, nullable=False)
+
+
+class PooledDeviceRow(Base):
+    __tablename__ = "pooled_devices"
+
+    host_id: Mapped[str] = mapped_column(String, primary_key=True)
+    device_id: Mapped[str] = mapped_column(String, primary_key=True)
+    driver_type: Mapped[str] = mapped_column(String, nullable=False)
+    status: Mapped[str] = mapped_column(String, nullable=False)
+    capability_tags_json: Mapped[str] = mapped_column(Text, nullable=False)
+    synced_at: Mapped[str | None] = mapped_column(String, nullable=True)
+
+
+class ScheduledTaskRow(Base):
+    __tablename__ = "scheduled_tasks"
+
+    id: Mapped[str] = mapped_column(String, primary_key=True)
+    goal: Mapped[str | None] = mapped_column(Text, nullable=True)
+    workflow_definition_id: Mapped[str | None] = mapped_column(String, nullable=True)
+    constraints_json: Mapped[str] = mapped_column(Text, nullable=False)
+    status: Mapped[str] = mapped_column(String, nullable=False)
+    assigned_device_id: Mapped[str | None] = mapped_column(String, nullable=True)
+    assigned_host_id: Mapped[str | None] = mapped_column(String, nullable=True)
+    created_at: Mapped[str] = mapped_column(String, nullable=False)
+
+
+class PluginRow(Base):
+    __tablename__ = "plugins"
+
+    name: Mapped[str] = mapped_column(String, primary_key=True)
+    version: Mapped[str] = mapped_column(String, nullable=False)
+    entry_point_kind: Mapped[str] = mapped_column(String, nullable=False)
+    target: Mapped[str] = mapped_column(String, nullable=False)
+    wired: Mapped[int] = mapped_column(Integer, nullable=False)