pools: Add a pool for buffers
This commit is contained in:
parent
a63bb2f669
commit
598de5b289
6 changed files with 56 additions and 28 deletions
|
|
@ -1,12 +1,12 @@
|
|||
package environment
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"io"
|
||||
"io/fs"
|
||||
"os"
|
||||
|
||||
"github.com/FAU-CDI/wisski-distillery/pkg/pools"
|
||||
"github.com/tkw1536/goprogram/stream"
|
||||
)
|
||||
|
||||
|
|
@ -60,8 +60,10 @@ func ReadFile(env Environment, path string) ([]byte, error) {
|
|||
defer file.Close()
|
||||
|
||||
// copy everything into a buffer!
|
||||
var buffer bytes.Buffer
|
||||
if _, err := io.Copy(&buffer, file); err != nil {
|
||||
buffer := pools.GetBuffer()
|
||||
defer pools.ReleaseBuffer(buffer)
|
||||
|
||||
if _, err := io.Copy(buffer, file); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
|
|
|
|||
|
|
@ -1,6 +1,8 @@
|
|||
// Package pools holds various pools for reuse
|
||||
package pools
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"strings"
|
||||
"sync"
|
||||
)
|
||||
|
|
@ -17,3 +19,16 @@ func ReleaseBuilder(builder *strings.Builder) {
|
|||
builder.Reset()
|
||||
builders.Put(builder)
|
||||
}
|
||||
|
||||
var buffers = sync.Pool{
|
||||
New: func() any { return new(bytes.Buffer) },
|
||||
}
|
||||
|
||||
func GetBuffer() *bytes.Buffer {
|
||||
return buffers.Get().(*bytes.Buffer)
|
||||
}
|
||||
|
||||
func ReleaseBuffer(buffer *bytes.Buffer) {
|
||||
buffer.Reset()
|
||||
buffers.Put(buffer)
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue