d71537d431
将流式响应中 bufio.Scanner 的 64KB buffer 从每次 make 分配改为 sync.Pool 复用，统一切片表达式为 [:0]、变量命名为 scanBuf， 并补充对应的单元测试。 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
25 lines
434 B
Go
25 lines
434 B
Go
package service
|
|
|
|
import "sync"
|
|
|
|
const sseScannerBuf64KSize = 64 * 1024
|
|
|
|
type sseScannerBuf64K [sseScannerBuf64KSize]byte
|
|
|
|
var sseScannerBuf64KPool = sync.Pool{
|
|
New: func() any {
|
|
return new(sseScannerBuf64K)
|
|
},
|
|
}
|
|
|
|
func getSSEScannerBuf64K() *sseScannerBuf64K {
|
|
return sseScannerBuf64KPool.Get().(*sseScannerBuf64K)
|
|
}
|
|
|
|
func putSSEScannerBuf64K(buf *sseScannerBuf64K) {
|
|
if buf == nil {
|
|
return
|
|
}
|
|
sseScannerBuf64KPool.Put(buf)
|
|
}
|