go-packages/iris
1
0
Fork 0
mirror of https://github.com/kataras/iris.git synced 2025-02-02 15:30:36 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Handle non-WS connections in WS handler with no panic

Browse Source 
Former-commit-id: f56665ee444bd16258edf6a34606efc132eced5a
This commit is contained in:
Alexey McSakoff 2018-03-10 19:31:40 +04:00
parent c746d631d7
commit 6c9e3c69ff
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
websocket/server.go
View File

@ -147,6 +147,9 @@ func New(cfg Config) *Server {
func (s *Server) Handler() context.Handler { func (s *Server) Handler() context.Handler {
return func(ctx context.Context) { return func(ctx context.Context) {
c := s.Upgrade(ctx) c := s.Upgrade(ctx)
if c.Err() != nil {
return
}
// NOTE TO ME: fire these first BEFORE startReader and startPinger // NOTE TO ME: fire these first BEFORE startReader and startPinger
// in order to set the events and any messages to send // in order to set the events and any messages to send
// the startPinger will send the OK to the client and only // the startPinger will send the OK to the client and only