statusresponsewriter.go

package httputils

import (
	"net/http"
	"sync"
)

type StatusResponseWriter interface {
	http.ResponseWriter
	StatusSent() bool
}

func ToStatusResponseWriter(w http.ResponseWriter) StatusResponseWriter {
	return &statusResponseWriter{w: w}
}

type statusResponseWriter struct {
	sync.Mutex
	w    http.ResponseWriter
	used bool
}

func (srw *statusResponseWriter) Header() http.Header {
	return srw.w.Header()
}

func (srw *statusResponseWriter) Write(p []byte) (n int, err error) {
	srw.Lock()
	srw.used = true
	srw.Unlock()
	return srw.w.Write(p)
}

func (srw *statusResponseWriter) WriteHeader(code int) {
	srw.Lock()
	srw.used = true
	srw.Unlock()
	srw.w.WriteHeader(code)
}

func (srw *statusResponseWriter) StatusSent() bool {
	srw.Lock()
	used := srw.used
	srw.Unlock()
	return used
}