go restful源码剖析-2

综述


调试样例为examples\restful-no-cache-filter.go,在该例子主要是对hello world样例的扩展,主要在hello world中添加Filter,用来消除http访问的cache,代码如下。

func main() {
    ws := new(restful.WebService)
    ws.Filter(restful.NoBrowserCacheFilter)
    ws.Route(ws.GET("/hello").To(hello))
    restful.Add(ws)
    log.Fatal(http.ListenAndServe(":8080", nil))
}

func hello(req *restful.Request, resp *restful.Response) {
    io.WriteString(resp, "world")
}

Filter添加流程


ws.Filter(restful.NoBrowserCacheFilter)从字面意思上理解就是讲一个NoBrowserCacheFilter对象添加到filter中,NoBrowserCacheFilter是定义的FilterFunction函数。

// FilterFunction definitions must call ProcessFilter on the FilterChain to pass on the control and eventually call the RouteFunction
type FilterFunction func(*Request, *Response, *FilterChain)

// NoBrowserCacheFilter is a filter function to set HTTP headers that disable browser caching
// See examples/restful-no-cache-filter.go for usage
func NoBrowserCacheFilter(req *Request, resp *Response, chain *FilterChain) {
    resp.Header().Set("Cache-Control", "no-cache, no-store, must-revalidate") // HTTP 1.1.
    resp.Header().Set("Pragma", "no-cache")                                   // HTTP 1.0.
    resp.Header().Set("Expires", "0")                                         // Proxies.
    chain.ProcessFilter(req, resp)
}

在ws结构体中存有记录Filter相关数据的字段,对应的添加接口为,并将FilterFunction:NoBrowserCacheFilter添加到type WebService struct { filters []FilterFunction }中进行维护,ws的启动流程同example/hello-world.go的流程相同。

func (w *WebService) Filter(filter FilterFunction) *WebService {
    w.filters = append(w.filters, filter)
    return w
}

Filter函数的执行


在http请求访问时,restful调用dispatch进行req的分发,在完成route select之后,执行绑定的hello函数前,需要对用户定的的Filter进行先行判定, 并将用户定义的Filter全部统计到allFilters中,并初始化FilterChain,在restful中Filter的规则是可以被覆盖掉的,按照优先级container filter < webservices < route执行。

func (c *Container) dispatch(httpWriter http.ResponseWriter, httpRequest *http.Request) {
    if len(c.containerFilters)+len(webService.filters)+len(route.Filters) > 0 {
        // compose filter chain
        allFilters := []FilterFunction{}
        allFilters = append(allFilters, c.containerFilters...)
        allFilters = append(allFilters, webService.filters...)
        allFilters = append(allFilters, route.Filters...)
        chain := FilterChain{Filters: allFilters, Target: func(req *Request, resp *Response) {
            // handle request by route after passing all filters
            route.Function(wrappedRequest, wrappedResponse)
        }}
        chain.ProcessFilter(wrappedRequest, wrappedResponse)
    } else {// example hello world在这里执行执行绑定函数
        // no filters, handle request by route
        route.Function(wrappedRequest, wrappedResponse)
    }
}

Filter Chain中实际上是对Filter规则和目标function的进一步包装,之所以叫做chain是因为对fliter的执行是通过递归调用的方式,并有Index控制Filter的递归层数,代码如下:

// FilterChain is a request scoped object to process one or more filters before calling the target RouteFunction.
type FilterChain struct {
    Filters []FilterFunction // ordered list of FilterFunction
    Index   int              // index into filters that is currently in progress
    Target  RouteFunction    // function to call after passing all filters
}

其中Filter字段中记录了所有的Filter规则,Targe绑定了玩家定义的需要执行的函数, 后面通过调用定义在Filter中的ProcessFilter函数执行对应的Filters及user function: hello。

// ProcessFilter passes the request,response pair through the next of Filters.
// Each filter can decide to proceed to the next Filter or handle the Response itself.
func (f *FilterChain) ProcessFilter(request *Request, response *Response) {
    if f.Index < len(f.Filters) {
        f.Index++
        f.Filters[f.Index-1](request, response, f)
    } else {
        f.Target(request, response)
    }
}

用户自定义filter


既然restful提供了Filter的通用接口,并提供NoBrowserCacheFilter样例,那么用户也可以自定义响应的Filter接口,在该小结中,将仿照NoBrowserCacheFilter的实现方式在package main中自定义用户的Filter函数,改变用户访问resp的header中的字段,代码如下:


func main() {
    ws := new(restful.WebService)
    ws.Filter(restful.NoBrowserCacheFilter)
    // 将自定义Filter添加到ws Filter中
    ws.Filter(UserCustomFilter)
    ws.Route(ws.GET("/hello").To(hello))
    restful.Add(ws)
    log.Fatal(http.ListenAndServe(":8080", nil))
}
   
func UserCustomFilter(req *restful.Request, resp *restful.Response, chain *restful.FilterChain) {
    //创建新字段
    resp.Header().Set("custom", "test") // HTTP 1.1.
    //覆盖掉NoBrowserCacheFilter中的pargma字段
    resp.Header().Set("Pragma", "test") // HTTP 1.0.
    //递归调用下一个pross Filter,取名为chain的原因
    chain.ProcessFilter(req, resp)
}

http访问后返回的http header如下:其中用户自定义的Filter覆盖掉了NoBrowserCacheFilter中的pargma字段,并新添加custom字段。

cache-control → no-cache, no-store, must-revalidate
custom → test
date →Tue, 21 Aug 2018 08:40:14 GMT
expires → 0
pragma → test
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 205,236评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 87,867评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,715评论 0 340
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,899评论 1 278
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,895评论 5 368
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,733评论 1 283
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,085评论 3 399
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,722评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 43,025评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,696评论 2 323
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,816评论 1 333
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,447评论 4 322
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,057评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,009评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,254评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,204评论 2 352
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,561评论 2 343

推荐阅读更多精彩内容

  • restful hello world 首次浏览下go-restful的工程结构,从工程组织上面来看,工程包括两个...
    tcuze阅读 579评论 0 1
  • Spring Web MVC Spring Web MVC 是包含在 Spring 框架中的 Web 框架,建立于...
    Hsinwong阅读 22,310评论 1 92
  • Spring Cloud为开发人员提供了快速构建分布式系统中一些常见模式的工具(例如配置管理,服务发现,断路器,智...
    卡卡罗2017阅读 134,598评论 18 139
  • 前几天几个朋友一起吃饭,谈起一个话题:人生中最无助的是什么事? 勾起了我的回忆,一个人走进医院,挂号、排队,然后一...
    每日小情书阅读 240评论 0 1
  • 青春易逝,岁月无痕,在最美的年华选择安逸,谁辜负了谁? 十年寒窗,双亲劳累,在远行的时候选择归乡,谁辜负了谁? 志...
    哎呦那个谁阅读 308评论 0 0