Go 미들웨어에서 HTTP 헤더를 효과적으로 제어하는 ​​방법은 무엇입니까?

외부 Go 미들웨어로 HTTP 헤더 제어

문제: 미들웨어에서 서버 헤더 재정의

Go에 미들웨어가 있는 시나리오에 직면해 있습니다. 사용자 정의 "서버" 헤더를 설정하려고 시도합니다. 그러나 애플리케이션이나 기타 미들웨어에 의해 설정된 기존 "Server" 헤더가 지속되어 응답에 여러 개의 "Server" 헤더가 생성됩니다.

허용되지 않는 헤더 수정

HTTP 의미 체계에 따라 응답 헤더 수정 ServeHTTP 메서드 반환 이후에는 금지됩니다. 이 제한은 요청 처리와 응답 마무리 사이의 명확한 경계를 설정하는 것을 목표로 합니다.

제안된 솔루션

1. 사용자 정의 ResponseWriter:

헤더 수정 작업을 가로채는 사용자 정의 ResponseWriter 래퍼를 정의합니다. 헤더를 작성하기 전에 래퍼는 사용자 정의 "서버" 헤더를 삽입합니다. 이 접근 방식은 간접 계층을 추가하지만 헤더 관리에 대한 세밀한 제어를 제공합니다.

다음은 사용자 정의 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:

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)
    })
}

2. 미들웨어 순서 반전:

다른 접근 방식은 "서버" 헤더 설정 미들웨어가 다른 모든 미들웨어 다음에 실행되도록 미들웨어의 순서를 바꾸는 것입니다. 이렇게 하면 응답이 완료되기 전 헤더 수정이 마지막 작업이 됩니다.

3. 가장 내부 미들웨어:

가능하다면 "서버" 헤더 설정 미들웨어를 미들웨어의 가장 안쪽 레이어로 배치하세요. 이렇게 하면 "Server" 헤더를 설정한 후 외부 미들웨어가 이를 수정할 가능성이 제거됩니다.

4. 응답 수정:

최후의 수단으로 응답을 가로채고 이에 따라 "Server" 헤더를 수정하는 사용자 정의 http.Handler 사용을 고려할 수 있습니다. 이 접근 방식을 사용하려면 애플리케이션의 다른 기능이 중단되지 않도록 주의해서 처리해야 합니다.

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