Function NewWorkerPool

internal/proxy/pool.go:33–59

NewWorkerPool creates a sharded worker pool. workers: number of workers/shards (0 = NumCPU * 2) queueSize: total queue capacity across all shards (0 = 10000)

(workers, queueSize int, handler func(*net.UDPAddr, []byte))

// workers: number of workers/shards (0 = NumCPU * 2)
// queueSize: total queue capacity across all shards (0 = 10000)
func NewWorkerPool(workers, queueSize int, handler func(*net.UDPAddr, []byte)) *WorkerPool {
	if workers <= 0 {
		workers = runtime.NumCPU() * 2
	}
	if queueSize <= 0 {
		queueSize = 10000
	}

	queuePerShard := queueSize / workers
	if queuePerShard < 100 {
		queuePerShard = 100
	}

	p := &WorkerPool{
		queues:        make([]chan WorkItem, workers),
		handler:       handler,
		workers:       workers,
		queuePerShard: queuePerShard,
		dropped:       make([]uint64, workers),
	}

	for i := 0; i < workers; i++ {
		p.queues[i] = make(chan WorkItem, queuePerShard)
	}

	return p
}

// Start launches all worker goroutines.
func (p *WorkerPool) Start() {