admin
980b91f41f
- 新增 subnet-aware 负载策略: 根据客户端IP匹配最近/16子网节点 - 节点注册端点 (POST /api/nodes/register): 边缘节点自动注册 - 节点心跳端点 (POST /api/nodes/heartbeat): 上报CPU/RAM/带宽/连接数 - 健康汇总 (GET /api/nodes/health): 服务端汇总所有节点状态 - 配置节点新增subnet/is_origin字段 - 前端NodeManager: 实时健康状态/CPU/RAM进度条/带宽/连接数 - X-Forwarded-For/X-Real-IP透传支持客户端IP
140 lines
4.0 KiB
Python
140 lines
4.0 KiB
Python
from fastapi import APIRouter, HTTPException, Depends
|
|
from pydantic import BaseModel
|
|
from typing import Optional
|
|
from config import settings
|
|
from routes.auth import require_auth
|
|
from services.load_balancer import lb
|
|
from services.multicast_detector import multicast_detector
|
|
|
|
router = APIRouter(prefix="/api/nodes", tags=["nodes"])
|
|
|
|
|
|
class NodeUpdate(BaseModel):
|
|
weight: int
|
|
enabled: bool | None = None
|
|
subnet: str | None = None
|
|
|
|
|
|
class StrategyUpdate(BaseModel):
|
|
strategy: str
|
|
multicast_enabled: bool | None = None
|
|
|
|
|
|
class NodeHeartbeat(BaseModel):
|
|
node_id: str
|
|
cpu_percent: float | None = None
|
|
memory_percent: float | None = None
|
|
bandwidth_mbps: float | None = None
|
|
active_connections: int | None = None
|
|
hls_readers: int | None = None
|
|
|
|
|
|
class NodeRegister(BaseModel):
|
|
node_id: str
|
|
name: str
|
|
host: str
|
|
hls_port: int
|
|
api_port: int
|
|
weight: int = 100
|
|
subnet: str = "0.0.0.0/0"
|
|
|
|
|
|
@router.get("")
|
|
async def list_nodes(_: bool = Depends(require_auth)):
|
|
nodes = []
|
|
for n in settings.edge_nodes:
|
|
nid = n["id"]
|
|
nodes.append({
|
|
"id": nid,
|
|
"name": n["name"],
|
|
"host": n["host"],
|
|
"hls_port": n["hls_port"],
|
|
"api_port": n["api_port"],
|
|
"weight": n["weight"],
|
|
"enabled": n.get("enabled", True),
|
|
"subnet": n.get("subnet", "0.0.0.0/0"),
|
|
"is_origin": n.get("is_origin", False),
|
|
"online": lb.is_online(nid),
|
|
"metrics": lb._node_health.get(nid),
|
|
})
|
|
return nodes
|
|
|
|
|
|
@router.get("/strategy")
|
|
async def get_strategy(_: bool = Depends(require_auth)):
|
|
return {
|
|
"strategy": settings.load_balance_strategy,
|
|
"multicast_enabled": settings.multicast_enabled,
|
|
"multicast_available": multicast_detector.is_available(),
|
|
}
|
|
|
|
|
|
@router.put("/strategy")
|
|
async def update_strategy(body: StrategyUpdate, _: bool = Depends(require_auth)):
|
|
if body.strategy in ("round-robin", "weighted", "least-conn", "subnet"):
|
|
settings.load_balance_strategy = body.strategy
|
|
if body.multicast_enabled is not None:
|
|
settings.multicast_enabled = body.multicast_enabled
|
|
return {"ok": True}
|
|
|
|
|
|
@router.put("/{node_id}")
|
|
async def update_node(node_id: str, body: NodeUpdate, _: bool = Depends(require_auth)):
|
|
found = False
|
|
for n in settings.edge_nodes:
|
|
if n["id"] == node_id:
|
|
n["weight"] = body.weight
|
|
if body.enabled is not None:
|
|
n["enabled"] = body.enabled
|
|
if body.subnet is not None:
|
|
n["subnet"] = body.subnet
|
|
found = True
|
|
break
|
|
if not found:
|
|
raise HTTPException(status_code=404, detail="Node not found")
|
|
return {"ok": True}
|
|
|
|
|
|
@router.post("/register")
|
|
async def register_node(body: NodeRegister):
|
|
for n in settings.edge_nodes:
|
|
if n["id"] == body.node_id:
|
|
n.update({
|
|
"name": body.name,
|
|
"host": body.host,
|
|
"hls_port": body.hls_port,
|
|
"api_port": body.api_port,
|
|
"weight": body.weight,
|
|
"subnet": body.subnet,
|
|
})
|
|
return {"ok": True, "action": "updated"}
|
|
settings.edge_nodes.append({
|
|
"id": body.node_id,
|
|
"name": body.name,
|
|
"host": body.host,
|
|
"hls_port": body.hls_port,
|
|
"api_port": body.api_port,
|
|
"weight": body.weight,
|
|
"enabled": True,
|
|
"subnet": body.subnet,
|
|
"is_origin": False,
|
|
})
|
|
return {"ok": True, "action": "registered"}
|
|
|
|
|
|
@router.post("/heartbeat")
|
|
async def node_heartbeat(body: NodeHeartbeat):
|
|
lb.report_health(body.node_id, {
|
|
"cpu_percent": body.cpu_percent,
|
|
"memory_percent": body.memory_percent,
|
|
"bandwidth_mbps": body.bandwidth_mbps,
|
|
"active_connections": body.active_connections,
|
|
"hls_readers": body.hls_readers,
|
|
})
|
|
return {"ok": True}
|
|
|
|
|
|
@router.get("/health")
|
|
async def get_all_health():
|
|
return lb.get_health_summary()
|