Go实现简单的带过期机制的缓存

我们都用过 Redis 吧，他有一个非常有用的功能，就是过期机制。

原子操作

一个带过期机制的 KV，肯定不能使用原生的 map，以为他不是线程安全的。

如果用非线程安全的 map，会怎么样呢？

你的定时器可能会出现并发多个的情况。

在这里你可以自己去实现一个线程安全的 map，也可以使用 sync 包里面的 map。

我这里使用后者。

我们先写一个简单的 sync.map 的 demo：

package main  
  
import (  
 "fmt"  
 "sync"  
)  
  
var KV sync.Map  
  
func Set(key string, val interface{}) {  
 KV.Store(key,val)  
}  
  
func main() {  
 Set("name","kobe")  
 Set("age", 10)  
 fmt.Println(KV.Load("name"))  
 fmt.Println(KV.Load("age"))  
}  
  
// 执行结果  
$ go run main.go   
kobe true  
10 true

计时器

除了原子操作，还有一个点应该是他想考察的，那就是计时器。

Go 语言里面 time 包里面有一个 AfterFunc 方法，可以处理延迟执行。

在上面的代码里面，我们只需要在 Set 方法里面写上这个方法即可：

func Set(key string, val interface{}) {
	KV.Store(key, val)
	time.AfterFunc(10 * time.Second, func() {
		KV.Delete(key)
	})
}

他有两个参数，一个是时间，多久后执行，一个是执行的方法。

但是这样写不够灵活，过期时间应该是由调用这个方法来传入：

func Set(key string, val interface{}, expire time.Duration) {
	KV.Store(key, val)
	time.AfterFunc(expire, func() {
		KV.Delete(key)
	})
}

这样就OK了，我们可以这样调用：

func main() {

	Set("name", "kobe", 3*time.Second)
	Set("age", 10, 6*time.Second)

	for {
		fmt.Println(KV.Load("name"))
		fmt.Println(KV.Load("age"))
		time.Sleep(2 * time.Second)
	}

}

运行结果：

10 true
kobe true
10 true
<nil> false
10 true
<nil> false
<nil> false
<nil> false
<nil> false
<nil> false
<nil> false
<nil> false
<nil> false

使用go自带的sync.Map和time.AfterFunc可以很简单的实现一个基于内存的缓存map。key不多的时候，效果还是很不错的。

完整代码

package main

import (
	"fmt"
	"sync"
	"sync/atomic"
	"time"
)

var KV sync.Map
var Len int64

func Set(key string, val interface{}, expire time.Duration) {
	KV.Store(key, val)
	atomic.AddInt64(&Len, 1)
	time.AfterFunc(expire, func() {
		atomic.AddInt64(&Len, -1)
		KV.Delete(key)
	})
}

func Get(key string) (interface{}, bool) {
	return KV.Load(key)
}

func Length() int {
	return int(Len)
}

func main() {

	Set("name", "kobe", 3*time.Second)
	Set("age", 10, 6*time.Second)

	for {
		fmt.Println(Get("name"))
		fmt.Println(Get("age"))
		time.Sleep(2 * time.Second)
	}

}