前言
早在几年前写过关于 LRU cache
的文章:
https://crossoverjie.top/2018/04/07/algorithm/LRU-cache/
当时是用 Java 实现的,最近我在完善 ptg 时正好需要一个最近最少使用的数据结构来存储历史记录。
ptg: Performance testing tool (Go), 用 Go 实现的 gRPC 客户端调试工具。
Go 官方库中并没有相关的实现,考虑到程序的简洁就不打算依赖第三方库,自己写一个;本身复杂度也不高,没有几行代码。
配合这个数据结构,我便在 ptg 中实现了请求历史记录的功能:
将每次的请求记录存储到 lru cache 中,最近使用到的历史记录排在靠前,同时也能提供相关的搜索功能;具体可见下图。
实现
实现原理没什么好说的,和 Java
的一样:
- 一个双向链表存储数据的顺序
- 一个
map
存储最终的数据 - 当数据达到上限时移除链表尾部数据
- 将使用到的
Node
移动到链表的头结点
虽然 Go 比较简洁,但好消息是基本的双向链表结构还是具备的。
所以基于此便定义了一个 LruCache
:
根据之前的分析:
size
存储缓存大小。- 链表存储数据顺序。
map
存储数据。lock
用于控制并发安全。
接下来重点是两个函数:写入、查询。
写入时判断是否达到容量上限,达到后删除尾部数据;否则就想数据写入头部。
而获取数据时,这会将查询到的结点移动到头结点。
这些结点操作都由 List 封装好了的。
所以使用起来也比较方便。
最终就是通过这个 LruCache
实现了上图的效果,想要了解更多细节的可以参考源码: