package utils

import (
	"sync"
	"time"
)

type KeyValueCache interface {
	Set(k, v string, tm *time.Time) error
	Get(k string) (string, bool)
	Del(k string) error
}

type MapCacheItem struct {
	Value  string
	Expire *time.Time
}

type MapCache struct {
	val sync.Map
}

func (m *MapCache) Set(k, v string, tm *time.Time) error {
	m.val.Store(k, &MapCacheItem{
		Value:  v,
		Expire: tm,
	})
	return nil
}

func (m *MapCache) Get(k string) (string, bool) {
	if vr, ok := m.val.Load(k); ok {
		if v, ok := vr.(*MapCacheItem); ok {
			if v.Expire == nil || v.Expire.After(time.Now()) {
				return v.Value, true
			}
		}
	}
	return "", false
}

func (m *MapCache) Del(k string) error {
	m.val.LoadAndDelete(k)
	return nil
}

func NewMapCache() *MapCache {
	return &MapCache{}
}