opensource/grpc-go
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Adds mutex lock on setting goaway flag

Browse Source
This commit is contained in:
Gian Biondi
2015-04-29 15:13:31 -04:00
parent 4067e7affb
commit dbe74dd6c5
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
transport/http2_server.go
View File

@ -146,7 +146,7 @@ func (t *http2Server) operateHeaders(hDec *hpackDecoder, s *Stream, frame header
hDec.state = decodeState{} hDec.state = decodeState{}
} }
}() }()
if goaway { if t.goaway {
//Stop creating streams on this transport //Stop creating streams on this transport
return nil return nil
} }
@ -410,7 +410,9 @@ func (t *http2Server) handleWindowUpdate(f *http2.WindowUpdateFrame) {
} }
func (t *http2Server) handleGoAway(f *http2.GoAwayFrame) { func (t *http2Server) handleGoAway(f *http2.GoAwayFrame) {
t.mu.Lock()
t.goaway = true t.goaway = true
t.mu.Unlock()
for id, _ := range t.activeStreams { for id, _ := range t.activeStreams {
if id > f.LastStreamID { if id > f.LastStreamID {
if s, ok := t.activeStreams[f.LastStreamID]; ok { if s, ok := t.activeStreams[f.LastStreamID]; ok {