手把手和你一起实现一个Web框架实战——EzWeb框架(四)[Go语言笔记]Go项目实战

代码仓库：
github
gitee
中文注释，非常详尽，可以配合食用
本篇代码，请选择demo4

这一篇文章主要实现路由组功能。实现路由组对路由进行更便利的管理，如应用中间件的应用。

一、路由组设计

这里进行的设计是以路由组前缀来进行区分的。嵌套

实现一个r.Group(prefix string)函数来构造路由组，使之在调用GET、POST等方法进行路由注册时，拥有共同的前缀。GET、POST等方法注册路由时调用了addRoute()方法，即如果我们在路由组中做注册路由的方法时，我们注册的路由是”Group.prefixt+last(这里的last是路由组内注册方法的路径)”。

type RouterGroup struct {
	prefix      string
	middlewares []HandlerFunc // 用于存储分组的中间件
	engine      *Engine       // 这里实现的嵌套让其拥有了engine的全部属性,所有的分组都使用了Engine实例,可以通过engine间接的访问各种接口
}

// Engine 实现了"net/http"标准库中的 Handler 接口中的ServeHTTP方法
type Engine struct {
	*RouterGroup	//嵌套，让Engine拥有RouterGroup的全部属性，这样做体现在我们使用r.Group()创建路由组的时候
	groups []*RouterGroup
	//用于存储路由处理方法
	//key是方法类型加路径，value是用户的处理方法
	router *router
}

当我们调用r.Group(prefix string),我们就构建了一个RouterGroup实例

engine *Engine

这让它们拥有一个共同的处理实例engine，当调用addGroup方法，为某个路由组路由注册时，我们可以通过先对”路由组前缀+剩余path”，这样的处理，再通过调用group.engine.router.addRoute(method, path, handler)进行路由注册。

func (group *RouterGroup) addRoute(method string, last string, handler HandlerFunc) {
   path := group.prefix + last
   group.engine.router.addRoute(method, path, handler)
}

二、方法修改

这里体现了为什么在RouterGroup结构体中要嵌套一个engine *Engine。我们在使用这个路由组功能的demo是这样的。

r := Ez.New()
api := r.Group("/api")
{
   api.POST("/hello", func(c *Ez.Context) {
      c.JSON(http.StatusOK,Ez.H{
         "message" : "hello",
      })
   })
}

r是一个Engine指针变量,我们的Group()方法,是属于RouterGroup的，在Engine结构体中嵌套了*RouterGroup使得我们可以直接调用它。而我们通过r.Group("/api")创建的这个api变量是一个Routergroup指针变量，我们们想要使用api.POST注册路由的时候，我们在Routergroup结构体中嵌套的engine *Engine就起了相关的作用。

// New Engine
func New() *Engine {
	//新建一个engine实例
	engine := &Engine{router: newRouter()}
	engine.RouterGroup = &RouterGroup{engine: engine}
	engine.groups = []*RouterGroup{engine.RouterGroup}
	return engine
}

// Group 新建Group
func (group *RouterGroup) Group(prefix string) *RouterGroup {
   engine := group.engine
   newGroup := &RouterGroup{
      prefix:      group.prefix + prefix,       //前缀
      engine:      engine,               //任何路由组都共享一个处理实例
   }
   engine.groups = append(engine.groups, newGroup)
   return newGroup
}

// engine.router.addRoute 方法封装在router中
func (group *RouterGroup) addRoute(method string, last string, handler HandlerFunc) {
   path := group.prefix + last
   group.engine.router.addRoute(method, path, handler)
}

func (group *RouterGroup) GET(path string, handler HandlerFunc) {
   group.addRoute("GET", path, handler)
}

func (group *RouterGroup) POST(path string, handler HandlerFunc) {
   group.addRoute("POST", path, handler)
}

demo测试:

/*
@Time : 2021/8/16 下午4:01
@Author : mrxuexi
@File : main
@Software: GoLand
*/
package main

import (
	"Ez"
	"net/http"
)
func main() {
	r := Ez.New()

	api := r.Group("/api")
	{
		api.POST("/hello", func(c *Ez.Context) {
			c.JSON(http.StatusOK,Ez.H{
				"message" : "hello",
			})
		})
		api.GET("/login", func(c *Ez.Context) {
			c.JSON(200,Ez.H{
				"name" : c.PostForm("name"),
			})
		})
	}


	r.Run(":9090")
}

成功！

参考：

[1]: https://github.com/geektutu/7days-golang/tree/master/gee-web “”gee””
[2]: https://github.com/gin-gonic/gin “”gin””