「Here's how CockroachDB keeps your database from collapsing under load」

其实是一篇关于 Golang Goroutine Scheduler 的文章。众所周知，Go 的 Goroutines 并没有优先级，对于类似于数据库这样的应用，如何在高并发的情况下保证重要任务的执行就会面临挑战。这篇文章就是介绍了 CockroachDB 的做法，通过在原生的 Go Scheduler 外增加一层名为 Admission Control 的调度层，来实现了 Goroutines 的优先级调度。

核心思想就是不要再将大量的 Goroutines 直接交给 Go Scheduler 去排队。而是先在应用层的 Admission Control 内的优先级队列里排队，只将当前系统能够负载的高优先级任务交给 Go Scheduler 去执行，这样就可以保证在高负载下系统仍然能保证重要任务的可用性。

具体的优先级实现有很多很巧妙的设计。比如优先级系统区分为 Token Based 和 Slot Based。对于 CPU 型任务，采用令牌桶算法，根据可用资源的情况来生成令牌，并且分发给任务使用，如果有任务超额使用了令牌，则扣减总的令牌数。而对于 I/O 型任务，因为 I/O 操作会阻塞 Goroutine，所以采用了 Slot Based 的方式，根据当前系统的 I/O 负载情况来动态调整 slot 的数量。而且动态调整的时间间隔也是动态变化的，尽量减少 CPU 消耗。

还观察到，即使使用这个优先级系统提交任务，偶尔也还是会导致 Go Scheduler 过载。而优化方法是将调度下一个任务的逻辑放到 task 的执行逻辑中，这样就只有在 task 被 Go Scheduler 实际交给 P & M 运行时，才会进行调度下一个任务的逻辑，从而确保了 task 不会在 Go Scheduler 堆积。

Refs:

* Source Code: cockroach Admit
* Go scheduler: Ms, Ps & Gs