user viper

main.go

@@ -1,47 +1,34 @@
 package main
 
 import (
-	"fmt"
-	conf "go_blog/config"
-	"go_blog/controllers"
+	"go_blog/config"
 	"go_blog/models"
 	"go_blog/routers"
-	"go_blog/serializers"
-	"log"
-	"net/http"
-	"strconv"
 
 	"go_blog/themes"
 
-	"gorm.io/driver/mysql"
 	"gorm.io/gorm"
-	"gorm.io/gorm/schema"
-	"honnef.co/go/tools/config"
 
 	"github.com/gin-gonic/gin"
 )
 
 const templatePath = "./templates/*"
 
-func init() {
-	conf.SetUp()
-	models.SetUp()
-}
-
 func main() {
-
+	conf, err := config.LoadConfig("config.yml")
+	if err != nil {
+		panic("配置文件加载失败: " + err.Error())
+	}
+	models.InitDatabase(conf)
 	// 4. 初始化主题管理器
 	themeManager := themes.NewManager()
-	if err := themeManager.LoadTheme(conf.GetCurrentTheme()); err != nil {
-		panic("主题加载失败: " + err.Error())
-	}
+	// if err := themeManager.LoadTheme(conf.GetCurrentTheme()); err != nil {
+	// 	panic("主题加载失败: " + err.Error())
+	// }
 
 	// 5. 创建Gin实例
 	router := gin.Default()
 	router.LoadHTMLGlob(templatePath)
-	// 注册WebSocket路由
-	registerRoutes(router)
-	router.GET("/events", esSSE)
 
 	// 6. 注册中间件
 	router.Use(
@@ -58,120 +45,7 @@ func main() {
 	// 8. 注册路由
 	routers.RegisterRoutes(router)
 	// 9. 启动服务
-	router.Run(":8910") //router.Run(":" + cfg.Server.Port)
-}
-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
-	}
-}
-
-func registerRoutes(r *gin.Engine) {
-
-	r.GET("/", func(c *gin.Context) {
-		var items []models.Content
-		models.DB.Select("*").Limit(5).Find(&items, "type = ?", "post")
-		c.HTML(http.StatusOK, "index.tmpl", gin.H{
-			"Items": items,
-		})
-	})
-
-	r.GET("/page", func(c *gin.Context) {
-		var items []models.Content
-		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")
-		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("/post/:id", func(c *gin.Context) {
-		id, err := strconv.ParseInt(c.Param("id"), 10, 32)
-		if err != nil {
-			return
-		}
-		var content = models.Content{Cid: int32(id)}
-		models.DB.First(&content)
-		c.HTML(http.StatusOK, "post.tmpl", content)
-	})
-	user := getUserInfo()
-	r.GET("/login", func(c *gin.Context) {
-		c.HTML(200, "login.tmpl", map[string]interface{}{
-			"title": "这个是titile,传入templates中的",
-			"user":  user,
-		})
-	})
-	r.GET("/register", func(c *gin.Context) {
-		c.HTML(200, "register.tmpl", map[string]interface{}{
-			"title": "这个是titile,传入templates中的",
-			"user":  user,
-		})
-	})
-
-	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)
-
-}
-
-func getUserInfo() models.User {
-	user := models.User{
-		Name:         "user",
-		Gender:       "male",
-		Age:          18,
-		Password:     "nothings",
-		PasswordHash: []byte("nothings"),
-	}
-	return user
-}
-
-func initDatabase() {
-	// 1. 初始化配置
-	cfg, err := config.Load("config.yaml")
-	if err != nil {
-		panic("加载配置失败: " + err.Error())
-	}
-
-	// 2. 初始化数据库
-	db, err := gorm.Open(mysql.Open(cfg.String()), &gorm.Config{
-		NamingStrategy: schema.NamingStrategy{
-			TablePrefix: conf.DataBase.Prefix,
-		},
-	})
-	if err != nil {
-		panic("数据库连接失败: " + err.Error())
-	}
-
-	// 3. 自动迁移数据模型
-	if err := db.AutoMigrate(&models.Article{}, &models.User{}); err != nil {
-		panic("数据库迁移失败: " + err.Error())
-	}
+	router.Run(":" + conf.Server.Port) //router.Run()
 }
 
 // 自定义模板渲染器