大猩猩多路复用器，静态文件服务器的中间件

Question

请参阅下面的代码。如您所见，这是一个具有两个端点的简单服务：

1. /test <-- 使用中间件保护基本身份验证
2. /static <-- 提供 ./static 目录中的所有文件

现在我也在尝试将基本的身份验证中间件模式添加到/static 端点，但由于某种原因，我不知道如何。我在将*route(r.PathPrefix 的结果(转换为middleware()函数理解的内容时遇到问题。(我也创建了一个游乐场，但由于外部导入，这行不通(

package main
import (
"encoding/base64"
"log"
"net/http"
"strings"
"github.com/gorilla/mux"
)
const (
username = "test"
password = "test"
)
func main() {
r := mux.NewRouter()
// add normal endpoint
r.HandleFunc("/test", middleWare(myHandler, basicAuth))
// add static
r.PathPrefix("/static/").Handler(http.StripPrefix("/static/", http.FileServer(http.Dir("./static"))))
srv := &http.Server{
Handler: r,
Addr:    "0.0.0.0:8080",
}
log.Print("listening on 0.0.0.0:8080")
log.Fatal(srv.ListenAndServe())
}
func myHandler(w http.ResponseWriter, r *http.Request) {
w.Write([]byte("yes!"))
return
}
func middleWare(h http.HandlerFunc, middleware ...func(http.HandlerFunc) http.HandlerFunc) http.HandlerFunc {
for _, m := range middleware {
h = m(h)
}
return h
}
func basicAuth(h http.HandlerFunc) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
w.Header().Set("WWW-Authenticate", `Basic realm="Restricted"`)
s := strings.SplitN(r.Header.Get("Authorization"), " ", 2)
if len(s) != 2 {
http.Error(w, "Not authorized", 401)
return
}
b, err := base64.StdEncoding.DecodeString(s[1])
if err != nil {
http.Error(w, err.Error(), 401)
return
}
pair := strings.SplitN(string(b), ":", 2)
if len(pair) != 2 {
http.Error(w, "Not authorized", 401)
return
}
if pair[0] != username || pair[1] != password {
http.Error(w, "Not authorized", 401)
return
}
h.ServeHTTP(w, r)
}
}

Answer 1

好吧，事实证明我使用了 mux 的古老版本，它缺少:)use()函数。