golang的http.HandleFunc使用正則匹配路由

因爲只是寫一個博客，自定義一個路由器沒(wo)有(hai)必(bu)要(hui)，實現一個簡單的路由便可

package router

import (
    "net/http"
    "zhfblog/controller"
    "regexp"
    "fmt"
)

// 路由定義
type routeInfo struct {
    pattern string                                       // 正則表達式
    f       func(w http.ResponseWriter, r *http.Request) //Controller函數
}

// 路由添加
var routePath = []routeInfo{
    routeInfo{"^/a/$", controller.Index},
}

// 使用正則路由轉發
func Route(w http.ResponseWriter, r *http.Request) {
    isFound := false
    for _, p := range routePath {
        // 這裏循環匹配Path，先添加的先匹配
        reg, err := regexp.Compile(p.pattern)
        if err != nil {
            continue
        }
        if reg.MatchString(r.URL.Path) {
            isFound = true
            p.f(w, r)
        }
    }
    if !isFound {
        // 未匹配到路由
        fmt.Fprint(w, "404 Page Not Found!")
    }
}

func init() {
    // 使用"/"匹配全部路由到自定義的正則路由函數Route
    // 只需在main包導入該路由包便可
    // import _ "zhfblog/router"
    http.HandleFunc("/", Route)
}