gosora/common/menu_item_store.go

package common

import "sync"

type DefaultMenuItemStore struct {
    items map[int]MenuItem
    lock  sync.RWMutex
}

func NewDefaultMenuItemStore() *DefaultMenuItemStore {
    return &DefaultMenuItemStore{
        items: make(map[int]MenuItem),
    }
}

func (s *DefaultMenuItemStore) Add(i MenuItem) {
    s.lock.Lock()
    defer s.lock.Unlock()
    s.items[i.ID] = i
}

func (s *DefaultMenuItemStore) Get(id int) (MenuItem, error) {
    s.lock.RLock()
    item, ok := s.items[id]
    s.lock.RUnlock()
    if ok {
        return item, nil
    }
    return item, ErrNoRows
}