vfdhbvdjh

controllers/main.py

@@ -3798,6 +3798,39 @@ def _get_admin_from_token(authorization: str | None) -> dict:
     return auth["user"]
 
 
+ATTENDANCE_CACHE_TTL_SECONDS = 15
+_attendance_cache: dict[str, tuple[float, dict]] = {}
+
+
+def _attendance_cache_get(key: str) -> dict | None:
+    cached = _attendance_cache.get(key)
+    if not cached:
+        return None
+    expires_at, payload = cached
+    if expires_at <= time.time():
+        _attendance_cache.pop(key, None)
+        return None
+    return payload
+
+
+def _attendance_cache_set(key: str, payload: dict, ttl: int = ATTENDANCE_CACHE_TTL_SECONDS) -> dict:
+    _attendance_cache[key] = (time.time() + ttl, payload)
+    if len(_attendance_cache) > 500:
+        now = time.time()
+        for item_key, (expires_at, _) in list(_attendance_cache.items()):
+            if expires_at <= now:
+                _attendance_cache.pop(item_key, None)
+    return payload
+
+
+def _attendance_cache_clear_person(person_id: str) -> None:
+    marker = f":{person_id}:"
+    suffix = f":{person_id}"
+    for key in list(_attendance_cache.keys()):
+        if marker in key or key.endswith(suffix):
+            _attendance_cache.pop(key, None)
+
+
 def _serialize_notification_rows(rows: list[dict]) -> list[dict]:
     for row in rows:
         for field in ("attendance_time", "created_at", "read_at"):
@@ -3925,6 +3958,9 @@ def save_log_to_db(log_queries: list) -> None:
             log_queries,
         )
         conn.commit()
+        for _, person_id, status, *_ in log_queries:
+            if person_id and status == "success":
+                _attendance_cache_clear_person(person_id)
 
         notification_queries = []
         for log_id, person_id, status, confidence, camera, action in log_queries:
@@ -4667,6 +4703,10 @@ async def employee_me(authorization: str | None = Header(default=None)):
 @app.get("/api/employee/notifications")
 async def employee_notifications(authorization: str | None = Header(default=None)):
     employee = _get_employee_from_token(authorization)
+    cache_key = f"employee_notifications:{employee['person_id']}"
+    cached = _attendance_cache_get(cache_key)
+    if cached:
+        return cached
     conn = cursor = None
     try:
         conn = get_db_connection()
@@ -4678,7 +4718,7 @@ async def employee_notifications(authorization: str | None = Header(default=None
             order="DESC",
         )
         unread = sum(1 for row in rows if row.get("status") == "unread")
-        return {"success": True, "data": rows, "unread": unread}
+        return _attendance_cache_set(cache_key, {"success": True, "data": rows, "unread": unread})
     except Exception as e:
         logger.error(f"[Employee Notifications] {employee.get('person_id')}: {e}", exc_info=True)
         return {
@@ -4707,6 +4747,11 @@ async def employee_attendance_calendar(
     if target_month < 1 or target_month > 12:
         return JSONResponse(status_code=400, content={"success": False, "error": "Thang khong hop le"})
 
+    cache_key = f"employee_attendance:{employee['person_id']}:{target_year}:{target_month}"
+    cached = _attendance_cache_get(cache_key)
+    if cached:
+        return cached
+
     start_date = date(target_year, target_month, 1)
     if target_month == 12:
         end_date = date(target_year + 1, 1, 1)
@@ -4725,7 +4770,7 @@ async def employee_attendance_calendar(
             include_day_key=True,
             order="ASC",
         )
-        return {"success": True, "data": rows, "year": target_year, "month": target_month}
+        return _attendance_cache_set(cache_key, {"success": True, "data": rows, "year": target_year, "month": target_month})
     except Exception as e:
         logger.error(f"[Employee Attendance] {employee.get('person_id')}: {e}", exc_info=True)
         return {
@@ -4775,6 +4820,7 @@ async def mark_employee_notifications_read(
             )
 
         conn.commit()
+        _attendance_cache_clear_person(employee["person_id"])
         return {"success": True, "updated": cursor.rowcount}
     finally:
         if cursor:
@@ -4878,6 +4924,10 @@ async def admin_employee_notifications(
     authorization: str | None = Header(default=None),
 ):
     _get_admin_from_token(authorization)
+    cache_key = f"admin_notifications:{person_id}"
+    cached = _attendance_cache_get(cache_key)
+    if cached:
+        return cached
     conn = cursor = None
     try:
         conn = get_db_connection()
@@ -4889,7 +4939,7 @@ async def admin_employee_notifications(
             order="DESC",
         )
         unread = sum(1 for row in rows if row.get("status") == "unread")
-        return {"success": True, "data": rows, "unread": unread}
+        return _attendance_cache_set(cache_key, {"success": True, "data": rows, "unread": unread})
     except Exception as e:
         logger.error(f"[Admin Employee Notifications] {person_id}: {e}", exc_info=True)
         return {
@@ -4919,6 +4969,11 @@ async def admin_employee_attendance_calendar(
     if target_month < 1 or target_month > 12:
         return JSONResponse(status_code=400, content={"success": False, "error": "Thang khong hop le"})
 
+    cache_key = f"admin_attendance:{person_id}:{target_year}:{target_month}"
+    cached = _attendance_cache_get(cache_key)
+    if cached:
+        return cached
+
     start_date = date(target_year, target_month, 1)
     end_date = date(target_year + 1, 1, 1) if target_month == 12 else date(target_year, target_month + 1, 1)
 
@@ -4934,7 +4989,7 @@ async def admin_employee_attendance_calendar(
             include_day_key=True,
             order="ASC",
         )
-        return {"success": True, "data": rows, "year": target_year, "month": target_month}
+        return _attendance_cache_set(cache_key, {"success": True, "data": rows, "year": target_year, "month": target_month})
     except Exception as e:
         logger.error(f"[Admin Employee Attendance] {person_id}: {e}", exc_info=True)
         return {