golang如何封装路由-木庄网络博客

本文摘自php中文网，作者尚，侵删。

封装方式一、路由写在 main函数中，数据库初始连接放在 init() 函数中。、

首先看 main.go

一个初始化函数，初始化 db

func init() {
   db.Connect()
   }

第二，路由

func main() {
 
   // Configure
   router := gin.Default()
 
   // Set html render options
   htmlRender := GinHTMLRender.New()
   htmlRender.Debug = gin.IsDebugging()
   htmlRender.Layout = "layouts/default"
   // htmlRender.TemplatesDir = "templates/" // default
   // htmlRender.Ext = ".html"               // default
 
   // Tell gin to use our html render
   router.HTMLRender = htmlRender.Create()
 
   router.RedirectTrailingSlash = true
   router.RedirectFixedPath = true
 
   // Middlewares
   router.Use(middlewares.Connect)
   router.Use(middlewares.ErrorHandler)
 
   // Statics
   router.Static("/public", "./public")
 
   // Routes
 
   router.GET("/", func(c *gin.Context) {
      c.Redirect(http.StatusMovedPermanently, "/articles")
   })
 
   // Articles
   router.GET("/new", articles.New)
   router.GET("/articles/:_id", articles.Edit)
   router.GET("/articles", articles.List)
   router.POST("/articles", articles.Create)
   router.POST("/articles/:_id", articles.Update)
   router.POST("/delete/articles/:_id", articles.Delete)
 
   // Start listening
   port := Port
   if len(os.Getenv("PORT")) > 0 {
      port = os.Getenv("PORT")
   }
   router.Run(":" + port)
}

封装方式二、路由和 db单独封装到一个函数中，以传参的形式将路由传给另一个函数。

func SetUpServer() {
    r := gin.Default()
 
    // secret 相当于签名
    store, _ := sessions.NewRedisStore(10, "tcp", CONFIG.Redis.Host, CONFIG.Redis.Password, []byte(CONFIG.Server.Secret))
    r.Use(sessions.Sessions("session", store))
 
    r.Use(gin.Recovery())
    r.Use(Uniquify())
 
    r.LoadHTMLGlob("templates/*")
    r.Static("/static", "public/")
 
    SetUpRoutes(r)
 
    r.GET("/incr", func(c *gin.Context) {
        c.JSON(200, gin.H{"count": c.MustGet("key").(string)})
    })
 
    // var port string
 
    r.Run(":" + CONFIG.Server.Port)
}

封装三、将路由等封装到 Server struct 中。对此 Server 创建一个 new函数，路由等相关内容只包含在方法中。

type Server struct {
    Ws     *melody.Melody
    Routes *gin.Engine
    Games  []*Game
}
 
func New() *Server {
    ws := melody.New()
    routes := gin.Default()
    games := make([]*Game, 0)
 
    return &Server{
        Ws:     ws,
        Routes: routes,
        Games:  games,
    }
}
 
 
...
...
 
 
func (srv *Server) RoutesInit() {
    srv.Routes.GET("/jq.js", func(c *gin.Context) {
        http.ServeFile(c.Writer, c.Request, "html/jq.js")
    })
 
    srv.Routes.GET("/goyaniv.js", func(c *gin.Context) {
        http.ServeFile(c.Writer, c.Request, "html/goyaniv.js")
    })
    srv.Routes.GET("/game/:name", func(c *gin.Context) {
        cookiekey, _ := c.Request.Cookie("goyanivkey")
        cookieid, _ := c.Request.Cookie("goyanivid")
        if cookieid == nil || cookiekey == nil {
            cookieid := CreateCookie("goyanivkey", GenerateUnique())
            cookiekey := CreateCookie("goyanivid", GenerateUnique())
            http.SetCookie(c.Writer, cookieid)
            http.SetCookie(c.Writer, cookiekey)
        }
        http.ServeFile(c.Writer, c.Request, "html/game.html")
    })
    srv.Routes.GET("/gamedev/:name", func(c *gin.Context) {
        cookiekey, _ := c.Request.Cookie("goyanivkey")
        cookieid, _ := c.Request.Cookie("goyanivid")
        if cookieid == nil || cookiekey == nil {
            cookieid := CreateCookie("goyanivkey", GenerateUnique())
            cookiekey := CreateCookie("goyanivid", GenerateUnique())
            http.SetCookie(c.Writer, cookieid)
            http.SetCookie(c.Writer, cookiekey)
        }
        http.ServeFile(c.Writer, c.Request, "html/gamedev.html")
    })
 
    srv.Routes.GET("/game/:name/ws", func(c *gin.Context) {
        srv.Ws.HandleRequest(c.Writer, c.Request)
    })
 
    srv.Routes.GET("/gamedev/:name/ws", func(c *gin.Context) {
        srv.Ws.HandleRequest(c.Writer, c.Request)
    })
 
    srv.Ws.HandleMessage(func(s *melody.Session, msg []byte) {
        FireMessage(srv, s, msg)
    })
 
    srv.Ws.HandleDisconnect(func(s *melody.Session) {
        FireDisconnect(srv, s)
    })
 
    srv.Ws.HandleConnect(func(s *melody.Session) {
        FireConnect(srv, s)
    })
}
 
func (s *Server) Run() {
    s.RoutesInit()
    s.Routes.Run(":5000")
}