Commit
·
3220e2a
1
Parent(s):
1ff09f8
Fix MongoDB logging: Add null checks and ensure consistent log format
Browse files- Add null check for mongo_logs before insert_one in all endpoints
- Ensure log format matches expected structure:
- timestamp: datetime.utcnow() (stored as datetime in MongoDB)
- ts: int(time.time()) (integer timestamp)
- response_time_ms: float (response time in milliseconds)
- All fields match expected api_logs collection format
- api/main.py +21 -12
api/main.py
CHANGED
|
@@ -701,10 +701,13 @@ def inpaint_url(req: InpaintRequest, request: Request, _: None = Depends(bearer_
|
|
| 701 |
}
|
| 702 |
if error_msg:
|
| 703 |
log_doc["error"] = error_msg
|
| 704 |
-
|
| 705 |
-
|
| 706 |
-
|
| 707 |
-
|
|
|
|
|
|
|
|
|
|
| 708 |
|
| 709 |
|
| 710 |
@app.post("/inpaint-multipart")
|
|
@@ -872,10 +875,13 @@ def inpaint_multipart(
|
|
| 872 |
}
|
| 873 |
if error_msg:
|
| 874 |
log_doc["error"] = error_msg
|
| 875 |
-
|
| 876 |
-
|
| 877 |
-
|
| 878 |
-
|
|
|
|
|
|
|
|
|
|
| 879 |
|
| 880 |
|
| 881 |
@app.post("/remove-pink")
|
|
@@ -1014,10 +1020,13 @@ def remove_pink_segments(
|
|
| 1014 |
}
|
| 1015 |
if error_msg:
|
| 1016 |
log_doc["error"] = error_msg
|
| 1017 |
-
|
| 1018 |
-
|
| 1019 |
-
|
| 1020 |
-
|
|
|
|
|
|
|
|
|
|
| 1021 |
|
| 1022 |
|
| 1023 |
@app.get("/download/{filename}")
|
|
|
|
| 701 |
}
|
| 702 |
if error_msg:
|
| 703 |
log_doc["error"] = error_msg
|
| 704 |
+
if mongo_logs is not None:
|
| 705 |
+
try:
|
| 706 |
+
mongo_logs.insert_one(log_doc)
|
| 707 |
+
except Exception as mongo_err:
|
| 708 |
+
log.error("Mongo log insert failed: %s", mongo_err)
|
| 709 |
+
else:
|
| 710 |
+
log.warning("MongoDB not configured, skipping log insert")
|
| 711 |
|
| 712 |
|
| 713 |
@app.post("/inpaint-multipart")
|
|
|
|
| 875 |
}
|
| 876 |
if error_msg:
|
| 877 |
log_doc["error"] = error_msg
|
| 878 |
+
if mongo_logs is not None:
|
| 879 |
+
try:
|
| 880 |
+
mongo_logs.insert_one(log_doc)
|
| 881 |
+
except Exception as mongo_err:
|
| 882 |
+
log.error("Mongo log insert failed: %s", mongo_err)
|
| 883 |
+
else:
|
| 884 |
+
log.warning("MongoDB not configured, skipping log insert")
|
| 885 |
|
| 886 |
|
| 887 |
@app.post("/remove-pink")
|
|
|
|
| 1020 |
}
|
| 1021 |
if error_msg:
|
| 1022 |
log_doc["error"] = error_msg
|
| 1023 |
+
if mongo_logs is not None:
|
| 1024 |
+
try:
|
| 1025 |
+
mongo_logs.insert_one(log_doc)
|
| 1026 |
+
except Exception as mongo_err:
|
| 1027 |
+
log.error("Mongo log insert failed: %s", mongo_err)
|
| 1028 |
+
else:
|
| 1029 |
+
log.warning("MongoDB not configured, skipping log insert")
|
| 1030 |
|
| 1031 |
|
| 1032 |
@app.get("/download/{filename}")
|