Как использовать мультиплексор гориллы с http.TimeoutHandler

В HTTP-сервере, написанном на go, я использую gorilla/mux для маршрутизации,
я хочу использовать http.TimeoutHandler (и/или другое "промежуточное ПО"), но я не могу понять, где их можно разместить.

Чтобы было ясно:

  1. Я создаю новый маршрутизатор gorillaMux := mux.NewRouter()
  2. добавить мои маршруты вызовами типа gorillaMux.HandleFunc("/", rootHandler)
  3. Я создаю сервер server := &http.Server{Addr:":1234"} и server.ListenAndServe()

Где я могу вставить http.TimeoutHandler или любое другое промежуточное программное обеспечение в этом отношении?


go middleware gorilla
person Ali    schedule 29.10.2013    source источник

Ответы (2)


arrow_upward
11
arrow_downward

Вот как вы можете это сделать:

package main

import (
    "fmt"
    "github.com/gorilla/mux"
    "net/http"
    "time"
)

func rootHandler(w http.ResponseWriter, r *http.Request) {
    time.Sleep(5 * time.Second)
    fmt.Fprintf(w, "Hello!")
}

func main() {
    mux := mux.NewRouter()
    mux.HandleFunc("/", rootHandler)

    muxWithMiddlewares := http.TimeoutHandler(mux, time.Second*3, "Timeout!")

    http.ListenAndServe(":8080", muxWithMiddlewares)
}

Если у вас есть более одного обработчика HTTP, вы можете объединить их:

// this is quite synthetic and ugly example, but it illustrates how Handlers works
muxWithMiddlewares := http.StripPrefix("/api", http.TimeoutHandler(mux, time.Second*3, "Timeout!"))
person Kavu    schedule 29.10.2013
comment
Обратите внимание, что это применяет тайм-аут для HTTP-запроса, но запущенная горутина, которая работает rootHandler, в конце концов завершается, даже если никто не слушает. - person zed; 13.01.2021

arrow_upward
1
arrow_downward

Это решение не отвечает на использование TimeoutHandler. Я разместил это здесь на случай, если кто-то захочет точно контролировать время ожидания своего сервера. Эта альтернатива позволит при необходимости определить IdleTimeout и RequestHeaderTimeout. В этом примере я использую как gorilla/mux, так и gorilla/handlers. Надеюсь, поможет.

// ReadTimeout is a timing constraint on the client http request imposed by the server from the moment
// of initial connection up to the time the entire request body has been read.
// [Accept] --> [TLS Handshake] --> [Request Headers] --> [Request Body] --> [Response]

// WriteTimeout is a time limit imposed on client connecting to the server via http from the
// time the server has completed reading the request header up to the time it has finished writing the response.
// [Accept] --> [TLS Handshake] --> [Request Headers] --> [Request Body] --> [Response]

func main() {
    mux := router.EpicMux()

    srv := &http.Server{
        Handler:      handlers.LoggingHandler(os.Stdout, mux),
        Addr:         "localhost:8080",
        WriteTimeout: 15 * time.Second,
        ReadTimeout:  15 * time.Second,
    }

    log.Fatal(srv.ListenAndServe())
}

func EpicMux() http.Handler {
    r := mux.NewRouter()
    r.HandleFunc("/", BaseURLRouter).Methods(http.MethodGet)
    // create the subroutes for v1 and v2
    v1 := r.PathPrefix("api/v1").Subrouter()
    // register handlers to appropriate version
    v1.HandleFunc("/person", PersonHandlerV1).Methods(http.MethodPost)

    v2 := r.PathPrefix("api/v2").Subrouter()
    v2.HandleFunc("/person",    PersonHandlerV2).Methods(http.MethodPost)
    return r
}
person Ramil    schedule 01.12.2017