deepdata: core product code (backend, frontend, runtime, deepflow module)

runtime/deepdata-python/app/module/shared/chunk_upload_repository.py

@@ -0,0 +1,95 @@
+"""
+Chunk Upload Repository
+分片上传数据访问层
+"""
+from typing import Optional, List
+
+from sqlalchemy import select, update, delete
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from app.db.models.chunk_upload import ChunkUploadPreRequest
+from app.core.logging import get_logger
+
+logger = get_logger(__name__)
+
+
+class ChunkUploadRepository:
+    """分片上传数据访问层"""
+
+    async def find_by_id(
+        self,
+        req_id: str,
+        db: AsyncSession
+    ) -> Optional[ChunkUploadPreRequest]:
+        """根据ID查询"""
+        result = await db.execute(
+            select(ChunkUploadPreRequest).where(ChunkUploadPreRequest.id == req_id)
+        )
+        return result.scalar_one_or_none()
+
+    async def find_by_service_id(
+        self,
+        service_id: str,
+        db: AsyncSession
+    ) -> List[ChunkUploadPreRequest]:
+        """根据服务ID查询"""
+        result = await db.execute(
+            select(ChunkUploadPreRequest).where(
+                ChunkUploadPreRequest.service_id == service_id
+            )
+        )
+        return result.scalars().all()
+
+    async def find_all(self, db: AsyncSession) -> List[ChunkUploadPreRequest]:
+        """查询所有"""
+        result = await db.execute(select(ChunkUploadPreRequest))
+        return result.scalars().all()
+
+    async def insert(
+        self,
+        request: ChunkUploadPreRequest,
+        db: AsyncSession
+    ) -> None:
+        """插入"""
+        db.add(request)
+
+    async def update(
+        self,
+        request: ChunkUploadPreRequest,
+        db: AsyncSession
+    ) -> int:
+        """更新"""
+        from datetime import datetime, timezone
+        result = await db.execute(
+            update(ChunkUploadPreRequest)
+            .where(ChunkUploadPreRequest.id == request.id)
+            .values(
+                uploaded_file_num=request.uploaded_file_num,
+                timeout=request.timeout,
+            )
+        )
+        return result.rowcount
+
+    async def delete_by_id(
+        self,
+        req_id: str,
+        db: AsyncSession
+    ) -> int:
+        """根据ID删除"""
+        result = await db.execute(
+            delete(ChunkUploadPreRequest).where(ChunkUploadPreRequest.id == req_id)
+        )
+        return result.rowcount
+
+    async def delete_by_service_id(
+        self,
+        service_id: str,
+        db: AsyncSession
+    ) -> int:
+        """根据服务ID删除"""
+        result = await db.execute(
+            delete(ChunkUploadPreRequest).where(
+                ChunkUploadPreRequest.service_id == service_id
+            )
+        )
+        return result.rowcount