增加网站统计信息

Author: liuhang

.gitignore

@@ -0,0 +1 @@
+mysql_data/

cmd/server/internal/handler/article_handler.go

@@ -4,14 +4,19 @@ import (
 	"blog_platform/cmd/server/internal/repository"	
 	"net/http"
 	"github.com/gin-gonic/gin"
+	"time"
 )
 
 type ArticleHandler struct {
 	Repo repository.ArticleRepository
+	StartTime time.Time
 }
 
-func NewArticleHandler(repo repository.ArticleRepository) *ArticleHandler {
-	return &ArticleHandler{Repo: repo}
+func NewArticleHandler(repo repository.ArticleRepository, startTime time.Time) *ArticleHandler {
+	return &ArticleHandler{
+		Repo:      repo,
+		StartTime: startTime,
+	}
 }
 
 // RecordArticleVisit 处理 POST /api/article/visit
@@ -39,4 +44,24 @@ func (h *ArticleHandler) RecordArticleVisit(c *gin.Context) {
 		"path":  req.Path,
 		"views": count,
 	})
+}
+
+// GetSiteInfo 处理 GET /api/site/info
+func (h *ArticleHandler) GetSiteInfo(c *gin.Context) {
+	// 从数据库查统计
+	totalViews, totalArticles, 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,
+		"total_articles": totalArticles,
+		"uptime_seconds": int64(uptime.Seconds()), // 返回秒数，前端处理成 "xx天xx小时" 更灵活
+		"start_time":     h.StartTime.Format(time.RFC3339),
+	})
 }

cmd/server/internal/repository/article_repo.go

@@ -6,9 +6,9 @@ import (
 	"gorm.io/gorm/clause"
 )
 
-// 定义接口，方便未来做单元测试或更换数据库
 type ArticleRepository interface {
 	IncreaseViewCount(path string) (int64, error)
+	GetSiteStats() (int64, int64, error)
 }
 
 type articleRepo struct {
@@ -38,4 +38,24 @@ func (r *articleRepo) IncreaseViewCount(path string) (int64, error) {
 	var stat model.ArticleStat
 	err = r.db.Select("view_count").Where("path = ?", path).First(&stat).Error
 	return stat.ViewCount, err
+}
+
+func (r *articleRepo) GetSiteStats() (int64, int64, error) {
+	var totalViews int64
+	var totalArticles int64
+
+	// 计算总阅读量 (SUM view_count)
+	// COALESCE 防止没有数据时返回 NULL 导致报错
+	err := r.db.Model(&model.ArticleStat{}).Select("COALESCE(SUM(view_count), 0)").Scan(&totalViews).Error
+	if err != nil {
+		return 0, 0, err
+	}
+
+	// 计算文章总数 (COUNT)
+	err = r.db.Model(&model.ArticleStat{}).Count(&totalArticles).Error
+	if err != nil {
+		return 0, 0, err
+	}
+
+	return totalViews, totalArticles, nil
 }

cmd/server/main.go

@@ -6,6 +6,7 @@ import (
 	"blog_platform/cmd/server/internal/repository"
 	"log"
 	"os"
+	"time"
 
 	"github.com/gin-contrib/cors"
 	"github.com/gin-gonic/gin"
@@ -23,11 +24,12 @@ func main() {
 	if err != nil {
 		log.Fatal("无法连接数据库: ", err)
 	}
+	siteFoundingDate := time.Date(2025, 12, 10, 0, 0, 0, 0, time.Local)
 
 	db.AutoMigrate(&model.ArticleStat{})
 
 	articleRepo := repository.NewArticleRepo(db)
-	articleHandler := handler.NewArticleHandler(articleRepo)
+	articleHandler := handler.NewArticleHandler(articleRepo, siteFoundingDate)
 
 	r := gin.Default()
 
@@ -42,6 +44,7 @@ func main() {
 	api := r.Group("/api")
 	{
 		api.POST("/article/visit", articleHandler.RecordArticleVisit)
+		api.GET("/site/info", articleHandler.GetSiteInfo)
 	}
 
 	log.Println("服务启动在 :8080")