在Go语言中,可以使用select和channel来实现任务调度。下面是一个示例代码,演示如何使用select和channel来实现任务调度:
package main
import (
"fmt"
"time"
)
func worker(id int, jobs <-chan int, results chan<- int) {
for job := range jobs {
fmt.Printf("Worker %d started job %d\n", id, job)
time.Sleep(time.Second) // 模拟任务执行时间
fmt.Printf("Worker %d finished job %d\n", id, job)
results <- job * 2 // 将任务结果发送到结果通道
}
}
func main() {
const numJobs = 10
jobs := make(chan int, numJobs)
results := make(chan int, numJobs)
// 启动多个worker goroutine
for w := 1; w <= 3; w++ {
go worker(w, jobs, results)
}
// 分发任务到任务通道
for j := 1; j <= numJobs; j++ {
jobs <- j
}
close(jobs)
// 收集任务结果
for r := 1; r <= numJobs; r++ {
result := <-results
fmt.Println("Result:", result)
}
}
在上述示例中,我们创建了两个通道jobs和results,jobs通道用于传递任务,results通道用于传递任务结果。我们启动了3个worker goroutine来处理任务。在主函数中,我们向jobs通道发送了10个任务,然后关闭该通道。接着,我们使用for循环从results通道中读取任务结果,并打印出来。
在worker函数中,使用range循环从jobs通道中接收任务,当jobs通道被关闭后,循环会自动退出。然后,我们模拟了任务的执行时间,将任务结果乘以2,并将结果发送到results通道中。
通过使用select和channel,我们可以实现并发式的任务调度,将任务分发给多个worker goroutine并收集任务结果。