pool.go

// pool.go
package main

import (
	"log"
)

// Job 表示一个需要被执行的任务
type Job struct {
	Task Task
}

// WorkerPool 维护一个任务通道和工作协程池
type WorkerPool struct {
	JobQueue chan Job
	quit     chan bool
}

// NewWorkerPool 创建一个新的工作池
func NewWorkerPool(maxWorkers int) *WorkerPool {
	pool := &WorkerPool{
		JobQueue: make(chan Job),
		quit:     make(chan bool),
	}

	for i := 0; i < maxWorkers; i++ {
		go func(workerID int) {
			for {
				select {
				case job := <-pool.JobQueue:
					log.Printf("Worker %d: started job %s\n", workerID, job.Task.SourceID)
					ProcessTask(&job.Task)
					log.Printf("Worker %d: finished job %s\n", workerID, job.Task.SourceID)
				case <-pool.quit:
					log.Printf("Worker %d: stopping\n", workerID)
					return
				}
			}
		}(i + 1)
	}
	return pool
}

// Submit 提交一个新任务到工作池
func (p *WorkerPool) Submit(job Job) {
	p.JobQueue <- job
}

// Stop 停止工作池
func (p *WorkerPool) Stop() {
	go func() {
		close(p.quit)
	}()
}