commons_go/websocket_fasthttp
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

ing

Browse Source
This commit is contained in:
crusader 2017-11-08 19:17:56 +09:00
parent b630f1721c
commit 64216d22c4
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
server.go
View File

@ -135,7 +135,7 @@ func (s *server) handleRequest(ctx *fasthttp.RequestCtx) {
soc := newSocket(conn, socketHandler) soc := newSocket(conn, socketHandler)
s.stopWg.Add(1) s.stopWg.Add(1)
handleConnection(s, soc) handleConnection(s, soc, socketHandler)
}) })
} }
@ -147,14 +147,14 @@ func (s *server) handleError(ctx *fasthttp.RequestCtx, status int, reason error)
s.sh.OnError(ctx, status, reason) s.sh.OnError(ctx, status, reason)
} }
func handleConnection(s *server, soc *Socket) { func handleConnection(s *server, soc *Socket, socketHandler SocketHandler) {
defer s.stopWg.Done() defer s.stopWg.Done()
logging.Logger.Debug(fmt.Sprintf("Server: Client[%s] is connected.", soc.RemoteAddr())) logging.Logger.Debug(fmt.Sprintf("Server: Client[%s] is connected.", soc.RemoteAddr()))
clientStopChan := make(chan struct{}) clientStopChan := make(chan struct{})
handleDoneChan := make(chan struct{}) handleDoneChan := make(chan struct{})
go soc.sh.Handle(soc, clientStopChan, handleDoneChan) go socketHandler.Handle(soc, clientStopChan, handleDoneChan)
select { select {
case <-s.stopChan: case <-s.stopChan: