// Write implementations of a cache with ttl (time to live) for the User structure. // You can use User.ID as the key // We want it to be thread-safe // We do not want to retrieve entries with expired ttl during read operations // We want expired ttl entries to be deleted and not stored // API get/set/del
sobes.tech AI
Answer from AI
go package main
import ( "sync" "time" )
type User struct { ID string // other fields }
type cacheItem struct { user User expiresAt time.Time }
type UserCache struct { mu sync.RWMutex items map[string]cacheItem }
func NewUserCache() *UserCache { c := &UserCache{ items: make(map[string]cacheItem), } // Start a goroutine for periodic cleanup of expired records go func() { for { time.Sleep(time.Minute) // cleanup interval c.cleanup() } }() return c }
func (c *UserCache) Set(user User, ttl time.Duration) { c.mu.Lock() defer c.mu.Unlock() c.items[user.ID] = cacheItem{ user: user, expiresAt: time.Now().Add(ttl), } }
func (c *UserCache) Get(id string) (User, bool) { c.mu.RLock() item, found := c.items[id] c.mu.RUnlock() if !found { return User{}, false } if time.Now().After(item.expiresAt) { // Record expired, delete c.mu.Lock() delete(c.items, id) c.mu.Unlock() return User{}, false } return item.user, true }
func (c *UserCache) Del(id string) { c.mu.Lock() defer c.mu.Unlock() delete(c.items, id) }
func (c *UserCache) cleanup() { c.mu.Lock() defer c.mu.Unlock() now := time.Now() for id, item := range c.items { if now.After(item.expiresAt) { delete(c.items, id) } } }
// Example usage // func main() { // cache := NewUserCache() // user := User{ID: "123"} // cache.Set(user, 5*time.Minute) // u, ok := cache.Get("123") // if ok { // fmt.Println("User found:", u) // } // }