使用了AI重构项目,并完善了一部分后台问题

This commit is contained in:
2026-08-05 04:47:11 +08:00
parent 146b6b1e6c
commit ac740c24fd
35 changed files with 2876 additions and 980 deletions
+27 -53
View File
@@ -2,75 +2,49 @@ package api
import (
"net/http"
"sync"
"time"
"strings"
"home-vue-go/internal/database"
"github.com/gin-gonic/gin"
)
// VisitRecord 表示一次访问记录(暂存在内存中)
type VisitRecord struct {
Path string
IP string
UserAgent string
Referer string
VisitTime time.Time
}
// 简单的内存存储与读写锁
var (
visitMutex sync.RWMutex
visitRecords []VisitRecord
maxVisits = 1000 // 最多保留的访问记录条数
)
// TrackVisit 记录访问
// TrackVisit stores a local visit only when local analytics is selected.
// Umami mode is tracked by the browser and must not be duplicated here.
func TrackVisit(db *database.Database) gin.HandlerFunc {
return func(c *gin.Context) {
settings, err := db.LoadSiteSettings(c.Request.Context())
if err != nil {
c.JSON(http.StatusInternalServerError, gin.H{"error": "加载统计配置失败"})
return
}
if settings.AnalyticsProvider == "umami" {
c.JSON(http.StatusNoContent, nil)
return
}
var req struct {
Path string `json:"path"`
Referer string `json:"referer"`
}
if err := c.ShouldBindJSON(&req); err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "参数错误"})
return
}
ip := c.ClientIP()
userAgent := c.GetHeader("User-Agent")
// 记录访问到内存
visitMutex.Lock()
visitRecords = append(visitRecords, VisitRecord{
Path: req.Path,
IP: ip,
UserAgent: userAgent,
Referer: req.Referer,
VisitTime: time.Now(),
})
// 限制记录数量
if len(visitRecords) > maxVisits {
visitRecords = visitRecords[len(visitRecords)-maxVisits:]
path := strings.TrimSpace(req.Path)
if path == "" || len(path) > 2048 {
c.JSON(http.StatusBadRequest, gin.H{"error": "访问路径无效"})
return
}
visitMutex.Unlock()
// TODO: 等Ent代码生成后,改用数据库存储
// ctx := c.Request.Context()
// if db.Client.Visit != nil {
// db.Client.Visit.Create()...
// }
c.JSON(http.StatusOK, gin.H{"message": "访问已记录"})
_, err = db.Client.Visit.Create().
SetPath(path).
SetIP(c.ClientIP()).
SetUserAgent(c.GetHeader("User-Agent")).
SetReferer(strings.TrimSpace(req.Referer)).
Save(c.Request.Context())
if err != nil {
c.JSON(http.StatusInternalServerError, gin.H{"error": "访问记录保存失败"})
return
}
c.JSON(http.StatusOK, gin.H{"message": "访问已记录", "analyticsSource": "local"})
}
}
// 获取访问记录(从内存)
func getVisitRecords() []VisitRecord {
visitMutex.RLock()
defer visitMutex.RUnlock()
return visitRecords
}