implement required interfaces for webtransport in log middleware

centrifugal · May 2, 2024 · 87bca11 · 87bca11
1 parent 8d3b489
commit 87bca11
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/internal/middleware/log.go b/internal/middleware/log.go
@@ -7,6 +7,7 @@ import (
 	"net/http"
 	"time"
 
+	"github.com/quic-go/quic-go/http3"
 	"github.com/rs/zerolog/log"
 )
 
@@ -61,10 +62,15 @@ func (lrw *logResponseWriter) Flush() {
 	lrw.ResponseWriter.(http.Flusher).Flush()
 }
 
-// // StreamCreator for WebTransport.
-// func (lrw *logResponseWriter) StreamCreator() http3.StreamCreator {
-// 	return lrw.ResponseWriter.(http3.Hijacker).StreamCreator()
-// }
+// Connection for WebTransport.
+func (lrw *logResponseWriter) Connection() http3.Connection {
+	return lrw.ResponseWriter.(http3.Hijacker).Connection()
+}
+
+// HTTPStream for WebTransport.
+func (lrw *logResponseWriter) HTTPStream() http3.Stream {
+	return lrw.ResponseWriter.(http3.HTTPStreamer).HTTPStream()
+}
 
 // CloseNotify as SockJS uses http.CloseNotifier.
 //