rogerferdinan/safe-web-socket
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: moving writeFunc as culprit of memory leak

Browse Source
This commit is contained in:
Roger Ferdinan
2026-02-08 20:50:55 +07:00
parent b296d73367
commit 632d79dd57
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
v1/server/server.go
View File

@@ -64,10 +64,12 @@ func (b *SafeWebsocketServerBuilder) HandleFunc(pattern string, fn func(http.Res
return b return b
} }
func (b *SafeWebsocketServerBuilder) HandleFuncWebsocket(pattern string, subscribedPath string, writeFunc func(chan []byte)) *SafeWebsocketServerBuilder { func (b *SafeWebsocketServerBuilder) HandleFuncWebsocket(pattern string, subscribedPath string, writeFunc func(writeChannel chan []byte)) *SafeWebsocketServerBuilder {
h := internal.NewHub() h := internal.NewHub()
h.Run() h.Run()
go writeFunc(h.Broadcast)
b.mux.HandleFunc(pattern+subscribedPath, func(w http.ResponseWriter, r *http.Request) { b.mux.HandleFunc(pattern+subscribedPath, func(w http.ResponseWriter, r *http.Request) {
conn, err := b.upgrader.Upgrade(w, r, nil) conn, err := b.upgrader.Upgrade(w, r, nil)
if err != nil { if err != nil {
@@ -85,7 +87,7 @@ func (b *SafeWebsocketServerBuilder) HandleFuncWebsocket(pattern string, subscri
go internal.WritePump(c, h) go internal.WritePump(c, h)
go internal.ReadPump(c, h) go internal.ReadPump(c, h)
go writeFunc(h.Broadcast)
}) })
return b return b
} }