kevin
50e1661c1b
流量统计从全站按日聚合升级为 per-user 维度，并引入 60 秒定时 落库机制，服务崩溃最多丢失 60 秒数据。 后端: - model/vpn.go: 新增 UserTrafficStat 模型（user_id + date 联合唯一索引） - db/db.go: AutoMigrate 注册新模型 - vpn/tunnel.go: tunnelConn 增加 flushedRx/flushedTx 快照字段 + flushDelta() 增量计算；recordTraffic 改为 per-user 双写 （user_traffic_stats + traffic_stats）；连接断开 defer 改增量落库 - vpn/service.go: 新增 flushDone 字段 + trafficFlusher（60s 定时 落库）+ flushAllTraffic；Stop() 先停 flusher 再最终 flush 全部 在线连接增量；TotalLiveTraffic 改返回未落库增量避免与 DB 重复； 新增 UserLiveTraffic(userID)；ClientInfo 增加 RxBytes/TxBytes - handler/traffic.go: 新建 5 个 handler（admin 3 个 + user 2 个） - router.go: 注册 5 条新路由 新增 API: - GET /api/admin/traffic/today 所有用户今日流量排行 - GET /api/admin/traffic/history?days=N 全站近 N 天流量历史 - GET /api/admin/traffic/users/:id?days=N 指定用户近 N 天流量 - GET /api/me/traffic/today 自己的今日流量 - GET /api/me/traffic?days=N 自己的近 N 天流量 前端: - 安装 chart.js + vue-chartjs - 新建 TrafficChart.vue 可复用柱状图组件（上行/下行双柱） - AdminView.vue: 在线客户端表格增加 RX/TX 列 + 全站 7 天流量 图表 + 用户今日流量排行表 - ProfileView.vue: 新增流量统计卡片（今日上行/下行/合计）+ 7 天 流量柱状图 - zh.ts/en.ts: 新增 traffic 相关国际化
77 lines
2.4 KiB
Go
77 lines
2.4 KiB
Go
package router
|
|
|
|
import (
|
|
"net/http"
|
|
"strings"
|
|
|
|
"lmvpn/internal/handler"
|
|
"lmvpn/internal/middleware"
|
|
"lmvpn/internal/vpn"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func Setup(r *gin.Engine) {
|
|
r.GET("/ws", vpn.HandleWS)
|
|
|
|
bodyLimit := middleware.BodyLimit(1 << 20) // 1 MiB
|
|
|
|
r.POST("/api/login", bodyLimit, middleware.LoginRateLimit(), handler.Login)
|
|
r.GET("/api/version", handler.GetVersion)
|
|
|
|
auth := r.Group("/api")
|
|
auth.Use(middleware.AuthMiddleware(), bodyLimit)
|
|
{
|
|
auth.GET("/me", handler.Me)
|
|
auth.PUT("/me/password", handler.ChangePassword)
|
|
auth.GET("/me/sessions", handler.ListMySessions)
|
|
auth.DELETE("/me/sessions/:sessionId", handler.RevokeMySession)
|
|
auth.GET("/me/vpn/connections", handler.GetMyVpnConnections)
|
|
auth.GET("/me/traffic/today", handler.GetMyTrafficToday)
|
|
auth.GET("/me/traffic", handler.GetMyTrafficHistory)
|
|
}
|
|
|
|
admin := r.Group("/api/admin")
|
|
admin.Use(middleware.AuthMiddleware(), middleware.AdminMiddleware(), bodyLimit)
|
|
{
|
|
admin.GET("/stats", handler.GetAdminStats)
|
|
admin.GET("/users/count", handler.GetUserCount)
|
|
admin.GET("/users", handler.ListUsers)
|
|
admin.POST("/users", handler.CreateUser)
|
|
admin.PUT("/users/:id", handler.UpdateUser)
|
|
admin.DELETE("/users/:id", handler.DeleteUser)
|
|
admin.DELETE("/users/:id/sessions", handler.AdminRevokeUserSessions)
|
|
|
|
admin.GET("/vpn/settings", handler.GetVpnSettings)
|
|
admin.PUT("/vpn/settings", handler.UpdateVpnSettings)
|
|
admin.GET("/vpn/status", handler.GetVpnStatus)
|
|
admin.GET("/vpn/diag", handler.GetVpnDiag)
|
|
admin.GET("/vpn/reservations", handler.ListVpnReservations)
|
|
admin.POST("/vpn/reservations", handler.CreateVpnReservation)
|
|
admin.DELETE("/vpn/reservations/:id", handler.DeleteVpnReservation)
|
|
admin.DELETE("/vpn/clients/:id", handler.KickUserClient)
|
|
|
|
admin.GET("/traffic/today", handler.GetAdminTrafficToday)
|
|
admin.GET("/traffic/history", handler.GetAdminTrafficHistory)
|
|
admin.GET("/traffic/users/:id", handler.GetAdminUserTraffic)
|
|
}
|
|
|
|
distDir := http.Dir("./dist")
|
|
fs := http.FileServer(distDir)
|
|
r.NoRoute(func(c *gin.Context) {
|
|
path := c.Request.URL.Path
|
|
if strings.HasPrefix(path, "/api") || strings.HasPrefix(path, "/ws") {
|
|
c.JSON(http.StatusNotFound, gin.H{"error": "not found"})
|
|
return
|
|
}
|
|
f, err := distDir.Open(path)
|
|
if err != nil {
|
|
c.Header("Content-Type", "text/html")
|
|
c.File("./dist/index.html")
|
|
return
|
|
}
|
|
f.Close()
|
|
fs.ServeHTTP(c.Writer, c.Request)
|
|
})
|
|
}
|