一个优雅的 LRU 缓存实现


学无止境

Golang 的各种组件很灵活也很强大,但对于初级入门的使用者来说,要用好着实不易。最近,在开发一个可以拿来即用的 golang 库。第一个组件选择了缓存,主要是因为这个组件非常的关键,但也非常不容易实现好。

第一步:定义 Cache 接口

设计一个高扩展的缓存包,就需要利用 里氏替换原则(Liskov Substitution Principle),做好抽象,将缓存定义为接口

type Cache interface {
    ...
}

第二步:组织包结构

然后,实现一个具体的 LRU 缓存,那么此时首先要组织好包结构,如下:

|-cache
| |-lru
| | |-lru.go
| | |-segment.go
| | |-options.go
| | |-expvar.go
|-cache.go

利用包划分层次,将接口放在根包下,作为所有子包的通用语言:

// cache.go
package edge
type Cache interface {
    ...
}

第三步:实现 LRU 缓存

  1. 为了防止锁竞争导致的性能低下,此处使用分段加锁的方式降低锁粒度以提高缓存性能
  2. 同时将 segmentnewSegmentcache 以小写命名,避免对外暴露实现细节
  3. 使用 Higher-order function,实现可扩展的配置参数
  4. 使用 expvar 暴露缓存的状态
// lru.go
package lru

type cache struct {
    ...
}
func New(opts ...Opt) (*cache, error) {
    ...
}

// segment.go
package lru

type segment struct {
    ...
}
func newSegment(c int) *segment {
    ...
}

// options.go
type options struct {
    ...
}

type Opt func(*options)

func WithConcurrency(c int) Opt {
	return func(o *options) {
		o.concurrency = c
	}
}

func WithCapacity(c int) Opt {
	return func(o *options) {
		o.capacity = c
	}
}

// expvar.go
var m = struct {
	Get    *expvar.Int
	Set    *expvar.Int
	Delete *expvar.Int
	Exists *expvar.Int
	Hit    *expvar.Int
	Evict  *expvar.Int
}{
	Get:    expvar.NewInt("cache.lru.get"),
	Set:    expvar.NewInt("cache.lru.set"),
	Delete: expvar.NewInt("cache.lru.delete"),
	Exists: expvar.NewInt("cache.lru.exists"),
	Hit:    expvar.NewInt("cache.lru.hit"),
	Evict:  expvar.NewInt("cache.lru.evict"),
}

第四步:结束了么?

当然没有,从以上可以看到,以下几点:

  1. options 可以做到多种实现共用,更应该放在 cache 文件夹下。
  2. 在使用时,lru.New() 赋值给 Cache 接口略微不自然
  3. segment.go 和 expvar.go 未对使用者开放但文件却对外暴露
  4. segment 可能后续会用来实现其他缓存算法,也不适合放在 lru 包下

基于以上原因,再次调整包结构如下:

|-cache
| |-options.go
| |-lru.go
|-cache.go
|-internal
| |-cache
| | |-lru
| | | |-expvar.go
| | | |-segment.go

同时,调整 LRU 缓存的接口为:

// cache.go
package cache

type cache struct {
    ...
}
func NewLRU(opts ...Opt) (*cache, error) {
    ...
}

是不是自然了很多,使用示例:

var c edge.Cache = cache.NewLRU()

总结

学以致用,此 LRU 的实现应用了很多之前的知识。追求优秀代码的路是没有尽头的,下课。

源代码:github.com/cyningsun/edge

本文作者 : cyningsun
本文地址https://www.cyningsun.com/07-26-2021/go-a-graceful-lru-implement.html
版权声明 :本博客所有文章除特别声明外,均采用 CC BY-NC-ND 3.0 CN 许可协议。转载请注明出处!

# Golang