golang mysql封装_golang如何封装路由

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

首先看 main.go

一个初始化函数，初始化 dbfunc 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")

}

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