github.com/treeverse/lakefs@v1.24.1-0.20240520134607-95648127bfb0/pkg/httputil/metrics.go (about)

     1  package httputil
     2  
     3  import "net/http"
     4  
     5  type MetricResponseWriter struct {
     6  	http.ResponseWriter
     7  	StatusCode int
     8  }
     9  
    10  func NewMetricResponseWriter(w http.ResponseWriter) *MetricResponseWriter {
    11  	return &MetricResponseWriter{ResponseWriter: w, StatusCode: http.StatusOK}
    12  }
    13  
    14  func (mrw *MetricResponseWriter) WriteHeader(code int) {
    15  	mrw.StatusCode = code
    16  	mrw.ResponseWriter.WriteHeader(code)
    17  }