1f349/violet
4
0
Fork 0
mirror of https://github.com/1f349/violet.git synced 2024-11-23 11:51:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix up this lock

Browse Source
This commit is contained in:
Melon 2023-08-17 14:57:41 +01:00
parent 1f72795f22
commit fbbcf0440c
Signed by: melon
GPG Key ID: 6C9D970C50D26A25
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
proxy/websocket/server.go
View File

@ -40,16 +40,17 @@ func (s *Server) Upgrade(rw http.ResponseWriter, req *http.Request) {
return return
} }
s.connLock.Lock() s.connLock.Lock()
defer s.connLock.Unlock()
// no more connections allowed // no more connections allowed
if s.connStop { if s.connStop {
s.connLock.Unlock()
_ = c.Close() _ = c.Close()
return return
} }
// save connection for shutdown // save connection for shutdown
s.conns[c.RemoteAddr().String()] = c s.conns[c.RemoteAddr().String()] = c
s.connLock.Unlock()
log.Printf("[Websocket] Dialing: '%s'\n", req.URL.String()) log.Printf("[Websocket] Dialing: '%s'\n", req.URL.String())