starred/godoxy-yusing
1
0
Fork 0
mirror of https://github.com/yusing/godoxy.git synced 2026-03-28 03:51:08 +01:00
Code Issues 18 Packages Projects Releases 10 Wiki Activity

refactor(websocket): simplify NewManagerWithUpgrade function to use context for custom upgrader

Browse Source
This commit is contained in:
yusing
2025-08-17 20:23:57 +08:00
parent beb1913285
commit 1c7e3e42f8
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
internal/net/gphttp/websocket/manager.go
View File

@@ -73,12 +73,12 @@ const (
// NewManagerWithUpgrade upgrades the HTTP connection to a WebSocket connection and returns a Manager.
// If the upgrade fails, the error is returned.
// If the upgrade succeeds, the Manager is returned.
func NewManagerWithUpgrade(c *gin.Context, upgrader ...websocket.Upgrader) (*Manager, error) {
var actualUpgrader websocket.Upgrader
if len(upgrader) == 0 {
actualUpgrader = defaultUpgrader
} else {
actualUpgrader = upgrader[0]
//
// To use a custom upgrader, set the "upgrader" context value to the upgrader.
func NewManagerWithUpgrade(c *gin.Context) (*Manager, error) {
actualUpgrader := &defaultUpgrader
if upgrader, ok := c.Get("upgrader"); ok {
actualUpgrader = upgrader.(*websocket.Upgrader)
}
conn, err := actualUpgrader.Upgrade(c.Writer, c.Request, nil)