欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

深入理解Go语言中的Dispatcher

程序员文章站 2022-06-24 09:20:53
介绍 go使用goroutines来处理connection的读写事件,不会阻塞: c, err := srv.newconn(rw) if err !=...

介绍

go使用goroutines来处理connection的读写事件,不会阻塞:

c, err := srv.newconn(rw)
  if err != nil {
    continue
  }
  go c.serve()

c即为创建的connection,保存了该次请求的信息,然后再传递到对应的handler,handler就可以读取到请求的header信息,保证了请求之间独立。

go中的servemux

上面代码中提到了c(这个c就是connection).serve()方法。其实内部是调用了http包默认的路由器,通过路由器把本次请求的信息传递到了后端的处理函数。

默认路由器servemux,结构如下:

type servemux struct {
 mu sync.rwmutex  //锁,由于请求涉及到并发处理,因此这里需要一个锁机制
 m map[string]muxentry // 路由规则,一个string对应一个mux实体,这里的string就是注册的路由表达式
 hosts bool // 是否在任意的规则中带有host信息
}

下面看一下muxentry:

type muxentry struct {
 explicit bool  // 是否精确匹配
 h    handler // 这个路由表达式对应哪个handler
 pattern string //匹配字符串
}

接着看一下handler的定义:

type handler interface {
 servehttp(responsewriter, *request) // 路由实现器
}

handler是一个接口,但是前一小节中的sayhelloname函数并没有实现servehttp这个接口,仍然能添加到路由表中,原因就是http包里还有一个handlerfunc,我们定义的函数sayhelloname就是这个handlerfunc调用的结果,而这个类型默认实现了servehttp这个接口,即我们调用了handlerfunc(f) ,强制类型转换f成为handlerfunc类型,这样f就拥有了servehttp方法。

type handlerfunc func(responsewriter, *request)
// servehttp calls f(w, r).
func (f handlerfunc) servehttp(w responsewriter, r *request) {
  f(w, r)
}

我们看一下handlerfunc的官方注解:

handlerfunc类型是一个适配器,允许使用普通的函数作为http处理程序。如果f是具有适当签名的函数,handlerfunc(f)是调用f的handler。

适当的签名,由于作者水平也不深厚(毕竟我本命语言是java),猜一下指的应该是函数的参数以及返回值,也就是说:如果函数的参数是两个,分别是responsewriter和一个指向request的指针,并且返回值为void类型的函数,可以强转为handlerfunc,而最终调用的f中的handler接口的方法也就是servehttp。

路由器里面存储好了相应的路由规则之后,那么具体的请求又是怎么分发的呢?请看下面的代码,默认的路由器实现了servehttp:

func (mux *servemux) servehttp(w responsewriter, r *request) {
 if r.requesturi == "*" {
 w.header().set("connection", "close")
 w.writeheader(statusbadrequest)
 return
 }
 h, _ := mux.handler(r)
 h.servehttp(w, r)
}

如上所示路由器接收到请求之后,如果是*那么关闭链接,不然调用mux.handler(r)返回对应设置路由的处理handler,然后执行h.servehttp(w, r) 。看一下servemux.handler(*request)的官方文档:

handler返回用于给定请求的处理程序,请咨询r.methodr.hostr.url.path。它总是返回一个非nil处理程序。如果路径不是其规范形式,处理程序将是重定向到规范路径的内部生成的处理程序。

handler还返回与请求匹配的注册模式,或者在内部生成的重定向的情况下,返回在跟随重定向之后匹配的模式。

如果没有适用于请求的注册处理程序,则handler返回“未找到页面”处理程序和空模式。

说白了,根据request的method、host和请求的url的路径返回一个处理程序,这个处理程序就是我们说过的handler,再看看handler接口的方法,我们就知道了,最终会跑到我们sayhelloname里面~。我们看看servemux.handler(*request)的实现:

func (mux *servemux) handler(r *request) (h handler, pattern string) {
 if r.method != "connect" {
 if p := cleanpath(r.url.path); p != r.url.path {
  _, pattern = mux.handler(r.host, p)
  return redirecthandler(p, statusmovedpermanently), pattern
 }
 }  
 return mux.handler(r.host, r.url.path)
}
func (mux *servemux) handler(host, path string) (h handler, pattern string) {
 mux.mu.rlock()
 defer mux.mu.runlock()
 // host-specific pattern takes precedence over generic ones
 if mux.hosts {
 h, pattern = mux.match(host + path)
 }
 if h == nil {
 h, pattern = mux.match(path)
 }
 if h == nil {
 h, pattern = notfoundhandler(), ""
 }
 return
}

为了不让读者懵逼,我们还是看一下match方法,这是个私有方法,循环迭代了mux中的map:

func (mux *servemux) match(path string) (h handler, pattern string) {
 var n = 0
 for k, v := range mux.m {
 if !pathmatch(k, path) {
  continue
 }
 if h == nil || len(k) > n {
  n = len(k)
  h = v.h
  pattern = v.pattern
 }
 }
 return
}

匹配到之后返回存储的handler,调用这个handler的servehttp接口就可以执行到相应的函数了。

go其实支持外部实现的路由器 listenandserve的第二个参数就是用以配置外部路由器的,它是一个handler接口,即外部路由器只要实现了handler接口就可以,我们可以在自己实现的路由器的servehttp里面实现自定义路由功能。

我们实现一个简易路由器:

package main
import (
 "fmt"
 "net/http"
)
type mymux struct {}
func (p *mymux) servehttp(w http.responsewriter, r *http.request) {
 if r.url.path == "/" {
 sayhelloname(w, r)
 return
 }
 http.notfound(w, r)
 return
}
func sayhelloname(w http.responsewriter, r *http.request) {
 fmt.fprintf(w, "hello myroute!")
}
func main() {
 mux := &mymux{}
 http.listenandserve(":9090", mux)
}

通过对http包的分析之后,现在让我们来梳理一下整个的代码执行过程:

1、首先调用http.handlefunc,按顺序做了几件事:

  • 调用了defaultservemux的handlefunc
  • 调用了defaultservemux的handle
  • 往defaultservemux的map[string]muxentry中增加对应的handler和路由规则

2、其次调用http.listenandserve(“:9090”, nil) ,按顺序做了几件事情:

  • 实例化server
  • 调用server的listenandserve()
  • 调用net.listen(“tcp”, addr)监听端口
  • 启动一个for循环,在循环体中accept请求
  • 对每个请求实例化一个conn,并且开启一个goroutine为这个请求进行服务go c.serve()
  • 读取每个请求的内容w, err := c.readrequest()
  • 判断handler是否为空,如果没有设置handler(这个例子就没有设置handler),handler就设置为defaultservemux
  • 调用handler的servehttp
  • 在这个例子中,下面就进入到defaultservemux.servehttp
  • 根据request选择handler,并且进入到这个handler的servehttp,mux.handler(r).servehttp(w, r)
  • 选择handler:
  • 判断是否有路由能满足这个request(循环遍历servermux的muxentry)
  • 如果有路由满足,调用这个路由handler的servehttp
  • 如果没有路由满足,调用notfoundhandler的servehttp

总结

以上就是这篇文章的全部内容了,希望本文的内容对大家的学习或者工作能带来一定的帮助,如果有疑问大家可以留言交流,谢谢大家对的支持。