go_blog/routers/router.go

package routers // Add the package declaration at the top

import (
	"fmt"
	"go_blog/controllers"
	"go_blog/models"
	"go_blog/serializers"
	"log"
	"net/http"

	"github.com/gin-gonic/gin"
	"gorm.io/gorm"
)

func RegisterRoutes(r *gin.Engine) {

	// 注册WebSocket路由
	r.GET("/events", esSSE)

	r.GET("/page", func(c *gin.Context) {
		var items []models.Content
		var pager serializers.Pager
		pager.InitPager(c)
		offset := (pager.Page - 1) * pager.PageSize
		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,
			"Title": "文章列表",
		})
	})
	r.GET("/createcontent", func(c *gin.Context) {
		c.HTML(http.StatusOK, "content.tmpl", nil)
	})

	r.GET("/ws", controllers.WebSocketHandler)
	r.POST("/content", controllers.CreateContentHandler)
	r.POST("/login", controllers.UsersLoginHandler)
	r.POST("/register", controllers.UsersRegisterHandler)
	r.POST("/setinfo", controllers.UsersSetInfoHandler)
	r.POST("/setpwd", controllers.UsersSetPwdHandler)
	// Frontend routes (dynamic themes)
	r.GET("/", controllers.Home)
	r.GET("/post/:id", controllers.ShowPost)

	// // Admin panel
	// admin := r.Group("/admin", middleware.AuthRequired())
	// {
	// 	admin.GET("/themes", controllers.ListThemes)
	// 	admin.POST("/themes/switch", controllers.SwitchTheme)
	// }

	// Static files for themes
	r.StaticFS("/themes", http.Dir("web/themes"))
}
func getUserInfo() models.User {
	user := models.User{
		Name:         "user",
		Gender:       "male",
		Age:          18,
		Password:     "nothings",
		PasswordHash: []byte("nothings"),
	}
	return user
}

func esSSE(c *gin.Context) {
	w := c.Writer

	w.Header().Set("Content-Type", "text/event-stream")
	w.Header().Set("Cache-Control", "no-cache")
	w.Header().Set("Connection", "keep-alive")
	w.Header().Set("Access-Control-Allow-Origin", "*")

	_, ok := w.(http.Flusher)

	if !ok {
		log.Panic("server not support") //浏览器不兼容
	}

	_, err := fmt.Fprintf(w, "id: aaa\ndata: %s\n\n", "dsdf")
	_, er1r := fmt.Fprintf(w, "event: connecttime\ndata: %s\n\n", "connecttime")
	if err != nil || er1r != nil {
		print("error", err)
		return
	}
}