Initial commit

app/repositories/memory_increment_repository.py

@@ -0,0 +1,110 @@
+from sqlalchemy import func
+from sqlalchemy.orm import Session, aliased
+from typing import List, Optional
+import uuid
+import datetime
+
+from app.models.memory_increment_model import MemoryIncrement
+
+from app.core.logging_config import get_db_logger
+
+# 获取数据库专用日志器
+db_logger = get_db_logger()
+
+
+class MemoryIncrementRepository:
+    def __init__(self, db: Session):
+        self.db = db
+
+    def get_memory_increments_by_workspace_id(self, workspace_id: uuid.UUID, limit: int) -> List[MemoryIncrement]:
+        """根据工作空间ID查询内存增量：通过 MemoryIncrement 关联查询 MemoryIncrement 列表"""
+        try:
+            # 使用窗口函数按日期分区并排序
+            subquery = (
+                self.db.query(
+                    MemoryIncrement,
+                    func.row_number().over(
+                        partition_by=func.date(MemoryIncrement.created_at),  # 按日期分区
+                        order_by=MemoryIncrement.created_at.desc()  # 按时间戳升序排序
+                    ).label('row_num')
+                )
+                .filter(MemoryIncrement.workspace_id == workspace_id)
+                .subquery()
+            )
+
+            memory_increment_alias = aliased(MemoryIncrement, subquery)
+
+            memory_increments = (
+                self.db.query(memory_increment_alias)
+                .filter(subquery.c.row_num == 1)  # 只取每个日期的第一条（最新的）
+                .order_by(memory_increment_alias.created_at.asc())  # 按时间戳降序排序
+                .limit(limit)
+                .all()
+            )
+            db_logger.info(f"成功查询工作空间 {workspace_id} 下的内存增量")
+            return memory_increments
+        except Exception as e:
+            db_logger.error(f"查询工作空间 {workspace_id} 下内存增量时出错: {str(e)}")
+            raise
+
+    def get_latest_memory_increment_by_workspace_id(self, workspace_id: uuid.UUID) -> Optional[MemoryIncrement]:
+        """根据工作空间ID查询最新的内存增量记录"""
+        try:
+            memory_increment = (
+                self.db.query(MemoryIncrement)
+                .filter(MemoryIncrement.workspace_id == workspace_id)
+                .order_by(MemoryIncrement.created_at.desc(), MemoryIncrement.id.desc())
+                .first()
+            )
+            if memory_increment:
+                db_logger.info(f"成功查询工作空间 {workspace_id} 下的最新内存增量")
+            else:
+                db_logger.warning(f"未找到工作空间 {workspace_id} 下的内存增量记录")
+            return memory_increment
+        except Exception as e:
+            db_logger.error(f"查询工作空间 {workspace_id} 下最新内存增量时出错: {str(e)}")
+            raise
+
+    def write_memory_increment(
+        self, 
+        workspace_id: uuid.UUID, 
+        total_num: int
+    ) -> MemoryIncrement:
+        """写入内存增量"""
+        try:
+            memory_increment = MemoryIncrement(
+                workspace_id=workspace_id,
+                total_num=total_num,
+                created_at=datetime.datetime.now(),
+                updated_at=datetime.datetime.now()
+            )
+            self.db.add(memory_increment)
+            self.db.commit()
+            self.db.refresh(memory_increment)
+            db_logger.info(f"成功写入内存增量: workspace_id={workspace_id}, total_num={total_num}")
+            return memory_increment
+        except Exception as e:
+            db_logger.error(f"写入内存增量失败: workspace_id={workspace_id}, total_num={total_num} - {str(e)}")
+            raise
+
+
+def get_memory_increments_by_workspace_id(db: Session, workspace_id: uuid.UUID, limit: int) -> List[MemoryIncrement]:
+    """根据工作空间ID查询内存增量（返回 MemoryIncrement ORM 列表）"""
+    repo = MemoryIncrementRepository(db)
+    memory_increments = repo.get_memory_increments_by_workspace_id(workspace_id, limit)
+    return memory_increments
+
+def write_memory_increment(
+    db: Session, 
+    workspace_id: uuid.UUID, 
+    total_num: int
+) -> MemoryIncrement:
+    """写入内存增量"""
+    repo = MemoryIncrementRepository(db)
+    memory_increment = repo.write_memory_increment(workspace_id, total_num)
+    return memory_increment
+
+def get_latest_memory_increment_by_workspace_id(db: Session, workspace_id: uuid.UUID) -> Optional[MemoryIncrement]:
+    """根据工作空间ID查询最新的内存增量记录"""
+    repo = MemoryIncrementRepository(db)
+    return repo.get_latest_memory_increment_by_workspace_id(workspace_id)