pip install fastapi uvicorn

python main.py

uvicorn main:app --reload --host 0.0.0.0 --port 8001

from fastapi import FastAPI, Request, Depends, HTTPException
from collections import defaultdict
from datetime import datetime, timedelta
import asyncio
import json
import os

app = FastAPI()

# 存儲(chǔ)客戶端的心跳數(shù)據(jù)
clients_last_heartbeat = defaultdict(datetime)

# 每個(gè)IP請(qǐng)求時(shí)間間隔限制為1秒
last_request_time = defaultdict(datetime)

# 在線客戶端統(tǒng)計(jì)
online_clients = set()

# 心跳超時(shí)時(shí)間設(shè)置為1分鐘
HEARTBEAT_TIMEOUT = timedelta(minutes=10)

# 用戶數(shù)據(jù)文件路徑
USER_DATA_FILE = "users_data.json"

# 加載用戶數(shù)據(jù)
def load_user_data():
    if os.path.exists(USER_DATA_FILE):
        with open(USER_DATA_FILE, "r") as f:
            return json.load(f)
    return {}

# 保存用戶數(shù)據(jù)
def save_user_data(data):
    with open(USER_DATA_FILE, "w") as f:
        json.dump(data, f)

# 初始化用戶數(shù)據(jù)
all_users = load_user_data()

@app.on_event("startup")
async def startup_event():
    # 啟動(dòng)后臺(tái)任務(wù)，每1分鐘檢查一次在線設(shè)備
    asyncio.create_task(remove_offline_clients())

async def remove_offline_clients():
    """
    定時(shí)任務(wù)：移除超過(guò)心跳超時(shí)時(shí)間未發(fā)送心跳的客戶端
    """
    while True:
        await asyncio.sleep(HEARTBEAT_TIMEOUT.total_seconds())
        now = datetime.utcnow()
        # 找出超過(guò)超時(shí)時(shí)間未發(fā)送心跳的設(shè)備，并將其從在線列表中移除
        offline_clients = {ip for ip, last_heartbeat in clients_last_heartbeat.items()
                           if now - last_heartbeat > HEARTBEAT_TIMEOUT}
        
        # 從在線設(shè)備中移除離線的客戶端
        for client in offline_clients:
            online_clients.discard(client)
            del clients_last_heartbeat[client]  # 刪除心跳記錄

        print(f"清除離線客戶端, 當(dāng)前在線客戶端數(shù)量: {len(online_clients)}")

# 請(qǐng)求頻率限制，1秒內(nèi)只能請(qǐng)求一次
def request_limit(request: Request):
    client_ip = request.client.host
    now = datetime.utcnow()
    
    if client_ip in last_request_time and (now - last_request_time[client_ip]).total_seconds() < 1:
        raise HTTPException(status_code=429, detail="Too Many Requests")
    
    last_request_time[client_ip] = now

@app.post("/heartbeat")
async def receive_heartbeat(request: Request, limit: None = Depends(request_limit)):
    """
    接受客戶端的心跳包
    """
    client_ip = request.client.host
    now = datetime.utcnow()
    # 更新心跳時(shí)間并將客戶端標(biāo)記為在線
    clients_last_heartbeat[client_ip] = now
    online_clients.add(client_ip)
    
    # 更新所有用戶數(shù)據(jù)并保存到文件
    all_users[client_ip] = now.isoformat()
    save_user_data(all_users)
    
    return {"message": "Heartbeat received", "ip": client_ip}

@app.get("/online_clients")
async def get_online_clients(request: Request, limit: None = Depends(request_limit)):
    """
    獲取當(dāng)前在線客戶端數(shù)量
    """
    return {"online_clients_count": len(online_clients)}

@app.post("/online_clients")
async def get_online_clients2(request: Request, limit: None = Depends(request_limit)):
    """
    獲取當(dāng)前在線客戶端數(shù)量
    """
    return {"online_clients_count": len(online_clients)}

@app.get("/total_users")
async def get_total_users(days: int = 7, request: Request = None, limit: None = Depends(request_limit)):
    """
    獲取最近n天活躍的用戶數(shù)
    """
    now = datetime.utcnow()
    cutoff_date = now - timedelta(days=days)
    
    # 篩選最近n天活躍的用戶
    recent_users_count = sum(
        1 for last_seen in all_users.values()
        if datetime.fromisoformat(last_seen) >= cutoff_date
    )
    
    return {"recent_users_count": recent_users_count, "days": days}

if __name__ == '__main__':
    import uvicorn
    uvicorn.run(app, host="0.0.0.0", port=8001)