Function Wrap

wrap_generated.go:107–1207

Wrap returns a wrapped version of w that provides the exact same interface as w. Specifically if w implements any combination of: - http.Flusher - httpFlushError - http.CloseNotifier - http.Hijacker - io.ReaderFrom - deadliner - fullDuplexEnabler - http.Pusher - io.StringWriter The wrapped version

(w http.ResponseWriter, hooks Hooks)

// The CaptureMetrics implementation serves as a working example for how the
// hooks can be used.
func Wrap(w http.ResponseWriter, hooks Hooks) http.ResponseWriter {
	state := &rwState{w: w}
	var combo uint16
	if hooks.Header != nil {
		state.header = hooks.Header(w.Header)
	}
	if hooks.WriteHeader != nil {
		state.writeHeader = hooks.WriteHeader(w.WriteHeader)
	}
	if hooks.Write != nil {
		state.write = hooks.Write(w.Write)
	}
	if t0, i0 := w.(http.Flusher); i0 {
		combo |= 1 << 8
		if hooks.Flush != nil {
			state.flush = hooks.Flush(t0.Flush)
		}
	}
	if t1, i1 := w.(httpFlushError); i1 {
		combo |= 1 << 7
		if hooks.FlushError != nil {
			state.flushError = hooks.FlushError(t1.FlushError)
		} else if state.flush != nil {
			state.flushError = func() (err error) { hooks.Flush(func() { err = t1.FlushError() })(); return err }
		}
	}
	if t2, i2 := w.(http.CloseNotifier); i2 {
		combo |= 1 << 6
		if hooks.CloseNotify != nil {
			state.closeNotify = hooks.CloseNotify(t2.CloseNotify)
		}
	}
	if t3, i3 := w.(http.Hijacker); i3 {
		combo |= 1 << 5
		if hooks.Hijack != nil {
			state.hijack = hooks.Hijack(t3.Hijack)
		}
	}
	if t4, i4 := w.(io.ReaderFrom); i4 {
		combo |= 1 << 4
		if hooks.ReadFrom != nil {
			state.readFrom = hooks.ReadFrom(t4.ReadFrom)
		}
	}
	if t5, i5 := w.(deadliner); i5 {
		combo |= 1 << 3
		if hooks.SetReadDeadline != nil {
			state.setReadDeadline = hooks.SetReadDeadline(t5.SetReadDeadline)
		}
		if hooks.SetWriteDeadline != nil {
			state.setWriteDeadline = hooks.SetWriteDeadline(t5.SetWriteDeadline)
		}
	}
	if t6, i6 := w.(fullDuplexEnabler); i6 {
		combo |= 1 << 2
		if hooks.EnableFullDuplex != nil {
			state.enableFullDuplex = hooks.EnableFullDuplex(t6.EnableFullDuplex)
		}