Add health endpoint

app/db.py

@@ -78,11 +78,11 @@ def get_visit_stats(
  statement = select(VisitStats)
  if start:
  # Ignore type because Visit.called_at can be None for default value
- statement.where(VisitStats.start >= start) # type: ignore
+ statement = statement.where(VisitStats.start >= start) # type: ignore
  if end:
- statement.where(VisitStats.end <= end) # type: ignore
+ statement = statement.where(VisitStats.end <= end) # type: ignore
  if limit:
- statement.limit(limit)
+ statement = statement.limit(limit)
  statement.order_by(VisitStats.start.desc()) # type: ignore
  return session.exec(statement).all()
 
@@ -129,3 +129,4 @@ def insert_download_stats(df: pd.DataFrame) -> pd.DataFrame:
  session.add(new_entry)
  session.commit()
  return df
+

app/main.py

@@ -1,5 +1,4 @@
 import logging
-from logzio.handler import LogzioHandler
 
 from typing import List, Dict, Optional
 
@@ -126,6 +125,21 @@ async def version(
  return models.VersionResponse(latest_release=app.latest_release)
 
 
+@app.get("/health")
+async def health():
+ """
+ Health check endpoint. Checks if the visits table contains records 
+ in the past 10 minutes.
+ """
+ try:
+ visits = db.get_visit_stats(start=datetime.datetime.now() - datetime.timedelta(minutes=10))
+ except Exception as e:
+ raise HTTPException(status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, detail=str(e))
+ if not visits:
+ raise HTTPException(status_code=status.HTTP_503_SERVICE_UNAVAILABLE, detail="No recent visits found")
+ return PlainTextResponse(content=str(len(visits)))
+
+
 def _log_visit(
  timestamp: str,
  version_multiqc: str = "",