说明

goblog 需要一款灵活的路由器来搭配 MVC 程序结构。

Go Web 开发有各式各样的路由器可供选择，我们先来看下 Go 标准库 net/http 包里的 http.ServeMux。

ServeMux 和 Handler

Go 语言中处理 HTTP 请求主要跟两个东西相关：ServeMux 和 Handler。

ServeMux 本质上是一个 HTTP 请求路由器（或者叫多路复用器，Multiplexor）。它把收到的请求与一组预先定义的 URL 路径列表做对比，然后在匹配到路径的时候调用关联的处理器（Handler）。

http 的 ServeMux 虽听起来陌生，事实上我们已经在使用它了。

重构：区分不同的 Handler

先来重构下我们的代码，修改如下：

main.go

package main

import (
	"fmt"
	"net/http"
)

func defaultHandler(w http.ResponseWriter, r *http.Request) {
	w.Header().Set("Content-Type", "text/html; charset=utf-8")
	if r.URL.Path == "/" {
		fmt.Fprint(w, "<h1>Hello, 欢迎来到 goblog！</h1>")
	} else {
		w.WriteHeader(http.StatusNotFound)
		fmt.Fprint(w, "<h1>请求页面未找到 :(</h1>"+
			"<p>如有疑惑，请联系我们。</p>")
	}
}

func aboutHandler(w http.ResponseWriter, r *http.Request) {
	w.Header().Set("Content-Type", "text/html; charset=utf-8")
	fmt.Fprint(w, "此博客是用以记录编程笔记，如您有反馈或建议，请联系 "+
		"<a href=\"mailto:summer@example.com\">summer@example.com</a>")
}

func main() {
	http.HandleFunc("/", defaultHandler)
	http.HandleFunc("/about", aboutHandler)
	http.ListenAndServe(":3000", nil)
}

浏览器访问以下三个链接，发现与之前一致：

localhost:3000/
localhost:3000/about
localhost:3000/no-where

在这里插入图片描述

查看 http.HandleFunc 源码

在 VSCode 编辑器里，把鼠标放置在 main 方法中的 http.HandleFunc 上，并同时按住 Ctrl 键（不要松开），当出现下划线的时候，点击进去：

在这里插入图片描述
此时我们看到的代码，是 net/http 包的源码。
显而易见，http 包也是由 Go 代码实现的。

仔细看下 HandleFunc 函数的定义：

func HandleFunc(pattern string, handler func(ResponseWriter, *Request)) {
    DefaultServeMux.HandleFunc(pattern, handler)
}

参数：

pattern 是 URI 的规则，例如 / 或者 about。
handler 是供调用的函数。

http.HandleFunc() 函数是对 DefaultServeMux.HandleFunc() 的封装，而我们在用相同的方法查看 http.ListenAndServe() 的源码：

在这里插入图片描述
注释里有这么一段：

The handler is typically nil, in which case the DefaultServeMux is used。

handler 通常为 nil，此种情况下会使用 DefaultServeMux。

重构：使用自定义的 ServeMux

main.go

.
.
.

func main() {
    router := http.NewServeMux()

    router.HandleFunc("/", defaultHandler)
    router.HandleFunc("/about", aboutHandler)

    http.ListenAndServe(":3000", router)
}

浏览器访问以下三个链接，与之前一致：

localhost:3000/
localhost:3000/about
localhost:3000/no-where

http.ServeMux 的局限性

http.ServeMux 在 goblog 中使用，会遇到以下几个问题：

不支持 URI 路径参数
不支持请求方法过滤
不支持路由命名

URI 路径参数

例如说博客详情页，我们的 URI 是 articles/1 这样来查看 ID 为 1 的文章。

http.ServeMux 也可以实现，新增一个路由作为示范：

main.go

func main() {
    router := http.NewServeMux()

    router.HandleFunc("/", defaultHandler)
    router.HandleFunc("/about", aboutHandler)

    // 文章详情
    router.HandleFunc("/articles/", func(w http.ResponseWriter, r *http.Request) {
        id := strings.SplitN(r.URL.Path, "/", 3)[2]
        fmt.Fprint(w, "文章 ID："+id)
    })

    http.ListenAndServe(":3000", router)
}

在这里插入图片描述

不够直观，且徒增了代码的维护成本。

请求方法过滤

无法直接从路由上区分 POST 或者 GET 等 HTTP 请求方法，只能手动判断，例如：

main.go

func main() {
	router := http.NewServeMux()

	router.HandleFunc("/", defaultHandler)
	router.HandleFunc("/about", aboutHandler)

	// 列表 or 创建
	router.HandleFunc("/articles", func(w http.ResponseWriter, r *http.Request) {
		switch r.Method {
		case "GET":
			fmt.Fprint(w, "访问文章列表")
		case "POST":
			fmt.Fprint(w, "创建新的文章")
		}
	})

	http.ListenAndServe(":3000", router)
}

在这里插入图片描述
我们使用 CURL 来测试下：

E:\golang\src>curl -X POST http://127.0.0.1:3000/articles
创建新的文章
E:\golang\src>

E:\golang\src>curl http://127.0.0.1:3000/articles
访问文章列表
E:\golang\src>