其他分享
首页 > 其他分享> > Ants线程池的优雅设计

Ants线程池的优雅设计

作者:互联网

先贴上原作者的仓库地址:https://github.com/panjf2000/ants
通俗来讲,线程池的存在就是为了实现高并发情况下的线程复用,从而减少内存开销和提升执行速度。
个人认为ants的设计有两个可学习之处:池调度模型和自旋锁。

池模型

调度流程

  1. 检查当前 Worker 队列中是否有可用的 Worker,如果有,取出执行当前的 task;
  2. 没有可用的 Worker,判断当前在运行的 Worker 是否已超过该 Pool 的容量:{是 —> 再判断工作池是否为非阻塞模式:[是 ——> 直接返回 nil,否 ——> 阻塞等待直至有 Worker 被放回 Pool],否 —> 新开一个 Worker(goroutine)处理};
  3. 每个 Worker 执行完任务之后,放回 Pool 的队列中等待。

自旋锁

定义:一个线程尝试去获取某一把锁的时候,如果这个锁此时已经被别人获取(占用),那么此线程就无法获取到这把锁,该线程将会等待,间隔一段时间后会再次尝试获取。
在ants中实现自旋锁使用二进制指数规避算法,代码如下:

type spinLock uint32

const maxBackoff = 16

func (sl *spinLock) Lock() {
	backoff := 1
	for !atomic.CompareAndSwapUint32((*uint32)(sl), 0, 1) {
		// Leverage the exponential backoff algorithm, see https://en.wikipedia.org/wiki/Exponential_backoff.
		for i := 0; i < backoff; i++ {
			runtime.Gosched()
		}
		if backoff < maxBackoff {
			backoff <<= 1
		}
	}
}

func (sl *spinLock) Unlock() {
	atomic.StoreUint32((*uint32)(sl), 0)
}

// NewSpinLock instantiates a spin-lock.
func NewSpinLock() sync.Locker {
	return new(spinLock)
}

标签:Worker,Ants,backoff,获取,线程,ants,优雅,Pool
来源: https://www.cnblogs.com/maxing486/p/16213378.html