minipaste/store/memory.go

package store

import (
	"bytes"
	"io"
	"io/ioutil"

	"github.com/google/uuid"
)

var _ Store = &MemoryStore{}

type MemoryStore struct {
	data map[string][]byte
}

func NewMemoryStore() *MemoryStore {
	return &MemoryStore{data: map[string][]byte{}}
}

func (s *MemoryStore) Add(r io.Reader) (string, error) {
	id := uuid.Must(uuid.NewRandom()).String()

	data, err := ioutil.ReadAll(r)
	if err != nil {
		return "", err
	}

	s.data[id] = data
	return id, nil
}

func (s *MemoryStore) Delete(id string) error {
	delete(s.data, id)
	return nil
}

func (s *MemoryStore) Get(id string) (io.ReadCloser, error) {
	data, ok := s.data[id]
	if !ok {
		return nil, ErrNoSuchItem
	}
	br := bytes.NewReader(data)

	r := io.NopCloser(br)

	return r, nil
}

func (s *MemoryStore) List() ([]string, error) {
	var ids []string
	for id := range s.data {
		ids = append(ids, id)
	}

	return ids, nil
}