drip/internal/shared/pool/header_pool.go

package pool

import (
	"net/http"
	"sync"
)

// HeaderPool manages a pool of http.Header objects for reuse
// This reduces GC pressure from repeated header map allocations
type HeaderPool struct {
	pool sync.Pool
}

// NewHeaderPool creates a new header pool
func NewHeaderPool() *HeaderPool {
	return &HeaderPool{
		pool: sync.Pool{
			New: func() interface{} {
				// Pre-allocate with capacity for common header count (8-12 headers)
				return make(http.Header, 12)
			},
		},
	}
}

// Get retrieves a header from the pool
// Returns a clean, empty header ready for use
func (p *HeaderPool) Get() http.Header {
	h := p.pool.Get().(http.Header)
	// Clear any existing data (headers might be dirty from previous use)
	for k := range h {
		delete(h, k)
	}
	return h
}

// Put returns a header to the pool
// The header will be reused by future Get() calls
func (p *HeaderPool) Put(h http.Header) {
	if h == nil {
		return
	}
	// Note: We don't clear here, clearing is done in Get() for better performance
	// (allows the GC to collect during idle time)
	p.pool.Put(h)
}

// Clone creates a copy of src into dst, reusing dst's underlying storage
// This is more efficient than creating a new header from scratch
func (p *HeaderPool) Clone(dst, src http.Header) {
	// Clear dst first
	for k := range dst {
		delete(dst, k)
	}

	// Copy all headers from src to dst
	for k, vv := range src {
		// Allocate new slice with exact capacity to avoid over-allocation
		dst[k] = make([]string, len(vv))
		copy(dst[k], vv)
	}
}

// CloneWithExtra clones src into dst and adds/overwrites extra headers
// This is optimized for the common pattern of cloning + adding Host header
func (p *HeaderPool) CloneWithExtra(dst, src http.Header, extraKey, extraValue string) {
	// Clear dst first
	for k := range dst {
		delete(dst, k)
	}

	// Copy all headers from src to dst
	for k, vv := range src {
		dst[k] = make([]string, len(vv))
		copy(dst[k], vv)
	}

	// Set extra header (overwrite if exists)
	dst.Set(extraKey, extraValue)
}

// globalHeaderPool is a package-level pool for convenience
var globalHeaderPool = NewHeaderPool()

// GetHeader retrieves a header from the global pool
func GetHeader() http.Header {
	return globalHeaderPool.Get()
}

// PutHeader returns a header to the global pool
func PutHeader(h http.Header) {
	globalHeaderPool.Put(h)
}