golang如何封装路由

封装方式一、路由写在 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 stringr.Run(":" + CONFIG.Server.Port)}

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

type Server struct {Ws     *melody.MelodyRoutes *gin.EngineGames  []*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")}

更多golang知识请关注PHP中文网golang教程栏目。

更多相关文章

随机推荐