fix: Write panic when conn Close (cloudwego#96)

NeVers1EEp · Dec 14, 2021 · 1344445 · 1344445
1 parent 985eb36
commit 1344445
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/connection_impl.go b/connection_impl.go
@@ -242,9 +242,15 @@ func (c *connection) Read(p []byte) (n int, err error) {
 
 // Write will Flush soon.
 func (c *connection) Write(p []byte) (n int, err error) {
+	if !c.lock(flushing) {
+		return 0, Exception(ErrConnClosed, "when write")
+	}
+	defer c.unlock(flushing)
+
 	dst, _ := c.outputBuffer.Malloc(len(p))
 	n = copy(dst, p)
-	err = c.Flush()
+	c.outputBuffer.Flush()
+	err = c.flush()
 	return n, err
 }