중복 없이 Go 미들웨어에서 HTTP 헤더를 재정의하는 방법은 무엇입니까?

외부 Go 미들웨어에서 HTTP 헤더 재정의

배경

Go의 미들웨어는 다음과 같은 용도로 사용됩니다. 요청과 응답이 핸들러에 도달하기 전에 가로채고 수정합니다. 그러나 여러 미들웨어 구성 요소가 동일한 헤더를 수정하면 해당 헤더의 여러 인스턴스가 응답에 설정될 수 있습니다.

다음 미들웨어 예시를 고려하세요.

func Server(h http.Handler, serverName string) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        w.Header().Set("Server", serverName)
        h.ServeHTTP(w, r)
    })
}

이 미들웨어가 "Server" 헤더를 설정하는 다른 미들웨어와 함께 사용되면 응답에 여러 개의 "Server" 헤더가 생성됩니다.

사용자 정의 ResponseWriter

이 문제를 해결하려면 WriteHeader() 호출을 가로채고 요청을 더 전달하기 전에 적절한 헤더를 설정하는 사용자 정의 ResponseWriter를 생성할 수 있습니다.

type serverWriter struct {
    w           http.ResponseWriter
    name        string
    wroteHeader bool
}

func (s serverWriter) WriteHeader(code int) {
    if s.wroteHeader == false {
        s.w.Header().Set("Server", s.name)
        s.wroteHeader = true
    }
    s.w.WriteHeader(code)
}

func (s serverWriter) Write(b []byte) (int, error) {
    return s.w.Write(b)
}

func (s serverWriter) Header() http.Header {
    return s.w.Header()
}

다음을 사용하여 서버 미들웨어의 이 사용자 정의 ResponseWriter를 사용하면 "Server" 헤더만 설정되어 있는지 확인할 수 있습니다.

func Server(h http.Handler, serverName string) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        sw := serverWriter{
            w:           w,
            name:        serverName,
            wroteHeader: false,
        }
        h.ServeHTTP(sw, r)
    })
}

이 솔루션을 사용하면 ServeHTTP의 의미를 위반하지 않고 외부 미들웨어에서 HTTP 헤더를 제어할 수 있습니다.

위 내용은 중복 없이 Go 미들웨어에서 HTTP 헤더를 재정의하는 방법은 무엇입니까?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!