Skip to content

Commit b490092

Browse files
authored
Merge pull request #245 from niksrc/websocket_103
sso-proxy: add websocket support
2 parents 917f880 + 9299caf commit b490092

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

internal/proxy/logging_handler.go

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,10 @@
44
package proxy
55

66
import (
7+
"bufio"
8+
"errors"
79
"io"
10+
"net"
811
"net/http"
912
"net/url"
1013
"strings"
@@ -38,6 +41,14 @@ func (l *responseLogger) extractUser() {
3841
}
3942
}
4043

44+
// Support Websockets
45+
func (l *responseLogger) Hijack() (rwc net.Conn, buf *bufio.ReadWriter, err error) {
46+
if hij, ok := l.w.(http.Hijacker); ok {
47+
return hij.Hijack()
48+
}
49+
return nil, nil, errors.New("http.Hijacker is not available on writer")
50+
}
51+
4152
func (l *responseLogger) Write(b []byte) (int, error) {
4253
if l.status == 0 {
4354
// The status will be StatusOK if WriteHeader has not been called yet

0 commit comments

Comments
 (0)