Avoid repeated closures

lxzan · Mar 22, 2024 · 88589b3 · 88589b3
1 parent e59de96
commit 88589b3
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/internal/pool.go b/internal/pool.go
@@ -40,8 +40,10 @@ func (c *CloserWrapper) Read(p []byte) (n int, err error) {
 }
 
 func (c *CloserWrapper) Close() error {
-	c.B.Reset()
-	bytebufferpool.Put(c.B)
-	c.B, c.R = nil, nil
+	// 避免重复关闭, 引发panic
+	if c.B != nil {
+		bytebufferpool.Put(c.B)
+		c.B, c.R = nil, nil
+	}
 	return nil
 }