Отображение пользовательской страницы ошибки 404 со стандартным пакетом http

Предполагая, что имеем:

http.HandleFunc("/smth", smthPage)
http.HandleFunc("/", homePage)

Пользователь видит, что "404 страница не найдена", когда они используют неправильный URL-адрес. Как я могу вернуть пользовательскую страницу для этого случая?

Обновление относительно gorilla/mux

Принятый ответ подходит для тех, кто использует чистый сетевой/http-пакет.

Если вы используете gorilla/mux, вы должны использовать что-то вроде этого:

func main() {
    r := mux.NewRouter()
    r.NotFoundHandler = http.HandlerFunc(notFound)
}

И реализуйте func notFound(w http.ResponseWriter, r *http.Request), как вы хотите.

Ответы

Ответ 1

Я обычно делаю это:

package main

import (
    "fmt"
    "net/http"
)

func main() {
    http.HandleFunc("/", homeHandler)
    http.HandleFunc("/smth/", smthHandler)
    http.ListenAndServe(":12345", nil)
}

func homeHandler(w http.ResponseWriter, r *http.Request) {
    if r.URL.Path != "/" {
        errorHandler(w, r, http.StatusNotFound)
        return
    }
    fmt.Fprint(w, "welcome home")
}

func smthHandler(w http.ResponseWriter, r *http.Request) {
    if r.URL.Path != "/smth/" {
        errorHandler(w, r, http.StatusNotFound)
        return
    }
    fmt.Fprint(w, "welcome smth")
}

func errorHandler(w http.ResponseWriter, r *http.Request, status int) {
    w.WriteHeader(status)
    if status == http.StatusNotFound {
        fmt.Fprint(w, "custom 404")
    }
}

Здесь я упростил код, чтобы показывать только 404, но на самом деле я делаю больше с этой настройкой: я обрабатываю все HTTP-ошибки с помощью errorHandler, в которых я записываю полезную информацию и отправляю электронную почту себе.

Ответ 2

Вам просто нужно создать собственный обработчик notFound и зарегистрировать его с помощью HandleFunc для пути, который вы не обрабатываете.

Если вы хотите больше всего контролировать свою логику маршрутизации, вам нужно будет использовать собственный сервер и собственный тип обработчика.

Это позволяет вам реализовать более сложную логику маршрутизации, чем позволяет HandleFunc.

Ответ 3

Ниже приводится подход, который я выбираю. Он основан на фрагменте кода, который я не могу подтвердить, так как потерял закладку браузера.

Пример кода: (я помещаю его в свой основной пакет)

type hijack404 struct {
    http.ResponseWriter
    R *http.Request
    Handle404 func (w http.ResponseWriter, r *http.Request) bool
}

func (h *hijack404) WriteHeader(code int) {
    if 404 == code && h.Handle404(h.ResponseWriter, h.R) {
        panic(h)
    }

    h.ResponseWriter.WriteHeader(code)
}

func Handle404(handler http.Handler, handle404 func (w http.ResponseWriter, r *http.Request) bool) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request){
        hijack := &hijack404{ ResponseWriter:w, R: r, Handle404: handle404 }

        defer func() {
            if p:=recover(); p!=nil {
                if p==hijack {
                    return
                }
                panic(p)
            }
        }()

        handler.ServeHTTP(hijack, r)
    })
}

func fire404(res http.ResponseWriter, req *http.Request) bool{
    fmt.Fprintf(res, "File not found. Please check to see if your URL is correct.");

    return true;
}

func main(){
    handler_statics := http.StripPrefix("/static/", http.FileServer(http.Dir("/Path_To_My_Static_Files")));

    var v_blessed_handler_statics http.Handler = Handle404(handler_statics, fire404);

    http.Handle("/static/", v_blessed_handler_statics);

    // add other handlers using http.Handle() as necessary

    if err := http.ListenAndServe(":8080", nil); err != nil{
        log.Fatal("ListenAndServe: ", err);
    }
}

Пожалуйста, настройте func fire404 для вывода собственной версии сообщения для ошибки 404.

Если вы используете Gorilla Mux, вы можете заменить основную функцию ниже:

func main(){
    handler_statics := http.StripPrefix("/static/", http.FileServer(http.Dir("/Path_To_My_Static_Files")));

    var v_blessed_handler_statics http.Handler = Handle404(handler_statics, fire404);

    r := mux.NewRouter();
    r.PathPrefix("/static/").Handler(v_blessed_handler_statics);

    // add other handlers with r.HandleFunc() if necessary...

    http.Handle("/", r);

    log.Fatal(http.ListenAndServe(":8080", nil));
}

Пожалуйста, не стесняйтесь исправить код, если он неправильный, так как я только новичок для Go. Спасибо.

Ответ 4

Древняя нить, но я просто сделал что-то для перехвата http.ResponseWriter, может быть здесь уместно.

package main

//GAE POC originally inspired by https://thornelabs.net/2017/03/08/use-google-app-engine-and-golang-to-host-a-static-website-with-same-domain-redirects.html

import (
    "net/http"
)

func init() {
    http.HandleFunc("/", handler)
}

// HeaderWriter is a wrapper around http.ResponseWriter which manipulates headers/content based on upstream response
type HeaderWriter struct {
    original http.ResponseWriter
    done     bool
}

func (hw *HeaderWriter) Header() http.Header {
    return hw.original.Header()
}

func (hw *HeaderWriter) Write(b []byte) (int, error) {
    if hw.done {
        //Silently let caller think they are succeeding in sending their boring 404...
        return len(b), nil
    }
    return hw.original.Write(b)
}

func (hw *HeaderWriter) WriteHeader(s int) {
    if hw.done {
        //Hmm... I don't think this is needed...
        return
    }
    if s < 400 {
        //Set CC header when status is < 400...
        //TODO: Use diff header if static extensions
        hw.original.Header().Set("Cache-Control", "max-age=60, s-maxage=2592000, public")
    }
    hw.original.WriteHeader(s)
    if s == 404 {
        hw.done = true
        hw.original.Write([]byte("This be custom 404..."))
    }
}

func handler(w http.ResponseWriter, r *http.Request) {
    urls := map[string]string{
        "/example-post-1.html": "https://example.com/post/example-post-1.html",
        "/example-post-2.html": "https://example.com/post/example-post-2.html",
        "/example-post-3.html": "https://example.com/post/example-post-3.html",
    }
    w.Header().Set("Strict-Transport-Security", "max-age=15768000")
    //TODO: Put own logic
    if value, ok := urls[r.URL.Path]; ok {
        http.Redirect(&HeaderWriter{original: w}, r, value, 301)
    } else {
        http.ServeFile(&HeaderWriter{original: w}, r, "static/"+r.URL.Path)
    }
}

Ответ 5

Возможно, я ошибаюсь, но я просто проверял источники: http://golang.org/src/pkg/net/http/server.go

Кажется, что указание пользовательской функции NotFound() вряд ли возможно: NotFoundHandler() возвращает жестко запрограммированную функцию NotFound().

Вероятно, вам следует отправить сообщение об этом.

В качестве обходного пути вы можете использовать обработчик "/", который является резервным, если не найдено никаких других обработчиков (так как это самый короткий). Итак, проверьте, существует ли страница в этом обработчике и возвращает пользовательскую ошибку 404.

Ответ 6

вы можете определить

http.HandleFunc("/", func(writer http.ResponseWriter, request *http.Request) {
    if request.URL.Path != "/" {
        writer.WriteHeader(404)
        writer.Write([]byte('not found, da xiong dei !!!'))
        return
    }
})

когда доступ не найден ресурс, он будет выполняться на http.HandleFunc("/", xxx)