本文总结了我平时在项目中积累的关于gin框架路由拆分与注册的若干方法。
标题:gin框架路由拆分与注册
1. 基本的路由注册
下面最基础的gin路由注册方式,适用于路由条目比较少的简单项目或者项目demo。
package mainimport ("net/http""github.com/gin-gonic/gin")func helloHandler(c *gin.Context) {c.JSON(http.StatusOK, gin.H{"message": "Hello q1mi!",})}func main() {r := gin.Default()r.GET("/hello", helloHandler)if err := r.Run(); err != nil {fmt.Println("startup service failed, err:%v\n", err)}}
2. 路由拆分成单独文件或包 - 调用一次,返回值为*gin.Engine
当项目的规模增大后就不太适合继续在项目的main.go文件中去实现路由注册相关逻辑了,我们会倾向于把路由部分的代码都拆分出来,形成一个单独的文件或包:
我们在routers.go文件中定义并注册路由信息:
package mainimport ("net/http""github.com/gin-gonic/gin")func helloHandler(c *gin.Context) {c.JSON(http.StatusOK, gin.H{"message": "Hello q1mi!",})}func setupRouter() *gin.Engine {r := gin.Default()r.GET("/hello", helloHandler)return r}
此时main.go中调用上面定义好的setupRouter函数:
func main() {r := setupRouter()if err := r.Run(); err != nil {fmt.Println("startup service failed, err:%v\n", err)}}
此时的目录结构:
gin_demo├── go.mod├── go.sum├── main.go└── routers.go
把路由部分的代码单独拆分成包的话也是可以的,拆分后的目录结构如下:
gin_demo├── go.mod├── go.sum├── main.go└── routers└── routers.go
routers/routers.go需要注意此时setupRouter需要改成首字母大写:— 相当于第三方
package routersimport ("net/http""github.com/gin-gonic/gin")func helloHandler(c *gin.Context) {c.JSON(http.StatusOK, gin.H{"message": "Hello q1mi!",})}// SetupRouter 配置路由信息func SetupRouter() *gin.Engine {r := gin.Default()r.GET("/hello", helloHandler)return r}
main.go文件内容如下:
package mainimport ("fmt""gin_demo/routers")func main() {r := routers.SetupRouter()if err := r.Run(); err != nil {fmt.Println("startup service failed, err:%v\n", err)}}
3. 路由拆分成多个文件 - 调用多个,参数是*gin.Engine
使用返回值和参数的原因:单个函数在路由文件引入一次;多个函数在main.go引入一次后当作参数传入即可。
当我们的业务规模继续膨胀,单独的一个routers文件或包已经满足不了我们的需求了,
func SetupRouter() *gin.Engine {r := gin.Default()r.GET("/hello", helloHandler)r.GET("/xx1", xxHandler1)// ...r.GET("/xx30", xxHandler30)return r}
因为我们把所有的路由注册都写在一个SetupRouter函数中的话就会太复杂了。
我们可以分开定义多个路由文件,例如:
gin_demo├── go.mod├── go.sum├── main.go└── routers├── blog.go└── shop.go
routers/shop.go中添加一个LoadShop 的函数,将shop相关的路由注册到指定的路由器:
func LoadShop(e *gin.Engine) {e.GET("/hello", helloHandler)e.GET("/goods", goodsHandler)e.GET("/checkout", checkoutHandler)// ...}
routers/blog.go中添加一个 LoadBlog 的函数,将blog相关的路由注册到指定的路由器:
func LoadBlog(e *gin.Engine) {e.GET("/post", postHandler)e.GET("/comment", commentHandler)// ...}
在main函数中实现最终的注册逻辑如下:
// main.gofunc main() {r := gin.Default()routers.LoadBlog(r);// 路由模块函数1routers.LoadShop(r);// 路由模块函数2if err := r.Run(); err != nil {fmt.Println("startup service failed, err:%v\n", err)}}
4. 路由拆分到不同的APP
有时候项目规模实在太大,那么我们就更倾向于把业务拆分的更详细一些,例如把不同的业务代码拆分成不同的APP。
因此我们在项目目录下单独定义一个app目录,用来存放我们不同业务线的代码文件,这样就很容易进行横向扩展。大致目录结构如下:
gin_demo├── app│ ├── blog│ │ ├── handler.go│ │ └── router.go│ └── shop│ ├── handler.go│ └── router.go├── go.mod├── go.sum├── main.go└── routers└── routers.go
其中app/blog/router.go用来定义post相关路由信息,具体内容如下:
func Routers(e *gin.Engine) {e.GET("/post", postHandler)e.GET("/comment", commentHandler)}
app/shop/router.go用来定义shop相关路由信息,具体内容如下:
func Routers(e *gin.Engine) {e.GET("/goods", goodsHandler)e.GET("/checkout", checkoutHandler)}
routers/routers.go中根据需要定义Include函数用来注册子app中定义的路由,Init函数用来进行路由的初始化操作:
type Option func(*gin.Engine)var options = []Option{}// 注册app的路由配置func Include(opts ...Option) {options = append(options, opts...)}// 初始化func Init() *gin.Engine {r := gin.New()for _, opt := range options {opt(r)}return r}
main.go中按上述方式先注册子app中的路由,然后再进行路由的初始化:
func main() {// 加载多个APP的路由配置routers.Include(shop.Routers, blog.Routers)// 初始化路由r := routers.Init()if err := r.Run(); err != nil {fmt.Println("startup service failed, err:%v\n", err)}}
总结
gin框架是一个非常容易扩展的web框架,本文是我在日常编码中总结的一点点经验,因为世界上不可能有完全相同的项目,每个人也都有自己的编程习惯,关于gin框架路由注册的方式我就在此抛砖引玉了。
