feat(mux): implement hijacker interface in response writer (#74)

.gitignore

@@ -19,6 +19,9 @@
 !.vscode/extensions.json
 *.code-workspace
 
+# Jetbrains
+.idea
+
 # Vim
 # Swap
 [._]*.s[a-v][a-z]

CHANGELOG.md

@@ -7,6 +7,10 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## Unreleased
 
+### Added
+
+- introduce support for Hijacker interface in **mux** middleware response writer
+
 ## v4.1.0 - 16-10-2023
 
 ### Added

middleware/mux/readableresponsewriter.go

@@ -17,6 +17,9 @@
 package mux
 
 import (
+	"bufio"
+	"fmt"
+	"net"
 	"net/http"
 )
 
@@ -60,3 +63,12 @@ func (r *readableResponseWriter) Flush() {
 		flusherWriter.Flush()
 	}
 }
+
+// Hijack func, calls the underlying ResponseWriter Hijack fn
+func (r *readableResponseWriter) Hijack() (net.Conn, *bufio.ReadWriter, error) {
+	if hijacker, ok := r.Writer.(http.Hijacker); ok {
+		return hijacker.Hijack()
+	}
+
+	return nil, nil, fmt.Errorf("the Hijacker interface is not supported")
+}

middleware/mux/readableresponsewriter_test.go

@@ -17,6 +17,8 @@
 package mux
 
 import (
+	"bufio"
+	"net"
 	"net/http"
 	"testing"
 )
@@ -26,6 +28,7 @@ type ResponseWriterMock struct {
 	writeHeaderCalled bool
 	writeCalled       bool
 	flushCalled       bool
+	hijackCalled      bool
 }
 
 func (r *ResponseWriterMock) Header() http.Header {
@@ -46,6 +49,12 @@ func (r *ResponseWriterMock) Flush() {
 	r.flushCalled = true
 }
 
+func (r *ResponseWriterMock) Hijack() (net.Conn, *bufio.ReadWriter, error) {
+	r.hijackCalled = true
+
+	return nil, nil, nil
+}
+
 func TestReadableResponseWriter(t *testing.T) {
 	mock := ResponseWriterMock{}
 	myw := readableResponseWriter{Writer: &mock}
@@ -73,4 +82,9 @@ func TestReadableResponseWriter(t *testing.T) {
 	if !mock.flushCalled {
 		t.Errorf("mock flush not called")
 	}
+
+	_, _, err = myw.Hijack()
+	if !mock.hijackCalled || err != nil {
+		t.Errorf("mock hijack not called successfully")
+	}
 }