fix a possible race condition

xtaci · xtaci · commit 3b54c0fc00fa · 2019-12-19T16:18:34.000+08:00
diff --git a/stream.go b/stream.go
@@ -103,8 +103,8 @@ func (s *Stream) WriteTo(w io.Writer) (n int64, err error) {
 
 		if buf != nil {
 			nw, ew := w.Write(buf)
-			defaultAllocator.Put(buf)
 			s.sess.returnTokens(len(buf))
+			defaultAllocator.Put(buf)
 			if nw > 0 {
 				n += int64(nw)
 			}

Original file line number	Diff line number	Diff line change
`@@ -103,8 +103,8 @@ func (s *Stream) WriteTo(w io.Writer) (n int64, err error) {`
`103`	`103`
`104`	`104`	`if buf != nil {`
`105`	`105`	`nw, ew := w.Write(buf)`
`106`		`- defaultAllocator.Put(buf)`
`107`	`106`	`s.sess.returnTokens(len(buf))`
	`107`	`+ defaultAllocator.Put(buf)`
`108`	`108`	`if nw > 0 {`
`109`	`109`	`n += int64(nw)`
`110`	`110`	`}`