gosora/common/widget_store.go

40 lines
642 B
Go

package common
import (
"database/sql"
"sync"
)
var Widgets *DefaultWidgetStore
type DefaultWidgetStore struct {
widgets map[int]*Widget
sync.RWMutex
}
func NewDefaultWidgetStore() *DefaultWidgetStore {
return &DefaultWidgetStore{widgets: make(map[int]*Widget)}
}
func (s *DefaultWidgetStore) Get(id int) (*Widget, error) {
s.RLock()
defer s.RUnlock()
w, ok := s.widgets[id]
if !ok {
return w, sql.ErrNoRows
}
return w, nil
}
func (s *DefaultWidgetStore) set(w *Widget) {
s.Lock()
defer s.Unlock()
s.widgets[w.ID] = w
}
func (s *DefaultWidgetStore) delete(id int) {
s.Lock()
defer s.Unlock()
delete(s.widgets, id)
}