Merge pull request #25 from pgulb/8-implement-mongo-connectivity

8 implement mongo connectivity

Author: pgulb

.github/workflows/deploy-api.yml

@@ -31,4 +31,4 @@ jobs:
         run: |
           echo ${{ secrets.FROG_KEY }} | base64 -d > frog &&
           chmod 600 frog &&
-          ssh -o StrictHostKeyChecking=no -i frog ${{ secrets.FROG_ADDRESS }} -p${{ secrets.FROG_PORT }} "docker run -d -p 30149:6789 --name flush-log-api --restart=unless-stopped ghcr.io/pgulb/flush-log:api"
+          ssh -o StrictHostKeyChecking=no -i frog ${{ secrets.FROG_ADDRESS }} -p${{ secrets.FROG_PORT }} "docker run -d -p 30149:6789 -e MONGO_URL=mock --name flush-log-api --restart=unless-stopped ghcr.io/pgulb/flush-log:api"

api/__init__.py

@@ -0,0 +1,25 @@
+from pymongo import MongoClient
+from pymongo.errors import ConnectionFailure
+
+
+class MockClient:
+    def __init__(self, not_avalaible: str):
+        self.admin = self
+        if not_avalaible == "true":
+            self.command = self.__bad_command
+
+    def __bad_command(self, command: str):
+        raise ConnectionFailure(f"Mock server not available, used command: {command}")
+
+    def command(self, command: str) -> str:
+        if command == "ping":
+            return "OK"
+        raise NotImplementedError
+
+
+def create_mongo_client(url: str) -> MongoClient:
+    return MongoClient(url)
+
+
+def create_mock_client(not_avalaible: str) -> MockClient:
+    return MockClient(not_avalaible)

api/db.py

@@ -1,7 +1,10 @@
+import logging
+import os
 from random import randint
 
 import fastapi
-from fastapi import Depends, HTTPException, status
+from db import create_mock_client, create_mongo_client
+from fastapi import Depends, HTTPException, Response, status
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.security import HTTPBasic, HTTPBasicCredentials
 
@@ -18,6 +21,20 @@
 )
 security = HTTPBasic()
 
+logging.basicConfig(level=logging.INFO, format="%(levelname)s %(asctime)s %(message)s")
+
+mongo_setting = os.getenv("MONGO_URL")
+if mongo_setting is None:
+    raise ValueError("MONGO_URL not set")
+if mongo_setting == "mock":
+    client = create_mock_client(os.getenv("MOCK_NOT_AVAILABLE"))
+    logging.info("Using mock client")
+else:
+    client = create_mongo_client(mongo_setting)
+    logging.info("Using mongo client")
+    if "/?" in mongo_setting:
+        logging.info(f"client options: {mongo_setting.split("/?")[1]}")
+
 
 def check_creds(credentials: HTTPBasicCredentials):
     if not (credentials.username == "admin" and credentials.password == "admin"):
@@ -32,3 +49,17 @@ def check_creds(credentials: HTTPBasicCredentials):
 def root(credentials: HTTPBasicCredentials = Depends(security)):
     check_creds(credentials)
     return f"Random string from api {randint(0, 10000)}"
+
+
+@app.get("/healthz", status_code=status.HTTP_200_OK)
+def healthz():
+    return "OK"
+
+
+@app.get("/readyz", status_code=status.HTTP_200_OK)
+def readyz():
+    try:
+        client.admin.command("ping")
+    except Exception:
+        return Response("NOT OK", status_code=status.HTTP_503_SERVICE_UNAVAILABLE)
+    return "OK"

api/main.py

@@ -2,11 +2,13 @@ annotated-types==0.7.0
 anyio==4.6.2.post1
 click==8.1.7
 colorama==0.4.6
+dnspython==2.7.0
 fastapi==0.115.2
 h11==0.14.0
 idna==3.10
 pydantic==2.9.2
 pydantic-core==2.23.4
+pymongo==4.10.1
 sniffio==1.3.1
 starlette==0.40.0
 typing-extensions==4.12.2

api/requirements.txt

@@ -17,5 +17,11 @@ services:
     ports:
       - 6789:6789
     restart: unless-stopped
+    environment:
+      - MONGO_URL=mongodb://mongo:27017/?connectTimeoutMS=3000&timeoutMS=4000&socketTimeoutMS=3000&serverSelectionTimeoutMS=4000
+      - MOCK_NOT_AVAILABLE=false
     volumes:
       - ./api:/src/
+  mongo:
+    image: mongo:8.0.1-noble
+    restart: unless-stopped