Go 미들웨어가 어떻게 중복 없이 HTTP 헤더를 안정적으로 제어할 수 있나요?

Outer Go 미들웨어에서 HTTP 헤더 제어

Go의 HTTP 미들웨어는 HTTP 요청과 응답을 가로채고 수정하는 편리한 방법을 제공합니다. 그러나 중복을 도입하지 않고 기존 헤더를 재정의해야 하기 때문에 외부 미들웨어에서 헤더를 제어하는 ​​것은 어려울 수 있습니다.

"서버" 헤더를 설정하는 다음 서버 미들웨어를 고려하세요.

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" 헤더를 설정하는 경우 중복된 헤더로 인해 응답이 발생합니다.

ServeHTTP가 요청이 완료된 후 ResponseWriter에 쓰기를 명시적으로 금지하기 때문에 문제가 발생합니다. 한 가지 접근 방식은 헤더 쓰기를 가로채고 첫 번째 쓰기 전에 "Server" 헤더를 삽입하는 사용자 지정 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()
}

// Server attaches a Server header to the response.
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)
    })
}

사용자 지정 ResponseWriter를 사용하면 "Server" 헤더가 추가되었는지 확인할 수 있습니다. 다른 핸들러의 동작에 관계없이 한 번만 가능합니다. 이 접근 방식은 추가 간접 계층을 도입하지만 원하는 기능은 유지합니다.

위 내용은 Go 미들웨어가 어떻게 중복 없이 HTTP 헤더를 안정적으로 제어할 수 있나요?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!