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

2022-05-10 09:08:12 浏览数 (1)

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

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

面试的时候问这个问题,肯定不可能让你实现一个非常复杂的,像 Redis 那么牛逼。

我想,他是想考察的点应该是以下这些:

一、原子操作

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

如果用非线程安全的 map,会怎么样呢?

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

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

我这里使用后者。

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

代码语言:javascript复制
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 方法里面写上这个方法即可:

代码语言:javascript复制
func Set(key string, val interface{}) {
 KV.Store(key,val)
 time.AfterFunc(10*time.Second, func() {
  KV.Delete(key)
 })
}

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

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

代码语言:javascript复制
func Set(key string, val interface{}, expire time.Duration) {
 KV.Store(key,val)
 time.AfterFunc(expire, func() {
  KV.Delete(key)
 })
}

这样就OK了,我们可以这样调用:

代码语言:javascript复制
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 里面的过期机制,比这复杂多了。

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

你学废了么?

0 人点赞