统计网站总访问量

Author: liuhang

cmd/server/internal/handler/article_handler.go

@@ -39,28 +39,38 @@ func (h *ArticleHandler) RecordArticleVisit(c *gin.Context) {
 		return
 	}
 
-	// 返回结果
 	c.JSON(http.StatusOK, gin.H{
 		"path":  req.Path,
 		"views": count,
 	})
 }
 
+func (h *ArticleHandler) RecordSiteVisit(c *gin.Context) {
+	// 调用 Repo 增加全站计数
+	err := h.Repo.IncreaseSiteVisit()
+	if err != nil {
+		c.JSON(http.StatusOK, gin.H{"status": "ok"})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{"status": "ok"})
+
+}
+
 // GetSiteInfo 处理 GET /api/site/info
 func (h *ArticleHandler) GetSiteInfo(c *gin.Context) {
 	// 从数据库查统计
-	totalViews, err := h.Repo.GetSiteStats()
+	totalViews, siteTotalVisits, err := h.Repo.GetSiteStats()
 	if err != nil {
 		c.JSON(http.StatusInternalServerError, gin.H{"error": "数据库查询失败"})
 		return
 	}
 
-	// 计算运行时间 (当前时间 - 启动时间)
 	uptime := time.Since(h.StartTime)
 
 	c.JSON(http.StatusOK, gin.H{
 		"total_views":    totalViews,
-		"uptime_seconds": int64(uptime.Seconds()), // 返回秒数，前端处理成 "xx天xx小时" 更灵活
+		"site_total_visits": siteTotalVisits,
+		"uptime_seconds": int64(uptime.Seconds()), 
 		"start_time":     h.StartTime.Format(time.RFC3339),
 	})
 }

cmd/server/internal/model/article.go

@@ -4,9 +4,13 @@ import "time"
 
 // ArticleStat 对应数据库中的 article_stats 表
 type ArticleStat struct {
-	// Path 是文章的 URL 路径，作为主键 (例如 /p/hello-world/)
 	Path      string    `gorm:"primaryKey;type:varchar(191)"` 
 	ViewCount int64     `gorm:"default:0"` // 浏览量
 	CreatedAt time.Time // 创建时间 (自动维护)
 	UpdatedAt time.Time // 更新时间 (自动维护)
+}
+
+type SiteStat struct {
+	ID        uint  `gorm:"primaryKey"` 
+	ViewCount int64 `gorm:"default:0"`
 }

cmd/server/internal/repository/article_repo.go

@@ -8,7 +8,8 @@ import (
 
 type ArticleRepository interface {
 	IncreaseViewCount(path string) (int64, error)
-	GetSiteStats() (int64, error)
+	IncreaseSiteVisit() error
+	GetSiteStats() (int64, int64, error)
 }
 
 type articleRepo struct {
@@ -40,15 +41,39 @@ func (r *articleRepo) IncreaseViewCount(path string) (int64, error) {
 	return stat.ViewCount, err
 }
 
-func (r *articleRepo) GetSiteStats() (int64, error) {
+func (r *articleRepo) IncreaseSiteVisit() error {
+	return r.db.Clauses(clause.OnConflict{
+		Columns:   []clause.Column{{Name: "id"}}, // 冲突列是 ID
+		DoUpdates: clause.Assignments(map[string]interface{}{
+			"view_count": gorm.Expr("view_count + 1"),
+		}),
+	}).Create(&model.SiteStat{ID: 1, ViewCount: 1}).Error
+}
+
+
+func (r *articleRepo) GetSiteStats() (int64, int64, error) {
 	var totalViews int64
+	var siteTotalVisits int64
 
 	// 计算总阅读量 (SUM view_count)
 	// COALESCE 防止没有数据时返回 NULL 导致报错
-	err := r.db.Model(&model.ArticleStat{}).Select("COALESCE(SUM(view_count), 0)").Scan(&totalViews).Error
+	err := r.db.Model(&model.ArticleStat{}).
+	Select("COALESCE(SUM(view_count), 0)").
+	Scan(&totalViews).Error
+
 	if err != nil {
-		return 0, err
+		return 0, 0, err
+	}
+
+	// 计算全站访问量 (SELECT view_count FROM site_stats WHERE id = 1)
+	var stat model.SiteStat
+	err = r.db.Model(&model.SiteStat{}).
+		Where("id = ?", 1).
+		First(&stat).Error
+
+	if err == nil {
+		siteTotalVisits = stat.ViewCount
 	}
 
-	return totalViews, nil
+	return totalViews, siteTotalVisits, nil
 }

cmd/server/main.go

@@ -17,7 +17,7 @@ import (
 func main() {
 	dsn := os.Getenv("DSN")
 	if dsn == "" {
-		dsn = "root:GB_UXa>cX*h!K2@tcp(127.0.0.1:3306)/blog_db?charset=utf8mb4&parseTime=True&loc=Local"
+		log.Fatal("DSN 环境变量未设置")
 	}
 
 	db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
@@ -26,7 +26,7 @@ func main() {
 	}
 	siteFoundingDate := time.Date(2025, 12, 10, 0, 0, 0, 0, time.Local)
 
-	db.AutoMigrate(&model.ArticleStat{})
+	db.AutoMigrate(&model.ArticleStat{}, &model.SiteStat{})
 
 	articleRepo := repository.NewArticleRepo(db)
 	articleHandler := handler.NewArticleHandler(articleRepo, siteFoundingDate)
@@ -35,7 +35,7 @@ func main() {
 
 	//  配置 CORS (允许跨域)
 	r.Use(cors.New(cors.Config{
-		AllowOrigins:     []string{"https://hangops.top", "http://localhost:1313"}, // 允许的域名
+		AllowOrigins:     []string{"https://hangops.top", "http://localhost:1313"}, 
 		AllowMethods:     []string{"POST", "GET", "OPTIONS"},
 		AllowHeaders:     []string{"Content-Type"},
 		AllowCredentials: true,
@@ -44,6 +44,7 @@ func main() {
 	api := r.Group("/api")
 	{
 		api.POST("/article/visit", articleHandler.RecordArticleVisit)
+		api.POST("/site/visit", articleHandler.RecordSiteVisit)
 		api.GET("/site/info", articleHandler.GetSiteInfo)
 	}