wrap_generated_gteq_1.8.go

Documentation: github.com/felixge/httpsnoop

     1  // +build go1.8
     2  // Code generated by "httpsnoop/codegen"; DO NOT EDIT.
     3  
     4  package httpsnoop
     5  
     6  import (
     7  	"bufio"
     8  	"io"
     9  	"net"
    10  	"net/http"
    11  )
    12  
    13  // HeaderFunc is part of the http.ResponseWriter interface.
    14  type HeaderFunc func() http.Header
    15  
    16  // WriteHeaderFunc is part of the http.ResponseWriter interface.
    17  type WriteHeaderFunc func(code int)
    18  
    19  // WriteFunc is part of the http.ResponseWriter interface.
    20  type WriteFunc func(b []byte) (int, error)
    21  
    22  // FlushFunc is part of the http.Flusher interface.
    23  type FlushFunc func()
    24  
    25  // CloseNotifyFunc is part of the http.CloseNotifier interface.
    26  type CloseNotifyFunc func() <-chan bool
    27  
    28  // HijackFunc is part of the http.Hijacker interface.
    29  type HijackFunc func() (net.Conn, *bufio.ReadWriter, error)
    30  
    31  // ReadFromFunc is part of the io.ReaderFrom interface.
    32  type ReadFromFunc func(src io.Reader) (int64, error)
    33  
    34  // PushFunc is part of the http.Pusher interface.
    35  type PushFunc func(target string, opts *http.PushOptions) error
    36  
    37  // Hooks defines a set of method interceptors for methods included in
    38  // http.ResponseWriter as well as some others. You can think of them as
    39  // middleware for the function calls they target. See Wrap for more details.
    40  type Hooks struct {
    41  	Header      func(HeaderFunc) HeaderFunc
    42  	WriteHeader func(WriteHeaderFunc) WriteHeaderFunc
    43  	Write       func(WriteFunc) WriteFunc
    44  	Flush       func(FlushFunc) FlushFunc
    45  	CloseNotify func(CloseNotifyFunc) CloseNotifyFunc
    46  	Hijack      func(HijackFunc) HijackFunc
    47  	ReadFrom    func(ReadFromFunc) ReadFromFunc
    48  	Push        func(PushFunc) PushFunc
    49  }
    50  
    51  // Wrap returns a wrapped version of w that provides the exact same interface
    52  // as w. Specifically if w implements any combination of:
    53  //
    54  // - http.Flusher
    55  // - http.CloseNotifier
    56  // - http.Hijacker
    57  // - io.ReaderFrom
    58  // - http.Pusher
    59  //
    60  // The wrapped version will implement the exact same combination. If no hooks
    61  // are set, the wrapped version also behaves exactly as w. Hooks targeting
    62  // methods not supported by w are ignored. Any other hooks will intercept the
    63  // method they target and may modify the call's arguments and/or return values.
    64  // The CaptureMetrics implementation serves as a working example for how the
    65  // hooks can be used.
    66  func Wrap(w http.ResponseWriter, hooks Hooks) http.ResponseWriter {
    67  	rw := &rw{w: w, h: hooks}
    68  	_, i0 := w.(http.Flusher)
    69  	_, i1 := w.(http.CloseNotifier)
    70  	_, i2 := w.(http.Hijacker)
    71  	_, i3 := w.(io.ReaderFrom)
    72  	_, i4 := w.(http.Pusher)
    73  	switch {
    74  	// combination 1/32
    75  	case !i0 && !i1 && !i2 && !i3 && !i4:
    76  		return struct {
    77  			Unwrapper
    78  			http.ResponseWriter
    79  		}{rw, rw}
    80  	// combination 2/32
    81  	case !i0 && !i1 && !i2 && !i3 && i4:
    82  		return struct {
    83  			Unwrapper
    84  			http.ResponseWriter
    85  			http.Pusher
    86  		}{rw, rw, rw}
    87  	// combination 3/32
    88  	case !i0 && !i1 && !i2 && i3 && !i4:
    89  		return struct {
    90  			Unwrapper
    91  			http.ResponseWriter
    92  			io.ReaderFrom
    93  		}{rw, rw, rw}
    94  	// combination 4/32
    95  	case !i0 && !i1 && !i2 && i3 && i4:
    96  		return struct {
    97  			Unwrapper
    98  			http.ResponseWriter
    99  			io.ReaderFrom
   100  			http.Pusher
   101  		}{rw, rw, rw, rw}
   102  	// combination 5/32
   103  	case !i0 && !i1 && i2 && !i3 && !i4:
   104  		return struct {
   105  			Unwrapper
   106  			http.ResponseWriter
   107  			http.Hijacker
   108  		}{rw, rw, rw}
   109  	// combination 6/32
   110  	case !i0 && !i1 && i2 && !i3 && i4:
   111  		return struct {
   112  			Unwrapper
   113  			http.ResponseWriter
   114  			http.Hijacker
   115  			http.Pusher
   116  		}{rw, rw, rw, rw}
   117  	// combination 7/32
   118  	case !i0 && !i1 && i2 && i3 && !i4:
   119  		return struct {
   120  			Unwrapper
   121  			http.ResponseWriter
   122  			http.Hijacker
   123  			io.ReaderFrom
   124  		}{rw, rw, rw, rw}
   125  	// combination 8/32
   126  	case !i0 && !i1 && i2 && i3 && i4:
   127  		return struct {
   128  			Unwrapper
   129  			http.ResponseWriter
   130  			http.Hijacker
   131  			io.ReaderFrom
   132  			http.Pusher
   133  		}{rw, rw, rw, rw, rw}
   134  	// combination 9/32
   135  	case !i0 && i1 && !i2 && !i3 && !i4:
   136  		return struct {
   137  			Unwrapper
   138  			http.ResponseWriter
   139  			http.CloseNotifier
   140  		}{rw, rw, rw}
   141  	// combination 10/32
   142  	case !i0 && i1 && !i2 && !i3 && i4:
   143  		return struct {
   144  			Unwrapper
   145  			http.ResponseWriter
   146  			http.CloseNotifier
   147  			http.Pusher
   148  		}{rw, rw, rw, rw}
   149  	// combination 11/32
   150  	case !i0 && i1 && !i2 && i3 && !i4:
   151  		return struct {
   152  			Unwrapper
   153  			http.ResponseWriter
   154  			http.CloseNotifier
   155  			io.ReaderFrom
   156  		}{rw, rw, rw, rw}
   157  	// combination 12/32
   158  	case !i0 && i1 && !i2 && i3 && i4:
   159  		return struct {
   160  			Unwrapper
   161  			http.ResponseWriter
   162  			http.CloseNotifier
   163  			io.ReaderFrom
   164  			http.Pusher
   165  		}{rw, rw, rw, rw, rw}
   166  	// combination 13/32
   167  	case !i0 && i1 && i2 && !i3 && !i4:
   168  		return struct {
   169  			Unwrapper
   170  			http.ResponseWriter
   171  			http.CloseNotifier
   172  			http.Hijacker
   173  		}{rw, rw, rw, rw}
   174  	// combination 14/32
   175  	case !i0 && i1 && i2 && !i3 && i4:
   176  		return struct {
   177  			Unwrapper
   178  			http.ResponseWriter
   179  			http.CloseNotifier
   180  			http.Hijacker
   181  			http.Pusher
   182  		}{rw, rw, rw, rw, rw}
   183  	// combination 15/32
   184  	case !i0 && i1 && i2 && i3 && !i4:
   185  		return struct {
   186  			Unwrapper
   187  			http.ResponseWriter
   188  			http.CloseNotifier
   189  			http.Hijacker
   190  			io.ReaderFrom
   191  		}{rw, rw, rw, rw, rw}
   192  	// combination 16/32
   193  	case !i0 && i1 && i2 && i3 && i4:
   194  		return struct {
   195  			Unwrapper
   196  			http.ResponseWriter
   197  			http.CloseNotifier
   198  			http.Hijacker
   199  			io.ReaderFrom
   200  			http.Pusher
   201  		}{rw, rw, rw, rw, rw, rw}
   202  	// combination 17/32
   203  	case i0 && !i1 && !i2 && !i3 && !i4:
   204  		return struct {
   205  			Unwrapper
   206  			http.ResponseWriter
   207  			http.Flusher
   208  		}{rw, rw, rw}
   209  	// combination 18/32
   210  	case i0 && !i1 && !i2 && !i3 && i4:
   211  		return struct {
   212  			Unwrapper
   213  			http.ResponseWriter
   214  			http.Flusher
   215  			http.Pusher
   216  		}{rw, rw, rw, rw}
   217  	// combination 19/32
   218  	case i0 && !i1 && !i2 && i3 && !i4:
   219  		return struct {
   220  			Unwrapper
   221  			http.ResponseWriter
   222  			http.Flusher
   223  			io.ReaderFrom
   224  		}{rw, rw, rw, rw}
   225  	// combination 20/32
   226  	case i0 && !i1 && !i2 && i3 && i4:
   227  		return struct {
   228  			Unwrapper
   229  			http.ResponseWriter
   230  			http.Flusher
   231  			io.ReaderFrom
   232  			http.Pusher
   233  		}{rw, rw, rw, rw, rw}
   234  	// combination 21/32
   235  	case i0 && !i1 && i2 && !i3 && !i4:
   236  		return struct {
   237  			Unwrapper
   238  			http.ResponseWriter
   239  			http.Flusher
   240  			http.Hijacker
   241  		}{rw, rw, rw, rw}
   242  	// combination 22/32
   243  	case i0 && !i1 && i2 && !i3 && i4:
   244  		return struct {
   245  			Unwrapper
   246  			http.ResponseWriter
   247  			http.Flusher
   248  			http.Hijacker
   249  			http.Pusher
   250  		}{rw, rw, rw, rw, rw}
   251  	// combination 23/32
   252  	case i0 && !i1 && i2 && i3 && !i4:
   253  		return struct {
   254  			Unwrapper
   255  			http.ResponseWriter
   256  			http.Flusher
   257  			http.Hijacker
   258  			io.ReaderFrom
   259  		}{rw, rw, rw, rw, rw}
   260  	// combination 24/32
   261  	case i0 && !i1 && i2 && i3 && i4:
   262  		return struct {
   263  			Unwrapper
   264  			http.ResponseWriter
   265  			http.Flusher
   266  			http.Hijacker
   267  			io.ReaderFrom
   268  			http.Pusher
   269  		}{rw, rw, rw, rw, rw, rw}
   270  	// combination 25/32
   271  	case i0 && i1 && !i2 && !i3 && !i4:
   272  		return struct {
   273  			Unwrapper
   274  			http.ResponseWriter
   275  			http.Flusher
   276  			http.CloseNotifier
   277  		}{rw, rw, rw, rw}
   278  	// combination 26/32
   279  	case i0 && i1 && !i2 && !i3 && i4:
   280  		return struct {
   281  			Unwrapper
   282  			http.ResponseWriter
   283  			http.Flusher
   284  			http.CloseNotifier
   285  			http.Pusher
   286  		}{rw, rw, rw, rw, rw}
   287  	// combination 27/32
   288  	case i0 && i1 && !i2 && i3 && !i4:
   289  		return struct {
   290  			Unwrapper
   291  			http.ResponseWriter
   292  			http.Flusher
   293  			http.CloseNotifier
   294  			io.ReaderFrom
   295  		}{rw, rw, rw, rw, rw}
   296  	// combination 28/32
   297  	case i0 && i1 && !i2 && i3 && i4:
   298  		return struct {
   299  			Unwrapper
   300  			http.ResponseWriter
   301  			http.Flusher
   302  			http.CloseNotifier
   303  			io.ReaderFrom
   304  			http.Pusher
   305  		}{rw, rw, rw, rw, rw, rw}
   306  	// combination 29/32
   307  	case i0 && i1 && i2 && !i3 && !i4:
   308  		return struct {
   309  			Unwrapper
   310  			http.ResponseWriter
   311  			http.Flusher
   312  			http.CloseNotifier
   313  			http.Hijacker
   314  		}{rw, rw, rw, rw, rw}
   315  	// combination 30/32
   316  	case i0 && i1 && i2 && !i3 && i4:
   317  		return struct {
   318  			Unwrapper
   319  			http.ResponseWriter
   320  			http.Flusher
   321  			http.CloseNotifier
   322  			http.Hijacker
   323  			http.Pusher
   324  		}{rw, rw, rw, rw, rw, rw}
   325  	// combination 31/32
   326  	case i0 && i1 && i2 && i3 && !i4:
   327  		return struct {
   328  			Unwrapper
   329  			http.ResponseWriter
   330  			http.Flusher
   331  			http.CloseNotifier
   332  			http.Hijacker
   333  			io.ReaderFrom
   334  		}{rw, rw, rw, rw, rw, rw}
   335  	// combination 32/32
   336  	case i0 && i1 && i2 && i3 && i4:
   337  		return struct {
   338  			Unwrapper
   339  			http.ResponseWriter
   340  			http.Flusher
   341  			http.CloseNotifier
   342  			http.Hijacker
   343  			io.ReaderFrom
   344  			http.Pusher
   345  		}{rw, rw, rw, rw, rw, rw, rw}
   346  	}
   347  	panic("unreachable")
   348  }
   349  
   350  type rw struct {
   351  	w http.ResponseWriter
   352  	h Hooks
   353  }
   354  
   355  func (w *rw) Unwrap() http.ResponseWriter {
   356  	return w.w
   357  }
   358  
   359  func (w *rw) Header() http.Header {
   360  	f := w.w.(http.ResponseWriter).Header
   361  	if w.h.Header != nil {
   362  		f = w.h.Header(f)
   363  	}
   364  	return f()
   365  }
   366  
   367  func (w *rw) WriteHeader(code int) {
   368  	f := w.w.(http.ResponseWriter).WriteHeader
   369  	if w.h.WriteHeader != nil {
   370  		f = w.h.WriteHeader(f)
   371  	}
   372  	f(code)
   373  }
   374  
   375  func (w *rw) Write(b []byte) (int, error) {
   376  	f := w.w.(http.ResponseWriter).Write
   377  	if w.h.Write != nil {
   378  		f = w.h.Write(f)
   379  	}
   380  	return f(b)
   381  }
   382  
   383  func (w *rw) Flush() {
   384  	f := w.w.(http.Flusher).Flush
   385  	if w.h.Flush != nil {
   386  		f = w.h.Flush(f)
   387  	}
   388  	f()
   389  }
   390  
   391  func (w *rw) CloseNotify() <-chan bool {
   392  	f := w.w.(http.CloseNotifier).CloseNotify
   393  	if w.h.CloseNotify != nil {
   394  		f = w.h.CloseNotify(f)
   395  	}
   396  	return f()
   397  }
   398  
   399  func (w *rw) Hijack() (net.Conn, *bufio.ReadWriter, error) {
   400  	f := w.w.(http.Hijacker).Hijack
   401  	if w.h.Hijack != nil {
   402  		f = w.h.Hijack(f)
   403  	}
   404  	return f()
   405  }
   406  
   407  func (w *rw) ReadFrom(src io.Reader) (int64, error) {
   408  	f := w.w.(io.ReaderFrom).ReadFrom
   409  	if w.h.ReadFrom != nil {
   410  		f = w.h.ReadFrom(f)
   411  	}
   412  	return f(src)
   413  }
   414  
   415  func (w *rw) Push(target string, opts *http.PushOptions) error {
   416  	f := w.w.(http.Pusher).Push
   417  	if w.h.Push != nil {
   418  		f = w.h.Push(f)
   419  	}
   420  	return f(target, opts)
   421  }
   422  
   423  type Unwrapper interface {
   424  	Unwrap() http.ResponseWriter
   425  }
   426  
   427  // Unwrap returns the underlying http.ResponseWriter from within zero or more
   428  // layers of httpsnoop wrappers.
   429  func Unwrap(w http.ResponseWriter) http.ResponseWriter {
   430  	if rw, ok := w.(Unwrapper); ok {
   431  		// recurse until rw.Unwrap() returns a non-Unwrapper
   432  		return Unwrap(rw.Unwrap())
   433  	} else {
   434  		return w
   435  	}
   436  }
   437
View as plain text