Use a buffer pool when proxying

Improves performance by reducing allocations.

internal/server/proxy_buffer_pool.go

@@ -0,0 +1,25 @@
+package server
+
+import "sync"
+
+func NewBufferPool(bufferSize int64) *BufferPool {
+ return &BufferPool{
+ pool: sync.Pool{
+ New: func() interface{} {
+ return make([]byte, bufferSize)
+ },
+ },
+ }
+}
+
+type BufferPool struct {
+ pool sync.Pool
+}
+
+func (b *BufferPool) Get() []byte {
+ return b.pool.Get().([]byte)
+}
+
+func (b *BufferPool) Put(content []byte) {
+ b.pool.Put(content)
+}

internal/server/service.go

@@ -32,6 +32,7 @@ const (
  DefaultHealthCheckTimeout = time.Second * 5
 
  MaxIdleConnsPerHost = 100
+ ProxyBufferSize = 32 * KB
 
  DefaultTargetTimeout = time.Second * 10
  DefaultMaxRequestMemoryBufferSize = 1 * MB

internal/server/target.go

@@ -251,7 +251,10 @@ func (t *Target) recordTargetNameForRequest(req *http.Request) {
 }
 
 func (t *Target) createProxyHandler() http.Handler {
+ bufferPool := NewBufferPool(ProxyBufferSize)
+
  return &httputil.ReverseProxy{
+ BufferPool: bufferPool,
  Rewrite: t.Rewrite,
  ErrorHandler: t.handleProxyError,
  Transport: &http.Transport{