带过期机制的map

实现带过期机制的map，类似redis的setex命令

实现：

package main

import (
	"fmt"
	"sync"
	"time"
)

var m sync.Map
var timeMap sync.Map
func set(key string,val interface{},exp time.Duration) {

	expT := time.Now().Add(exp)
	m.Store(key,val)
	timeMap.Store(key,&expT)
	time.AfterFunc(exp, func() {
		v,ok := timeMap.Load(key)
		if !ok {
			return
		}
		if v.(*time.Time).Sub(time.Now()).Seconds() > 0 {
			return
		}
		m.Delete(key)
	})
}

测试：

func main() {
	set("name","lufei",time.Second*10)
	start := time.Now()
	i := 0
	for {
		i++
		time.Sleep(time.Second)
		if i == 3 {
			fmt.Println("执行")
			set("name","lufei",time.Second*10)
		}
		val, ok := m.Load("name")
		if ok {
			fmt.Println(val)
		}else {
			break
		}
	}
	t := time.Since(start)
	fmt.Println(t)
}