通过 collection.Cache 进行缓存

go-zero微服务框架中提供了许多开箱即用的工具，好的工具不仅能提升服务的性能而且还能提升代码的鲁棒性避免出错，实现代码风格的统一方便他人阅读等等，本系列文章将分别介绍go-zero框架中工具的使用及其实现原理

进程内缓存工具collection.Cache

在做服务器开发的时候，相信都会遇到使用缓存的情况，go-zero 提供的简单的缓存封装 collection.Cache，简单使用方式如下

  1. // 初始化 cache，其中 WithLimit 可以指定最大缓存的数量
  2. c, err := collection.NewCache(time.Minute, collection.WithLimit(10000))
  3. if err != nil {
  4.   panic(err)
  5. }
  7. // 设置缓存
  8. c.Set("key", user)
  10. // 获取缓存，ok：是否存在
  11. v, ok := c.Get("key")
  13. // 删除缓存
  14. c.Del("key")
  16. // 获取缓存，如果 key 不存在的，则会调用 func 去生成缓存
  17. v, err := c.Take("key", func() (interface{}, error) {
  18.   return user, nil
  19. })

cache 实现的建的功能包括

  • 缓存自动失效，可以指定过期时间
  • 缓存大小限制，可以指定缓存个数
  • 缓存增删改
  • 缓存命中率统计
  • 并发安全
  • 缓存击穿

实现原理： Cache 自动失效，是采用 TimingWheel(https://github.com/zeromicro/zeromicro/blob/master/core/collection/timingwheel.go) 进行管理的

  1. timingWheel, err := NewTimingWheel(time.Second, slots, func(k, v interface{}) {
  2.         key, ok := k.(string)
  3.         if !ok {
  4.             return
  5.         }
  7.         cache.Del(key)
  8. })

Cache 大小限制，是采用 LRU 淘汰策略，在新增缓存的时候会去检查是否已经超出过限制，具体代码在 keyLru 中实现

  1. func (klru *keyLru) add(key string) {
  2.     if elem, ok := klru.elements[key]; ok {
  3.         klru.evicts.MoveToFront(elem)
  4.         return
  5.     }
  7.     // Add new item
  8.     elem := klru.evicts.PushFront(key)
  9.     klru.elements[key] = elem
  11.     // Verify size not exceeded
  12.     if klru.evicts.Len() > klru.limit {
  13.         klru.removeOldest()
  14.     }
  15. }

Cache 的命中率统计，是在代码中实现 cacheStat,在缓存命中丢失的时候自动统计，并且会定时打印使用的命中率, qps 等状态.

打印的具体效果如下

  1. cache(proc) - qpm: 2, hit_ratio: 50.0%, elements: 0, hit: 1, miss: 1

缓存击穿包含是使用 syncx.SingleFlight(https://github.com/zeromicro/zeromicro/blob/master/core/syncx/singleflight.go) 进行实现的，就是将同时请求同一个 key 的请求, 关于 SingleFlight 后续会继续补充。 相关具体实现是在:

  1. func (c *Cache) Take(key string, fetch func() (interface{}, error)) (interface{}, error) {
  2.     val, fresh, err := c.barrier.DoEx(key, func() (interface{}, error) {
  3.         v, e := fetch()
  4.         if e != nil {
  5.             return nil, e
  6.         }
  8.         c.Set(key, v)
  9.         return v, nil
  10.     })
  11.     if err != nil {
  12.         return nil, err
  13.     }
  15.     if fresh {
  16.         c.stats.IncrementMiss()
  17.         return val, nil
  18.     } else {
  19.         // got the result from previous ongoing query
  20.         c.stats.IncrementHit()
  21.     }
  23.     return val, nil
  24. }

本文主要介绍了go-zero框架中的 Cache 工具，在实际的项目中非常实用。用好工具对于提升服务性能和开发效率都有很大的帮助，希望本篇文章能给大家带来一些收获。