Golang中的路由使用详解

脚本专栏 2024/11/17 佚名

2 0 1

极乐门资源网 Design By www.ioogu.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在接口这方面确实设计的很精妙。

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

Golang,路由

标签：

Golang,路由

极乐门资源网 Design By www.ioogu.com

极乐门资源网 免责声明：本站文章均来自网站采集或用户投稿，网站不提供任何软件下载或自行开发的软件！如有用户或公司发现本站内容信息存在侵权行为，请邮件告知！ 858582#qq.com

极乐门资源网 Design By www.ioogu.com

评论“Golang中的路由使用详解”

暂无Golang中的路由使用详解的评论...

《魔兽世界》大逃杀！60人新游玩模式《强袭风暴》3月21日上线

暴雪近日发布了《魔兽世界》10.2.6 更新内容，新游玩模式《强袭风暴》即将于3月21 日在亚服上线，届时玩家将前往阿拉希高地展开一场 60 人大逃杀对战。

艾泽拉斯的冒险者已经征服了艾泽拉斯的大地及遥远的彼岸。他们在对抗世界上最致命的敌人时展现出过人的手腕，并且成功阻止终结宇宙等级的威胁。当他们在为即将于《魔兽世界》资料片《地心之战》中来袭的萨拉塔斯势力做战斗准备时，他们还需要在熟悉的阿拉希高地面对一个全新的敌人──那就是彼此。在《巨龙崛起》10.2.6 更新的《强袭风暴》中，玩家将会进入一个全新的海盗主题大逃杀式限时活动，其中包含极高的风险和史诗级的奖励。
《强袭风暴》不是普通的战场，作为一个独立于主游戏之外的活动，玩家可以用大逃杀的风格来体验《魔兽世界》，不分职业、不分装备（除了你在赛局中捡到的），光是技巧和战略的强弱之分就能决定出谁才是能坚持到最后的赢家。本次活动将会开放单人和双人模式，玩家在加入海盗主题的预赛大厅区域前，可以从强袭风暴角色画面新增好友。游玩游戏将可以累计名望轨迹，《巨龙崛起》和《魔兽世界：巫妖王之怒经典版》的玩家都可以获得奖励。

更新日志

2024年11月17日

Golang中的路由使用详解

Golang,路由

详解Go语言RESTful JSON API创建

Go语言中TCP/IP网络编程的深入讲解

评论“Golang中的路由使用详解”

《魔兽世界》大逃杀！60人新游玩模式《强袭风暴》3月21日上线

更新日志

友情链接