cachex/cachex.go

package cachex

import (
	"errors"
	"sync"
	"time"
)

// 应用内缓存
// 特点
// 1.全局单实例
// 到设定的点就删除

type cache struct {
	store sync.Map
}

var one sync.Once
var c *cache

type cacheData struct {
	key    string
	data   interface{}
	expire time.Time
}

var ErrorEmpty error = errors.New("empty cache")

func NewCache() *cache {
	one.Do(func() {
		c = &cache{}

		go func() {
			for {
				c.store.Range(func(key, value interface{}) bool {
					if value.(*cacheData).expire.Before(time.Now()) {
						c.store.Delete(key)
					}
					return true
				})

				time.Sleep(time.Second * 5)
			}
		}()
	})
	return c
}

// 设置缓存
func (c *cache) Set(key string, value interface{}, expire time.Duration) {
	if expire == 0 {
		expire = time.Hour * 24 * 365
	}
	cd := &cacheData{key, value, time.Now().Add(expire)}
	c.store.Store(key, cd)
}

// 读取缓存
func (c *cache) Get(key string) (interface{}, error) {
	if v, ok := c.store.Load(key); ok {

		cc := v.(*cacheData)
		if cc.expire.Before(time.Now()) {
			c.store.Delete(key)
			return nil, ErrorEmpty
		}
		return cc.data, nil
	}
	return nil, ErrorEmpty
}

// 删除缓存
func (c *cache) Delete(key string) {
	c.store.Delete(key)
}