use mysql

routers/router.go

@@ -9,6 +9,7 @@ import (
 	"net/http"
 
 	"github.com/gin-gonic/gin"
+	"gorm.io/gorm"
 )
 
 func RegisterRoutes(r *gin.Engine) {
@@ -21,7 +22,20 @@ func RegisterRoutes(r *gin.Engine) {
 		var pager serializers.Pager
 		pager.InitPager(c)
 		offset := (pager.Page - 1) * pager.PageSize
-		models.DB.Select("*").Offset(offset).Limit(pager.PageSize).Find(&items, "type = ?", "post")
+		if dbInterface, ok := c.Get("DB"); ok {
+			if db, ok := dbInterface.(*gorm.DB); ok {
+				db.Select("*").Offset(offset).Limit(pager.PageSize).Find(&items, "type = ?", "post")
+			} else {
+				log.Println("无法将 DB 转换为 *gorm.DB 类型")
+				c.JSON(http.StatusInternalServerError, gin.H{"error": "内部服务器错误"})
+				return
+			}
+		} else {
+			log.Println("未找到键 'DB' 的上下文值")
+			c.JSON(http.StatusInternalServerError, gin.H{"error": "内部服务器错误"})
+			return
+		}
+
 		c.HTML(http.StatusOK, "index.tmpl", gin.H{
 			"Items": items,
 			"Pager": pager,