Mirrors/hostyoself
1
0
Fork 0
mirror of https://github.com/schollz/hostyoself.git synced 2026-01-23 02:15:14 +00:00
Code Issues Projects Releases Wiki Activity

return nil from websocket error so no superfluos writes

Browse source
This commit is contained in:
Zack Scholl 2019-07-12 12:18:40 +00:00
parent 79bf4213e4
commit ae6121a085
1 changed files with 7 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

10
pkg/server/server.go
View file

@ -232,7 +232,8 @@ func (s *server) handleWebsocket(w http.ResponseWriter, r *http.Request) (err er
// handle websockets on this page
c, errUpgrade := wsupgrader.Upgrade(w, r, nil)
if errUpgrade != nil {
return errUpgrade
log.Error(errUpgrade)
return nil
}
ws := wsconn.New(c)
@ -250,7 +251,7 @@ func (s *server) handleWebsocket(w http.ResponseWriter, r *http.Request) (err er
err = fmt.Errorf("got wrong type/domain: %s/%s", p.Type, p.Message)
log.Debug(err)
ws.Close()
return
return nil
}
domain := strings.Replace(strings.ToLower(strings.TrimSpace(p.Message)), " ", "-", -1)
@ -276,7 +277,10 @@ func (s *server) handleWebsocket(w http.ResponseWriter, r *http.Request) (err er
Message: domain,
Success: true,
})
return
if err != nil {
log.Error(err)
}
return nil
}
func (s *server) isdomain(domain string) bool {