91. Go实现一个最简单的带过期机制的KV存储：time.AfterFunc妙用

一、原子操作

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

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

你的定时器可能会出现并发使用map的情况，产生竞争引发panic。

所以我们使用 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","zs")
 Set("age", 10)
 fmt.Println(KV.Load("name"))
 fmt.Println(KV.Load("age"))
}

// 执行结果

$ go run main.go 
zs true
10 true

二、计时器

要带过期机制，自然想到的就是计时器。

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

我们只需要在上面的 Set 方法里面写上 time.AfterFunc方法即可：

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","zs", 4*time.Second)
 Set("age", 10, 6*time.Second)
 for  {
  fmt.Println(KV.Load("name"))
  fmt.Println(KV.Load("age"))
  time.Sleep(2*time.Second)
 }
}

// 执行结果

$ go run kv.go 
zs true
10 true
zs true
10 true
<nil> false
10 true
<nil> false
<nil> false

我们的带定时器的 KV 就完成了。

我们这里只是实现了一个非常简单的定时器，复杂场景下，考虑过期机制还需要考虑很多其他策略，比如Redis 里面的过期机制，比这复杂多了。