Koldun/mapserver
1
0
Fork 0
forked from MTSR/mapserver
Code Pull Requests Activity

ws close

Browse Source
This commit is contained in:
NatureFreshMilk 2019-02-07 08:08:54 +01:00
parent 5608be900e
commit 7b5b88a75e
1 changed files with 8 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
server/web/ws.go
View File

@ -64,17 +64,16 @@ func (t *WS) ServeHTTP(resp http.ResponseWriter, req *http.Request) {
t.channels[id] = ch
t.mutex.Unlock()
defer func() {
for {
data := <-ch
err := conn.WriteMessage(websocket.TextMessage, data)
if err != nil {
break
}
}
t.mutex.Lock()
delete(t.channels, id)
close(ch)
t.mutex.Unlock()
}()
for {
data := <-ch
conn.WriteMessage(websocket.TextMessage, data)
}
}