清风细雨楼

Golang中的路由使用详解
脚本专栏 2024/12/27 佚名
3 1
清风细雨楼 Design By www.eepep.com

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

我们现在已经明白了DefaultServeMux就是存放pattern和handler的地方，我们称其为路由，那么我们可能会想，既然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在接口这方面确实设计的很精妙。

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

清风细雨楼 Design By www.eepep.com
广告合作：本站广告合作请联系QQ：858582 申请时备注：广告合作（否则不回）
免责声明：本站资源来自互联网收集,仅供用于学习和交流,请遵循相关法律法规,本站一切资源不代表本站立场,如有侵权、后门、不妥请联系本站删除！
清风细雨楼 Design By www.eepep.com
www.eepep.com 清风细雨楼
120,135影音资源
344,641技术资源
22,817软件资源
435,032站长资源

更新日志

友情链接

杰晶网络 DDR爱好者之家 桃源资源网 杰网资源 富贵资源网 南强小屋 铁雪资源网 幽灵资源网 万梅资源网 狼山资源网 白云岛资源网 昆仑资源网 相思资源网 明霞山资源网 内蒙古资源网 黑松山资源网 茶园资源网 饿虎岗资源网 大旗谷资源网 常春岛资源网 岱庙资源网 兴国资源网 快活林资源网 蝙蝠岛资源网 帝王谷资源网 白云城资源网 伏龙阁资源网 清风细雨楼 天枫庄资源网 圆月山庄资源网 无争山庄资源网 神水资源网 移花宫资源网 神剑山庄资源网 无为清净楼资源网 金钱帮资源网 丐帮资源网 华山资源网 极乐门资源网 小李飞刀资源网 凤求凰客栈 风云阁资源网 金狮镖局 鸳鸯亭资源网 千金楼资源网 更多链接
清风细雨楼 Design By www.eepep.com