fix: incomplete data caused by buffer pool

krau · Nov 30, 2024 · 348c0a3 · 348c0a3
1 parent 2d2008f
commit 348c0a3
Showing 1 changed file with 1 addition and 9 deletions.
diff --git a/common/image.go b/common/image.go
@@ -275,15 +275,7 @@ func CompressImageForTelegramByFFmpegFromBytes(input []byte, maxDepth uint, extr
 	}
 	vfKwArg := ffmpeg.KwArgs{"vf": fmt.Sprintf("scale=%d:%d:flags=lanczos", newWidth, newHeight)}
 
-	buf := imageBufferPool.Get().(*bytes.Buffer)
-	buf.Reset()
-	defer func() {
-		buf.Reset()
-		imageBufferPool.Put(buf)
-	}()
-	if buf.Cap() < inputLen {
-		buf.Grow(inputLen - buf.Cap())
-	}
+	buf := bytes.NewBuffer(nil)
 
 	err = ffmpeg.Input("pipe:").Output("pipe:", vfKwArg, ffmpeg.KwArgs{"format": "mjpeg"}, ffmpeg.MergeKwArgs(extraFFmpegKwArgs)).WithInput(bytes.NewReader(input)).WithOutput(buf).Run()