34 lines
819 B
Python
34 lines
819 B
Python
from fastapi import FastAPI
|
|
from pydantic import BaseModel
|
|
from redis import Redis
|
|
from datetime import datetime
|
|
from pymongo import MongoClient
|
|
import json
|
|
|
|
app = FastAPI()
|
|
|
|
redis_client = Redis(host="redis", port=6379, decode_responses=True)
|
|
mongo_client = MongoClient("mongodb://mongo:27017")
|
|
mongo_db = mongo_client["logs"]
|
|
mongo_col = mongo_db["actions"]
|
|
|
|
class Action(BaseModel):
|
|
user: str
|
|
type: str
|
|
|
|
@app.post("/action")
|
|
async def log_action(action: Action):
|
|
entry = {
|
|
"user": action.user,
|
|
"type": action.type,
|
|
"timestamp": datetime.utcnow().isoformat()
|
|
}
|
|
redis_client.rpush("user_actions", json.dumps(entry))
|
|
return {"status": "saved to redis"}
|
|
|
|
@app.get("/actions")
|
|
async def get_actions():
|
|
actions = list(mongo_col.find({}, {"_id": 0}))
|
|
return actions
|
|
|