Add function to upgrade connection in http router

AmanAgnihotri · Apr 12, 2024 · 03a99b1 · 03a99b1
1 parent c3c57ce
commit 03a99b1
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pkg/infra/web/http/router.go b/pkg/infra/web/http/router.go
@@ -3,7 +3,12 @@
 
 package http
 
+import (
+	"github.com/lesismal/nbio/nbhttp/websocket"
+)
+
 type Router interface {
 	Handle(pattern string, handler func(ctx *Context))
 	HandleWithAuth(pattern string, handler func(ctx *Context))
+	Upgrade(ctx *Context) (*websocket.Conn, error)
 }