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

Golang中的路由使用详解

程序员文章站 2024-01-23 11:30:10
之前有篇文章比较浅显的分析了一下golang的服务器如何实现,还有handler, defaultservemux,handlerfunc的用处。 我们现在已经明白了de...

之前有篇文章比较浅显的分析了一下golang的服务器如何实现,还有handler, defaultservemux,handlerfunc的用处。

我们现在已经明白了defaultservemux就是存放patternhandler的地方,我们称其为路由,那么我们可能会想,既然golang能够实现这个路由,我们能否也模仿一个呢?

首先我们需要一个能够保存客户端的请求的一个容器(路由)。

创建路由结构体

type copyrouter struct {
  router map[string]map[string]http.handlerfunc
}

在这里我们创建了一个像defaultservemux的路由。

客户端请求存入路由

func (c *copyrouter) handlefunc(method, pattern string, handle http.handlerfunc) {
  if method == "" {
    panic("method can not be null!")
  }

  if pattern == "" {
    panic("pattern can not be null!")
  }

  if _, ok := c.router[method][pattern]; ok {
    panic("pattern exists!")
  }

  if c.router == nil {
    c.router = make(map[string]map[string]http.handlerfunc)
  }

  if c.router[method] == nil {
    c.router[method] = make(map[string]http.handlerfunc)
  }
  c.router[method][pattern] = handle
}

这里我们模仿源码中的servemux将每一个url所对应的handler保存起来。

实现handler接口

func (c *copyrouter) servehttp(w http.responsewriter, r *http.request) {
  if f, ok := c.router[r.method][r.url.string()]; ok {
    f.servehttp(w, r)
  }
}

在这里为什么要实现这个handler接口,因为我们发现在listenandserve方法中,最后会调用h.servehttp(w, r),那么我们就只需要让我们定义的路由实现handler接口就可以了。

获取一个路由

func newrouter() *copyrouter {
  return new(copyrouter)
}

到这里,我们自己定义的路由就完成了,我们来看看使用方法。

func sayhi(w http.responsewriter, r *http.request) {
  fmt.fprint(w,"hi")
}

func main() {
  copyrouter := copyrouter.newrouter()
  copyrouter.handlefunc("get","/sayhi", sayhi)
  log.fatal(http.listenandserve("localhost:8080", copyrouter))
}

这样就完成了一个高仿版的自定义路由,是不是和golang提供给我们的servemux很像,当然我们这个路由是一个低配版的,还有很多细节没有处理。

现在再看看,我们的main函数里面的代码不是很美观,每一次都要写get或者post方法,那么我们能否提供一个比较美观的方式呢?可以,那么我们再封装一下。

func (c *copyrouter) get(pattern string, handler http.handlerfunc){
  c.handlefunc("get", pattern, handler)
}

func (c *copyrouter) post(pattern string, handler http.handlerfunc){
  c.handlefunc("post", pattern, handler)
}

...

然后再修改一下调用方式。

copyrouter.get("/sayhi",sayhi)

现在看起来是不是就美观很多了?是的,很多web框架也是这样,为什么用起来就感觉很流畅,因为这些大神们就是站在我们开发者的角度来考虑问题,提供了很方便的一些用法,封装的很完善。

再考虑一下,我们这个自定义的路由还能做些什么,如果我们要记录每一次的访问请求,该如何处理呢?也很简单,我们只需要将逻辑写在servehttp方法中就可以了,稍微修改一下我们的代码。

func (c *copyrouter) servehttp(w http.responsewriter, r *http.request) {
  if f, ok := c.router[r.method][r.url.string()]; ok {
    func (handler http.handler){
      start := time.now()
      log.printf(" 请求 [%s] 开始时间为 : %v\n", r.url.string(), start)
      f.servehttp(w, r)
      log.printf(" 请求 [%s] 完成时间为 : %v\n", r.url.string(), time.since(start))
    }(f)
  }
}

这里我们又加入了一个记录请求时间的功能,所以在这个自定义的路由里面还可以做更多的事情。

还有一点,就是我们在定义这个路由结构体的时候,能否将这个类型修改为handler呢?也就是将这个类型map[string]map[string]http.handlerfunc修改为map[string]map[string]http.handler,是可以的,但是我们在调用的时候就需要在main方法里面做一下修改。

copyrouter.get("/sayhi",handlerfunc(sayhi))

在这里做一个强制转换即可,但是这样也不是很美观。

看到这里,我们应该对一个源码中的类型重点关注一下,那就是handlerfunc。

type handlerfunc func(responsewriter, *request)

func (f handlerfunc) servehttp(w responsewriter, r *request) {
  f(w, r)
}

这里handlerfunc起到了一个适配器的作用,这是一个非常巧妙的设计,不得不说golang在接口这方面确实设计的很精妙。

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。