fix a possible race condition

xtaci · Dec 19, 2019 · 3b54c0f · 3b54c0f
1 parent 3627305
commit 3b54c0f
Showing 1 changed file with 1 addition and 1 deletion.
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)
 			}