fix: header middleware response writer.
This commit is contained in:
parent
2d1a973ee5
commit
eb7a6d925b
1 changed files with 19 additions and 0 deletions
|
@ -1,6 +1,9 @@
|
||||||
package headers
|
package headers
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"bufio"
|
||||||
|
"fmt"
|
||||||
|
"net"
|
||||||
"net/http"
|
"net/http"
|
||||||
|
|
||||||
"github.com/containous/traefik/v2/pkg/log"
|
"github.com/containous/traefik/v2/pkg/log"
|
||||||
|
@ -73,3 +76,19 @@ func (w *responseModifier) Write(b []byte) (int, error) {
|
||||||
|
|
||||||
return w.w.Write(b)
|
return w.w.Write(b)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Hijack hijacks the connection.
|
||||||
|
func (w *responseModifier) Hijack() (net.Conn, *bufio.ReadWriter, error) {
|
||||||
|
if h, ok := w.w.(http.Hijacker); ok {
|
||||||
|
return h.Hijack()
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil, nil, fmt.Errorf("not a hijacker: %T", w.w)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Flush sends any buffered data to the client.
|
||||||
|
func (w *responseModifier) Flush() {
|
||||||
|
if flusher, ok := w.w.(http.Flusher); ok {
|
||||||
|
flusher.Flush()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in a new issue