6

假设您有两个 的实例http.ServeMux,并且您希望它们在相同的端口号上提供服务,如下所示:

    muxA, muxB http.ServeMux
    //initialise muxA
    //initialise muxB
    combinedMux := combineMux([muxA, muxB])
    http.ListenAndServe(":8080", combinedMux)

combinedMux如上所述,如何编写函数?

...或者是否有另一种方法来完成同样的事情?

4

2 回答 2

15

因为 anhttp.ServeMux也是 anhttp.Handler您可以轻松地将一个多路复用器嵌套在另一个内部,即使在相同的端口和相同的主机名上。这是一个这样做的例子:

rootMux := http.NewServeMux()
subMux := http.NewServeMux()

// This will end up handling /top_path/sub_path
subMux.HandleFunc("/sub_path", myHandleFunc)

// Use the StripPrefix here to make sure the URL has been mapped
// to a path the subMux can read
rootMux.Handle("/top_path/", http.StripPrefix("/top_path", subMux))

http.ListenAndServe(":8000", rootMux)

请注意,如果没有该http.StripPrefix()调用,您将需要处理较低多路复用器中的整个路径。

于 2017-04-12T21:34:43.453 回答
3

SeverMux 类型本身就是一个 http.Handler,因此您可以轻松地嵌套它们。例如:

mux := NewServeMux()
mux.AddHandler("server1.com:8080/", muxA)
mux.AddHandler("server2.com:8080/", muxB)

我不太确定“组合”它们的确切含义。如果您想先尝试一个处理程序,然后在 404 的情况下尝试另一个处理程序,您可以这样做(未经测试):

mux := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
    rec := httptest.NewRecorder()
    rec.Body = &bytes.Buffer{}
    muxA.ServeHTTP(rec, r)
    if rec.Code == 404 {
        muxB.ServeHTTP(w, r)
        return
    }
    for key, val := range rec.HeaderMap {
        w.Header().Set(key, val)
    }
    w.WriteHeader(rec.Code)
    rec.Body.WriteTo(w)
})

这显然有一些缺点,例如将整个响应存储在内存中。或者,如果您不介意调用您的处理程序两次,您也可以设置rec.Body = nil、检查并在成功的情况下再次rec.Code调用。muxA.ServeHTTP(w, r)但最好重组您的应用程序,以便第一种方法(嵌套的 ServerMux)就足够了。

于 2014-05-16T14:58:59.507 回答