http.HandleFunc에서 HTTP 응답 로깅

이 문서에서는 요청을 위조하거나 http.ResponseWriter를 수정하지 않고 로깅 목적으로 HTTP 응답을 얻는 대체 접근 방식에 대해 설명합니다. 기능적 스타일 구현을 제공하는 미들웨어 체이닝 개념을 소개합니다.

미들웨어 체이닝

미들웨어 체이닝에는 기본 요청 실행 전후에 특정 작업을 수행하는 핸들러 체인에 제어권을 전달하는 작업이 포함됩니다. 미들웨어라고 알려진 이러한 핸들러는 요청과 체인의 다음 핸들러를 수신하여 순서대로 실행되도록 합니다.

HTTP 핸들러 조합기 역할을 하는 사용자 정의 미들웨어 기능을 정의합니다.

func NewResponseLoggingHandler(next http.HandlerFunc) http.HandlerFunc {
    return func(w http.ResponseWriter, r *http.Request) {
        // Replace the response writer with a recorder for subsequent handlers
        c := httptest.NewRecorder()
        next(c, r)

        // Copy data from the recorder to the original response writer
        for k, v := range c.HeaderMap {
            w.Header()[k] = v
        }
        w.WriteHeader(c.Code)
        c.Body.WriteTo(w)
    }
}

솔루션 적용

특정 카테고리의 모든 핸들러에 대한 자동 응답 로깅을 보장하기 위해 로깅 미들웨어를 캡슐화하는 또 다른 핸들러 결합자를 만듭니다.

func NewDefaultHandler(next http.HandlerFunc) http.HandlerFunc {
    return NewResponseLoggingHandler(NewOtherStuffHandler(next))
}

이제 NewDefaultHandler를 사용하여 인스턴스화된 모든 핸들러 체인에는 응답 로깅 및 기타 기본 동작이 자동으로 포함됩니다.

h := NewDefaultHandler(...)

결론

미들웨어 체이닝을 사용하면 요청을 위조하거나 수정할 필요 없이 HTTP 응답을 투명하게 가로채서 기록할 수 있습니다. http.ResponseWriter. 이 접근 방식은 모듈식 로깅을 허용하고 핸들러 관리를 단순화합니다.