Robot
/
croc
mirror of https://github.com/schollz/croc.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

relay log more

This commit is contained in:
Zack Scholl 2018-09-22 09:48:22 -07:00
parent 73791a4dc7
commit 1acbe9c475
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/relay/conn.go
View File

@ -81,14 +81,19 @@ func (s *subscription) writePump() {
select {
case message, ok := <-c.send:
if !ok {
c.write(websocket.CloseMessage, []byte{})
err := c.write(websocket.CloseMessage, []byte{})
if err != nil {
log.Debug(err)
}
return
}
if err := c.write(message.messageType, message.data); err != nil {
log.Debug(err)
return
}
case <-ticker.C:
if err := c.write(websocket.PingMessage, []byte{}); err != nil {
log.Debug(err)
return
}
}
@ -99,7 +104,7 @@ func (s *subscription) writePump() {
func serveWs(w http.ResponseWriter, r *http.Request) {
ws, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Error(err)
log.Debug(err)
return
}
vals := r.URL.Query()