一、简介
在使用过beego、gin、kratos等常用框架后,他们的设计理念和提供的功能有很多共性,为了更好提高自己的一些理解,决定简单的写一个web框架。
二、编写
1. 路由
参考gin框架,我们先写一个engine,将路由和对应的执行方法利用map的方式绑定起来,并且把engine的生成方法收束起来由框架控制。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 package zedimport ( "fmt" "net/http" )type HandlerFunc func (http.ResponseWriter, *http.Request) type Engine struct { router map [string ]HandlerFunc }func (engine *Engine) ServeHTTP (w http.ResponseWriter, req *http.Request) { key := req.Method + "-" + req.URL.Path if handler, ok := engine.router[key]; ok { handler(w, req) } else { fmt.Fprintf(w, "404 not found:%s\n" , req.URL) } }func (engine *Engine) Run (addr string ) (err error) { return http.ListenAndServe(addr, engine) }func (engine *Engine) addRoute (method string , pattern string , handler HandlerFunc) { key := method + "-" + pattern engine.router[key] = handler }func (engine *Engine) GET (pattern string , handler HandlerFunc) { engine.addRoute("GET" , pattern, handler) }func (engine *Engine) POST (pattern string , handler HandlerFunc) { engine.addRoute("POST" , pattern, handler) }func New () *Engine { return &Engine{router: map [string ]HandlerFunc{}} }
这样一个简单的路由方法就写好了,这样有什么问题呢,首先,在高并发的场景下,可能有大量的请求不会马上返回,这时候我们就需要一个超时控制,其次,在原生的net/http中每个请求都需要对它的header进行设置,代码繁杂,初次之外,对于框架来说,可能还要支撑一些其他功能,例如中间件,那么中间件的消息怎么传递呢,这时候我们就需要context。
2. context
新增了一个Context的结构体,把请求参数和返回参数都放到里面,另外再封装一些常用信息和方法,这样可以给用户一些简洁的写法。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 package zedimport ( "encoding/json" "fmt" "net/http" )type H map [string ]interface {}type Context struct { Writer http.ResponseWriter Req *http.Request Method string Path string StatusCode int }func newContext (w http.ResponseWriter, req *http.Request) *Context { return &Context{ Writer: w, Req: req, Method: req.Method, Path: req.URL.Path, } }func (c *Context) PostForm (key string ) string { return c.Req.FormValue(key) }func (c *Context) Query (key string ) string { return c.Req.URL.Query().Get(key) }func (c *Context) Status (code int ) { c.StatusCode = code c.Writer.WriteHeader(code) }func (c *Context) SetHeader (key string , value string ) { c.Writer.Header().Set(key, value) }func (c *Context) String (code int , format string , value ...interface {}) { c.SetHeader("Content-Type" , "text/plain" ) c.Status(code) c.Writer.Write([]byte (fmt.Sprintf(format, value...))) }func (c *Context) JSON (code int , obj interface {}) { c.SetHeader("Content-Type" , "application/json" ) c.Status(code) encoder := json.NewEncoder(c.Writer) err := encoder.Encode(obj) if err != nil { http.Error(c.Writer, err.Error(), 500 ) } }func (c *Context) Data (code int , data []byte ) { c.Status(code) c.Writer.Write(data) }func (c *Context) HTML (code int , html string ) { c.SetHeader("Content-Type" , "text/html" ) c.Status(code) c.Writer.Write([]byte (html)) }
同时重新封装一下路由,为后面路由的扩展做准备
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 package zedimport ( "log" "net/http" )type router struct { handlers map [string ]HandlerFunc }func newRouter () *router { return &router{handlers: make (map [string ]HandlerFunc)} }func (r *router) addRoute (method string , pattern string , handler HandlerFunc) { log.Printf("Route %4s - %s" , method, pattern) key := method + "-" + pattern r.handlers[key] = handler }func (r *router) handle (c *Context) { key := c.Method + "-" + c.Path if handler, ok := r.handlers[key]; ok { handler(c) } else { c.String(http.StatusNotFound, "404 NOT FOUND: %s\n" , c.Path) } }
3. 前缀树路由
Trie树
之前使用的是map的方式用来匹配路由,虽然索引很高效,但是这种方式不够灵活,只能检索动态路由。
实现动态路由最常用的数据结构,被称为前缀树,每一个节点的所有的子节点都拥有相同的前缀。而HTTP的请求路径是由/
分隔开的,每一段都可以作为一个前缀树的节点,我们通过树结构查询,如果某一层的节点都不满足条件,就说明没有匹配到的路由,查询结束。
实现
首先我们需要设计树节点上应该存储那些信息。
1 2 3 4 5 6 type node struct { pattern string part string children []*node isWild bool }
与普通的树不同,为了实现动态路由匹配,加上了isWild
这个参数。即当我们匹配 /p/go/doc/
这个路由时,第一层节点,p
精准匹配到了p
,第二层节点,go
模糊匹配到:lang
,那么将会把lang
这个参数赋值为go
,继续下一层匹配。我们将匹配的逻辑,包装为一个辅助函数。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 func (n *node) matchChild (part string ) *node { for _, child := range n.children { if child.part == part || child.isWild { return child } } return nil }func (n *node) matchChildren (part string ) []*node { nodes := make ([]*node, 0 ) for _, child := range n.children { if child.part == part || child.isWild { nodes = append (nodes, child) } } return nodes }
对于路由来说,最重要的当然是注册与匹配了。开发服务时,注册路由规则,映射handler;访问时,匹配路由规则,查找到对应的handler。因此,Trie 树需要支持节点的插入与查询。插入功能很简单,递归查找每一层的节点,如果没有匹配到当前part
的节点,则新建一个,有一点需要注意,/p/:lang/doc
只有在第三层节点,即doc
节点,pattern
才会设置为/p/:lang/doc
。p
和:lang
节点的pattern
属性皆为空。因此,当匹配结束时,我们可以使用n.pattern == ""
来判断路由规则是否匹配成功。例如,/p/python
虽能成功匹配到:lang
,但:lang
的pattern
值为空,因此匹配失败。查询功能,同样也是递归查询每一层的节点,退出规则是,匹配到了*
,匹配失败,或者匹配到了第len(parts)
层节点。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 func (n *node) insert (pattern string , parts []string , height int ) { if len (parts) == height { n.pattern = pattern return } part := parts[height] child := n.matchChild(part) if child == nil { child = &node{part: part, isWild: part[0 ] == ':' || part[0 ] == '*' } n.children = append (n.children, child) } child.insert(pattern, parts, height+1 ) }func (n *node) search (parts []string , height int ) *node { if len (parts) == height || strings.HasPrefix(n.part, "*" ) { if n.pattern == "" { return nil } return n } part := parts[height] children := n.matchChildren(part) for _, child := range children { result := child.search(parts, height+1 ) if result != nil { return result } } return nil }
Router
Trie 树的插入与查找都成功实现了,接下来我们将 Trie 树应用到路由中去吧。我们使用 roots 来存储每种请求方式的Trie 树根节点。使用 handlers 存储每种请求方式的 HandlerFunc 。getRoute 函数中,还解析了:
和*
两种匹配符的参数,返回一个 map 。例如/p/go/doc
匹配到/p/:lang/doc
,解析结果为:{lang: "go"}
,/static/css/geektutu.css
匹配到/static/*filepath
,解析结果为{filepath: "css/geektutu.css"}
。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 type router struct { roots map [string ]*node handlers map [string ]HandlerFunc } func newRouter () *router { return &router{ roots: make (map [string ]*node), handlers: make (map [string ]HandlerFunc), } }func parsePattern (pattern string ) []string { vs := strings.Split(pattern, "/" ) parts := make ([]string , 0 ) for _, item := range vs { if item != "" { parts = append (parts, item) if item[0 ] == '*' { break } } } return parts }func (r *router) addRoute (method string , pattern string , handler HandlerFunc) { parts := parsePattern(pattern) key := method + "-" + pattern _, ok := r.roots[method] if !ok { r.roots[method] = &node{} } r.roots[method].insert(pattern, parts, 0 ) r.handlers[key] = handler }func (r *router) getRoute (method string , path string ) (*node, map [string ]string ) { searchParts := parsePattern(path) params := make (map [string ]string ) root, ok := r.roots[method] if !ok { return nil , nil } n := root.search(searchParts, 0 ) if n != nil { parts := parsePattern(n.pattern) for index, part := range parts { if part[0 ] == ':' { params[part[1 :]] = searchParts[index] } if part[0 ] == '*' && len (part) > 1 { params[part[1 :]] = strings.Join(searchParts[index:], "/" ) break } } return n, params } return nil , nil }
Context与handle的变化
在 HandlerFunc 中,希望能够访问到解析的参数,因此,需要对 Context 对象增加一个属性和方法,来提供对路由参数的访问。我们将解析后的参数存储到Params
中,通过c.Param("lang")
的方式获取到对应的值。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 type Context struct { Writer http.ResponseWriter Req *http.Request Path string Method string Params map [string ]string StatusCode int }func (c *Context) Param (key string ) string { value, _ := c.Params[key] return value }
1 2 3 4 5 6 7 8 9 10 func (r *router) handle (c *Context) { n, params := r.getRoute(c.Method, c.Path) if n != nil { c.Params = params key := c.Method + "-" + n.pattern r.handlers[key](c) } else { c.String(http.StatusNotFound, "404 NOT FOUND: %s\n" , c.Path) } }
router.go
的变化比较小,比较重要的一点是,在调用匹配到的handler
前,将解析出来的路由参数赋值给了c.Params
。这样就能够在handler
中,通过Context
对象访问到具体的值了。
<<<<<<< HEAD
4. 实现路由分组控制(Route Group Control)
=======
4. 分组控制
ba26a3a9458192af5af8922514b11e1b18dc576b
分组的意义
分组控制(Group Control)是 Web 框架应提供的基础功能之一。所谓分组,是指路由的分组。如果没有路由分组,我们需要针对每一个路由进行控制。但是真实的业务场景中,往往某一组路由需要相似的处理。例如:
以/post
开头的路由匿名可访问。
以/admin
开头的路由需要鉴权。
以/api
开头的路由是 RESTful 接口,可以对接第三方平台,需要三方平台鉴权。
大部分情况下的路由分组,是以相同的前缀来区分的。因此,我们今天实现的分组控制也是以前缀来区分,并且支持分组的嵌套。例如/post
是一个分组,/post/a
和/post/b
可以是该分组下的子分组。作用在/post
分组上的中间件(middleware),也都会作用在子分组,子分组还可以应用自己特有的中间件。
中间件可以给框架提供无限的扩展能力,应用在分组上,可以使得分组控制的收益更为明显,而不是共享相同的路由前缀这么简单。例如/admin
的分组,可以应用鉴权中间件;/
分组应用日志中间件,/
是默认的最顶层的分组,也就意味着给所有的路由,即整个框架增加了记录日志的能力。
分组嵌套
一个 Group 对象需要具备哪些属性呢?首先是前缀(prefix),比如/
,或者/api
;要支持分组嵌套,那么需要知道当前分组的父亲(parent)是谁;当然了,按照我们一开始的分析,中间件是应用在分组上的,那还需要存储应用在该分组上的中间件(middlewares)。还记得,我们之前调用函数(*Engine).addRoute()
来映射所有的路由规则和 Handler 。如果Group对象需要直接映射路由规则的话,比如我们想在使用框架时,这么调用:
1 2 3 4 5 6 7 8 9 10 11 12 <<<<<<< HEAD z := zed.New() v1 := z.Group("/v1" ) v1.GET("/" , func (c *zed.Context) { c.HTML(http.StatusOK, "<h1>Hello zed</h1>" ) ======= r := gee.New() v1 := r.Group("/v1" ) v1.GET("/" , func (c *gee.Context) { c.HTML(http.StatusOK, "<h1>Hello Gee</h1>" ) >>>>>>> ba26a3a9458192af5af8922514b11e1b18dc576b })
那么Group对象,还需要有访问Router
的能力,为了方便,我们可以在Group中,保存一个指针,指向Engine
,整个框架的所有资源都是由Engine
统一协调的,那么就可以通过Engine
间接地访问各种接口了。
<<<<<<< HEAD
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 type RouterGroup struct { ======= 所以,最后的 Group 的定义是这样的:`` `go RouterGroup struct { >>>>>>> ba26a3a9458192af5af8922514b11e1b18dc576b prefix string middlewares []HandlerFunc // support middleware parent *RouterGroup // support nesting engine *Engine // all groups share a Engine instance } <<<<<<< HEAD // HandlerFunc defines the request handler used by zed type HandlerFunc func(ctx *Context) // Engine implement the interface of ServeHTTP type Engine struct { =======
我们还可以进一步地抽象,将Engine
作为最顶层的分组,也就是说Engine
拥有RouterGroup
所有的能力。
1 2 3 4 5 6 7 8 9 10 Engine struct { >>>>>>> ba26a3a9458192af5af8922514b11e1b18dc576b *RouterGroup router *router groups []*RouterGroup } <<<<<<< HEAD =======
那我们就可以将和路由有关的函数,都交给RouterGroup
实现了。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 >>>>>>> ba26a3a9458192af5af8922514b11e1b18dc576bfunc New () *Engine { engine := &Engine{router: newRouter()} engine.RouterGroup = &RouterGroup{engine: engine} engine.groups = []*RouterGroup{engine.RouterGroup} return engine } <<<<<<< HEAD ======= >>>>>>> ba26a3a9458192af5af8922514b11e1b18dc576bfunc (group *RouterGroup) Group (prefix string ) *RouterGroup { engine := group.engine newGroup := &RouterGroup{ prefix: group.prefix + prefix, parent: group, engine: engine, } engine.groups = append (engine.groups, newGroup) return newGroup }func (group *RouterGroup) addRoute (method string , comp string , handler HandlerFunc) { pattern := group.prefix + comp log.Printf("Route %4s - %s" , method, pattern) group.engine.router.addRoute(method, pattern, handler) }func (group *RouterGroup) GET (pattern string , handler HandlerFunc) { group.addRoute("GET" , pattern, handler) }func (group *RouterGroup) POST (pattern string , handler HandlerFunc) { group.addRoute("POST" , pattern, handler) }
<<<<<<< HEAD
5. 中间件
=======
可以仔细观察下addRoute
函数,调用了group.engine.router.addRoute
来实现了路由的映射。由于Engine
从某种意义上继承了RouterGroup
的所有属性和方法,因为 (*Engine).engine 是指向自己的。这样实现,我们既可以像原来一样添加路由,也可以通过分组添加路由。
5. 中间件
中间件是什么
中间件(middlewares),简单说,就是非业务的技术类组件。Web 框架本身不可能去理解所有的业务,因而不可能实现所有的功能。因此,框架需要有一个插口,允许用户自己定义功能,嵌入到框架中,仿佛这个功能是框架原生支持的一样。因此,对中间件而言,需要考虑2个比较关键的点:
插入点在哪?使用框架的人并不关心底层逻辑的具体实现,如果插入点太底层,中间件逻辑就会非常复杂。如果插入点离用户太近,那和用户直接定义一组函数,每次在 Handler 中手工调用没有多大的优势了。
中间件的输入是什么?中间件的输入,决定了扩展能力。暴露的参数太少,用户发挥空间有限。
那对于一个 Web 框架而言,中间件应该设计成什么样呢?接下来的实现,基本参考了 Gin 框架。
中间件设计
Gee 的中间件的定义与路由映射的 Handler 一致,处理的输入是Context
对象。插入点是框架接收到请求初始化Context
对象后,允许用户使用自己定义的中间件做一些额外的处理,例如记录日志等,以及对Context
进行二次加工。另外通过调用(*Context).Next()
函数,中间件可等待用户自己定义的 Handler
处理结束后,再做一些额外的操作,例如计算本次处理所用时间等。即 Gee 的中间件支持用户在请求被处理的前后,做一些额外的操作。举个例子,我们希望最终能够支持如下定义的中间件,c.Next()
表示等待执行其他的中间件或用户的Handler
:
1 2 3 4 5 6 7 8 9 10 func Logger () HandlerFunc { return func (c *Context) { t := time.Now() c.Next() log.Printf("[%d] %s in %v" , c.StatusCode, c.Req.RequestURI, time.Since(t)) } }
另外,支持设置多个中间件,依次进行调用。
我们在分组控制 Group Control中讲到,中间件是应用在RouterGroup
上的,应用在最顶层的 Group,相当于作用于全局,所有的请求都会被中间件处理。那为什么不作用在每一条路由规则上呢?作用在某条路由规则,那还不如用户直接在 Handler 中调用直观。只作用在某条路由规则的功能通用性太差,不适合定义为中间件。
我们之前的框架设计是这样的,当接收到请求后,匹配路由,该请求的所有信息都保存在Context
中。中间件也不例外,接收到请求后,应查找所有应作用于该路由的中间件,保存在Context
中,依次进行调用。为什么依次调用后,还需要在Context
中保存呢?因为在设计中,中间件不仅作用在处理流程前,也可以作用在处理流程后,即在用户定义的 Handler 处理完毕后,还可以执行剩下的操作。
为此,我们给Context
添加了2个参数,定义了Next
方法:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 type Context struct { Writer http.ResponseWriter Req *http.Request Path string Method string Params map [string ]string StatusCode int handlers []HandlerFunc index int }func newContext (w http.ResponseWriter, req *http.Request) *Context { return &Context{ Path: req.URL.Path, Method: req.Method, Req: req, Writer: w, index: -1 , } }func (c *Context) Next () { c.index++ s := len (c.handlers) for ; c.index < s; c.index++ { c.handlers[c.index](c) } }
index
是记录当前执行到第几个中间件,当在中间件中调用Next
方法时,控制权交给了下一个中间件,直到调用到最后一个中间件,然后再从后往前,调用每个中间件在Next
方法之后定义的部分。如果我们将用户在映射路由时定义的Handler
添加到c.handlers
列表中,结果会怎么样呢?想必你已经猜到了。
1 2 3 4 5 6 7 8 9 10 func A (c *Context) { part1 c.Next() part2 }func B (c *Context) { part3 c.Next() part4 }
假设我们应用了中间件 A 和 B,和路由映射的 Handler。c.handlers
是这样的[A, B, Handler],c.index
初始化为-1。调用c.Next()
,接下来的流程是这样的:
c.index++,c.index 变为 0
0 < 3,调用 c.handlers[0],即 A
执行 part1,调用 c.Next()
c.index++,c.index 变为 1
1 < 3,调用 c.handlers[1],即 B
执行 part3,调用 c.Next()
c.index++,c.index 变为 2
2 < 3,调用 c.handlers[2],即Handler
Handler 调用完毕,返回到 B 中的 part4,执行 part4
part4 执行完毕,返回到 A 中的 part2,执行 part2
part2 执行完毕,结束。
一句话说清楚重点,最终的顺序是part1 -> part3 -> Handler -> part 4 -> part2
。恰恰满足了我们对中间件的要求,接下来看调用部分的代码,就能全部串起来了。
代码实现
定义Use
函数,将中间件应用到某个 Group 。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 func (group *RouterGroup) Use (middlewares ...HandlerFunc) { group.middlewares = append (group.middlewares, middlewares...) }func (engine *Engine) ServeHTTP (w http.ResponseWriter, req *http.Request) { var middlewares []HandlerFunc for _, group := range engine.groups { if strings.HasPrefix(req.URL.Path, group.prefix) { middlewares = append (middlewares, group.middlewares...) } } c := newContext(w, req) c.handlers = middlewares engine.router.handle(c) }
ServeHTTP 函数也有变化,当我们接收到一个具体请求时,要判断该请求适用于哪些中间件,在这里我们简单通过 URL 的前缀来判断。得到中间件列表后,赋值给 c.handlers
。
handle 函数中,将从路由匹配得到的 Handler 添加到 c.handlers
列表中,执行c.Next()
。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 func (r *router) handle (c *Context) { n, params := r.getRoute(c.Method, c.Path) if n != nil { key := c.Method + "-" + n.pattern c.Params = params c.handlers = append (c.handlers, r.handlers[key]) } else { c.handlers = append (c.handlers, func (c *Context) { c.String(http.StatusNotFound, "404 NOT FOUND: %s\n" , c.Path) }) } c.Next() }
6. 模板
服务端渲染
现在越来越流行前后端分离的开发模式,即 Web 后端提供 RESTful 接口,返回结构化的数据(通常为 JSON 或者 XML)。前端使用 AJAX 技术请求到所需的数据,利用 JavaScript 进行渲染。Vue/React 等前端框架持续火热,这种开发模式前后端解耦,优势非常突出。后端童鞋专心解决资源利用,并发,数据库等问题,只需要考虑数据如何生成;前端童鞋专注于界面设计实现,只需要考虑拿到数据后如何渲染即可。使用 JSP 写过网站的童鞋,应该能感受到前后端耦合的痛苦。JSP 的表现力肯定是远不如 Vue/React 等专业做前端渲染的框架的。而且前后端分离在当前还有另外一个不可忽视的优势。因为后端只关注于数据,接口返回值是结构化的,与前端解耦。同一套后端服务能够同时支撑小程序、移动APP、PC端 Web 页面,以及对外提供的接口。随着前端工程化的不断地发展,Webpack,gulp 等工具层出不穷,前端技术越来越自成体系了。
但前后分离的一大问题在于,页面是在客户端渲染的,比如浏览器,这对于爬虫并不友好。Google 爬虫已经能够爬取渲染后的网页,但是短期内爬取服务端直接渲染的 HTML 页面仍是主流。
今天的内容便是介绍 Web 框架如何支持服务端渲染的场景。
静态文件(Serve Static Files)
网页的三剑客,JavaScript、CSS 和 HTML。要做到服务端渲染,第一步便是要支持 JS、CSS 等静态文件。还记得我们之前设计动态路由的时候,支持通配符*
匹配多级子路径。比如路由规则/assets/*filepath
,可以匹配/assets/
开头的所有的地址。例如/assets/js/geektutu.js
,匹配后,参数filepath
就赋值为js/geektutu.js
。
那如果我么将所有的静态文件放在/usr/web
目录下,那么filepath
的值即是该目录下文件的相对地址。映射到真实的文件后,将文件返回,静态服务器就实现了。
找到文件后,如何返回这一步,net/http
库已经实现了。因此,gee 框架要做的,仅仅是解析请求的地址,映射到服务器上文件的真实地址,交给http.FileServer
处理就好了。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 func (group *RouterGroup) createStaticHandler (relativePath string , fs http.FileSystem) HandlerFunc { absolutePath := path.Join(group.prefix, relativePath) fileServer := http.StripPrefix(absolutePath, http.FileServer(fs)) return func (c *Context) { file := c.Param("filepath" ) if _, err := fs.Open(file); err != nil { c.Status(http.StatusNotFound) return } fileServer.ServeHTTP(c.Writer, c.Req) } }func (group *RouterGroup) Static (relativePath string , root string ) { handler := group.createStaticHandler(relativePath, http.Dir(root)) urlPattern := path.Join(relativePath, "/*filepath" ) group.GET(urlPattern, handler) }
我们给RouterGroup
添加了2个方法,Static
这个方法是暴露给用户的。用户可以将磁盘上的某个文件夹root
映射到路由relativePath
。例如:
1 2 3 4 z := zed.New() z.Static("/assets" , "/usr/geektutu/blog/static" ) z.Run(":9999" )
用户访问localhost:9999/assets/js/zed.js
,最终返回/usr/geektutu/blog/static/js/zed.js
。
HTML 模板渲染
Go语言内置了text/template
和html/template
2个模板标准库,其中html/template为 HTML 提供了较为完整的支持。包括普通变量渲染、列表渲染、对象渲染等。gee 框架的模板渲染直接使用了html/template
提供的能力。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 Engine struct { *RouterGroup router *router groups []*RouterGroup htmlTemplates *template.Template funcMap template.FuncMap }func (engine *Engine) SetFuncMap (funcMap template.FuncMap) { engine.funcMap = funcMap }func (engine *Engine) LoadHTMLGlob (pattern string ) { engine.htmlTemplates = template.Must(template.New("" ).Funcs(engine.funcMap).ParseGlob(pattern)) }
首先为 Engine 示例添加了 *template.Template
和 template.FuncMap
对象,前者将所有的模板加载进内存,后者是所有的自定义模板渲染函数。
另外,给用户分别提供了设置自定义渲染函数funcMap
和加载模板的方法。
接下来,对原来的 (*Context).HTML()
方法做了些小修改,使之支持根据模板文件名选择模板进行渲染。
1 2 3 4 5 6 7 8 9 10 11 12 13 type Context struct { engine *Engine }func (c *Context) HTML (code int , name string , data interface {}) { c.SetHeader("Content-Type" , "text/html" ) c.Status(code) if err := c.engine.htmlTemplates.ExecuteTemplate(c.Writer, name, data); err != nil { c.Fail(500 , err.Error()) } }
我们在 Context
中添加了成员变量 engine *Engine
,这样就能够通过 Context 访问 Engine 中的 HTML 模板。实例化 Context 时,还需要给 c.engine
赋值。
1 2 3 4 5 6 7 func (engine *Engine) ServeHTTP (w http.ResponseWriter, req *http.Request) { c := newContext(w, req) c.handlers = middlewares c.engine = engine engine.router.handle(c) }
7. 错误恢复
panic
Go 语言中,比较常见的错误处理方法是返回 error,由调用者决定后续如何处理。但是如果是无法恢复的错误,可以手动触发 panic,当然如果在程序运行过程中出现了类似于数组越界的错误,panic 也会被触发。panic 会中止当前执行的程序,退出。
下面是主动触发的例子:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 func main () { fmt.Println("before panic" ) panic ("crash" ) fmt.Println("after panic" ) } $ go run hello.go before panic panic : crash goroutine 1 [running]: main.main() ~/go_demo/hello/hello.go :7 +0x95 exit status 2
下面是数组越界触发的 panic
1 2 3 4 5 6 7 func main () { arr := []int {1 , 2 , 3 } fmt.Println(arr[4 ]) } $ go run hello.go panic : runtime error: index out of range [4 ] with length 3
defer
panic 会导致程序被中止,但是在退出前,会先处理完当前协程上已经defer 的任务,执行完成后再退出。效果类似于 java 语言的 try...catch
。
1 2 3 4 5 6 7 8 9 10 11 12 func main () { defer func () { fmt.Println("defer func" ) }() arr := []int {1 , 2 , 3 } fmt.Println(arr[4 ]) } $ go run hello.go defer func panic : runtime error: index out of range [4 ] with length 3
可以 defer 多个任务,在同一个函数中 defer 多个任务,会逆序执行。即先执行最后 defer 的任务。
在这里,defer 的任务执行完成之后,panic 还会继续被抛出,导致程序非正常结束。
recover
Go 语言还提供了 recover 函数,可以避免因为 panic 发生而导致整个程序终止,recover 函数只在 defer 中生效。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 func test_recover () { defer func () { fmt.Println("defer func" ) if err := recover (); err != nil { fmt.Println("recover success" ) } }() arr := []int {1 , 2 , 3 } fmt.Println(arr[4 ]) fmt.Println("after panic" ) }func main () { test_recover() fmt.Println("after recover" ) } $ go run hello.go defer func recover success after recover
我们可以看到,recover 捕获了 panic,程序正常结束。test_recover() 中的 after panic 没有打印,这是正确的,当 panic 被触发时,控制权就被交给了 defer 。就像在 java 中,try
代码块中发生了异常,控制权交给了 catch
,接下来执行 catch 代码块中的代码。而在 main() 中打印了 after recover ,说明程序已经恢复正常,继续往下执行直到结束。
zed 的错误处理机制
对一个 Web 框架而言,错误处理机制是非常必要的。可能是框架本身没有完备的测试,导致在某些情况下出现空指针异常等情况。也有可能用户不正确的参数,触发了某些异常,例如数组越界,空指针等。如果因为这些原因导致系统宕机,必然是不可接受的。
今天,我们将在 zed中添加一个非常简单的错误处理机制,即在此类错误发生时,向用户返回 Internal Server Error ,并且在日志中打印必要的错误信息,方便进行错误定位。
我们之前实现了中间件机制,错误处理也可以作为一个中间件,增强 zed框架的能力,在zed中实现中间件 Recovery
。
1 2 3 4 5 6 7 8 9 10 11 12 13 func Recovery () HandlerFunc { return func (c *Context) { defer func () { if err := recover (); err != nil { message := fmt.Sprintf("%s" , err) log.Printf("%s\n\n" , trace(message)) c.Fail(http.StatusInternalServerError, "Internal Server Error" ) } }() c.Next() } }
Recovery
的实现非常简单,使用 defer 挂载上错误恢复的函数,在这个函数中调用 recover() ,捕获 panic,并且将堆栈信息打印在日志中,向用户返回 Internal Server Error 。
你可能注意到,这里有一个 trace() 函数,这个函数是用来获取触发 panic 的堆栈信息,完整代码如下:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 package geeimport ( "fmt" "log" "net/http" "runtime" "strings" )func trace (message string ) string { var pcs [32 ]uintptr n := runtime.Callers(3 , pcs[:]) var str strings.Builder str.WriteString(message + "\nTraceback:" ) for _, pc := range pcs[:n] { fn := runtime.FuncForPC(pc) file, line := fn.FileLine(pc) str.WriteString(fmt.Sprintf("\n\t%s:%d" , file, line)) } return str.String() }func Recovery () HandlerFunc { return func (c *Context) { defer func () { if err := recover (); err != nil { message := fmt.Sprintf("%s" , err) log.Printf("%s\n\n" , trace(message)) c.Fail(http.StatusInternalServerError, "Internal Server Error" ) } }() c.Next() } }
在 trace() 中,调用了 runtime.Callers(3, pcs[:])
,Callers 用来返回调用栈的程序计数器, 第 0 个 Caller 是 Callers 本身,第 1 个是上一层 trace,第 2 个是再上一层的 defer func
。因此,为了日志简洁一点,我们跳过了前 3 个 Caller。
接下来,通过 runtime.FuncForPC(pc)
获取对应的函数,在通过 fn.FileLine(pc)
获取到调用该函数的文件名和行号,打印在日志中。
ba26a3a9458192af5af8922514b11e1b18dc576b