原来sync.Once还能这么用
介绍
sync.Once
估计大家都不陌生,官方介绍中,
Once is an object that will perform exactly one action
正是因为这个特性,Once
常常被用于单例对象的初始化场景。
也正是因为这个特性,其实它还能做一些其他的事情。
缓存击穿
日常背诵八股文,我相信你们对缓存击穿这个词特别熟悉。
缓存击穿一般代指热点key
缓存失效(到期|删了),同一时刻大量对热点key
的并发请求。缓存找不到数据,所有请求都打入到DB
层。此时,身为开发的你,明天和意外就不知道哪个先到了。
为了防止这种情况发生,针对相同key
的请求,只需要一个请求(A)到达DB
层取数据,其他请求等待A
通知就行了。
就像这样,
图片来源:[1]
singleflight
Go
里有很多防缓存击穿的工具,比如singleflight
库。
通过上面简单的代码大概能看出,其实就是对key
做了缓存。
把一个key
对应call
结构存储在map
中。保证只有一个key
真正执行fn()
服务 ,其他请求则通过sync.waitGroup
的wait
等待结果。
至于g.docall(c,key,fn)
,
当带着全村人希望的那个请求,获取到数据,给对应key
的call
赋值,最终执行done
,通知等待这个key
全村的村民获取数据。
代码并不复杂。
自定义singleflight
我们也可以实现一个简易版本的。
代码整体不难,主要的点在于我们是通过通道来实现通知自家兄弟取数据。
最后,让我们使用Once
来达到同样的效果,不然标题不白起了嘛。
上面核心代码都写出来了,实际开发中需要对请求资源做一些超时控制等操作。
总结
平常对Once
的使用只停留在初始化工作上,而弱化了它的使用场景。对于其他工具也是一个道理,这就需要去积累和挖掘了。
附录
[1]https://medium.com/codex/caching-system-stability-766bf5fff69f
https://blog.chuie.io/posts/synconce/
推荐阅读