面试官：请实现一个简单带过期机制的KV

小锟哥哥

发布于 2022-05-10 09:08:12

3100

发布于 2022-05-10 09:08:12

文章被收录于专栏：GoLang全栈

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

如果面试的时候，让你实现一个带过期机制的 KV 我们该注意哪些点呢？

面试的时候问这个问题，肯定不可能让你实现一个非常复杂的，像 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","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 方法里面写上这个方法即可：

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

你如果还有其他的实现方案，欢迎向我们反馈，我们可以后期再更文扩展！

你学废了么？

本文参与腾讯云自媒体同步曝光计划，分享自微信公众号。

原始发表：2022-02-25，如有侵权请联系 cloudcommunity@tencent.com 删除

云数据库 Redis

安全

本文分享自 GoLang全栈微信公众号，前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与腾讯云自媒体同步曝光计划，欢迎热爱写作的你一起参与！

云数据库 Redis

安全

登录后参与评论

0 条评论

热度