Go语言原生不支持goroutine优先级调度，需在应用层通过container/heap实现优先级队列。定义Task结构体包含Priority字段，数值越小优先级越高，利用最大堆或最小堆维护任务顺序，配合工作协程按优先级取出并执行任务，从而实现高优先级任务先执行的调度机制。
在Go语言中，原生的goroutine调度器并不直接支持任务优先级的概念。所有goroutine由调度器统一管理，按协作式调度机制运行，无法像操作系统线程那样设置高、低优先级执行。但在实际开发中，我们常常需要实现带优先级的任务调度，比如高优先级的消息通知要早于日志写入处理。这时候就需要在应用层设计优先级队列与调度逻辑。
使用带优先级的队列实现任务调度
最常见的方式是结合优先级队列和一个或多个工作协程（worker goroutines），通过比较任务的优先级来决定执行顺序。
可以使用最小堆或最大堆结构来维护任务队列。Go标准库container/heap提供了堆的操作接口，我们可以基于它构建一个优先级任务队列。
示例：定义带优先级的任务
type Task struct {
Priority int // 数值越小，优先级越高
Data string
Fn func()
}
// 任务队列，实现 heap.Interface
type PriorityQueue []*Task
func (pq PriorityQueue) Len() int { return len(pq) }
func (pq PriorityQueue) Less(i, j int) bool {
return pq[i].Priority < pq[j].Priority // 小顶堆：优先级数值小的排前面
}
func (pq PriorityQueue) Swap(i, j int) {
pq[i], pq[j] = pq[j], pq[i]
}
func (pq *PriorityQueue) Push(x interface{}) {
task := x.(*Task)
*pq = append(*pq, task)
}
func (pq *PriorityQueue) Pop() interface{} {
old := *pq
n := len(old)
task := old[n-1]
*pq = old[0 : n-1]
return task
}
调度器控制任务分发
有了优先级队列后，需要一个调度器来安全地向队列中添加任务，并由工作协程取出执行。
立即学习“go语言免费学习笔记（深入）”；
通常使用互斥锁 + 条件变量（sync.Mutex 和 sync.Cond）或channel来协调生产者和消费者。
示例：启动调度器与工作协程
func NewScheduler() *Scheduler {
pq := make(PriorityQueue, 0)
heap.Init(&pq)
return &Scheduler{
tasks: &pq,
mu: new(sync.Mutex),
cond: sync.NewCond(&sync.Mutex{}),
}
}
type Scheduler struct {
tasks *PriorityQueue
mu *sync.Mutex
cond *sync.Cond
}
func (s *Scheduler) Submit(priority int, f func()) {
s.mu.Lock()
heap.Push(s.tasks, &Task{Priority: priority, Fn: f})
s.mu.Unlock()
s.cond.Signal() // 唤醒等待的工作协程
}
func (s *Scheduler) Run() {
for {
s.cond.L.Lock()
for s.tasks.Len() == 0 {
s.cond.Wait() // 队列为空时等待
}
s.mu.Lock()
task := heap.Pop(s.tasks).(*Task)
s.mu.Unlock()
s.cond.L.Unlock()
// 执行任务
task.Fn()
}
}
实际使用示例
启动调度器并提交不同优先级的任务：
func main() {
scheduler := NewScheduler()
// 启动调度器
go scheduler.Run()
// 提交任务
scheduler.Submit(2, func() {
fmt.Println("低优先级任务")
})
scheduler.Submit(1, func() {
fmt.Println("高优先级任务")
})
scheduler.Submit(3, func() {
fmt.Println("最低优先级任务")
})
// 简单延时让任务执行
time.Sleep(time.Second)
}
输出结果会优先执行Priority=1的任务，然后是2，最后是3。
进阶建议
- 可扩展支持任务超时、取消（结合context.Context）
- 多个优先级等级可对应多个channel（如highCh, normalCh, lowCh），用select配合default做非阻塞降级消费
- 在高并发场景下注意锁竞争，可考虑分片队列或无锁队列优化
- 若对延迟敏感，可限制队列长度，防止堆积
基本上就这些。Go虽然没有内置优先级调度，但通过组合堆、goroutine和同步原语，完全可以实现灵活高效的优先级任务系统。关键是理解优先级判定逻辑与调度公平性之间的平衡。不复杂但容易忽略细节。
