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

最新推荐文章于 2025-05-02 11:13:23 发布

原创最新推荐文章于 2025-05-02 11:13:23 发布 · 621 阅读

7 ·

CC 4.0 BY-SA版权

文章标签：

#golang #开发语言 #后端

go 专栏收录该内容

174 篇文章

订阅专栏

本文介绍了如何在Go语言中使用sync.Map进行线程安全的键值存储，并结合time包的AfterFunc实现带过期机制的KV。作者还讨论了更复杂的过期策略，如Redis中的实现。

文章目录

一、原子操作
二、计时器

一、原子操作

一个带过期机制的 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 里面的过期机制，比这复杂多了。