memory_store.go

package queued

// Iterator

type MemoryIterator struct {
}

func (it *MemoryIterator) NextRecord() (*Record, bool) {
	return nil, false
}

// Store

type MemoryStore struct {
	id      int
	records map[int]*Record
}

func NewMemoryStore() *MemoryStore {
	records := make(map[int]*Record)

	return &MemoryStore{
		id:      0,
		records: records,
	}
}

func (s *MemoryStore) Get(id int) (*Record, error) {
	if record, ok := s.records[id]; ok {
		return record, nil
	} else {
		return nil, nil
	}
}

func (s *MemoryStore) Put(record *Record) error {
	record.id = s.id + 1
	s.records[record.id] = record
	s.id = record.id
	return nil
}

func (s *MemoryStore) Remove(id int) error {
	delete(s.records, id)
	return nil
}

func (s *MemoryStore) Iterator() Iterator {
	return &MemoryIterator{}
}

func (s *MemoryStore) Drop() {
	s.records = make(map[int]*Record)
}