Files
clean-architecture-go-v2/infra/repository/book/repository_inmem.go
T
2020-10-01 08:30:36 -03:00

77 lines
1.4 KiB
Go

package book
import (
"strings"
"github.com/eminetto/clean-architecture-go-v2/domain"
"github.com/eminetto/clean-architecture-go-v2/domain/entity"
)
//InmemRepo in memory repo
type InmemRepo struct {
m map[entity.ID]*entity.Book
}
//NewInmemRepository create new repository
func NewInmemRepository() *InmemRepo {
var m = map[entity.ID]*entity.Book{}
return &InmemRepo{
m: m,
}
}
//Create a book
func (r *InmemRepo) Create(e *entity.Book) (entity.ID, error) {
r.m[e.ID] = e
return e.ID, nil
}
//Get a book
func (r *InmemRepo) Get(id entity.ID) (*entity.Book, error) {
if r.m[id] == nil {
// return nil, fmt.Errorf("not found")
return nil, domain.ErrNotFound
}
return r.m[id], nil
}
//Update a book
func (r *InmemRepo) Update(e *entity.Book) error {
_, err := r.Get(e.ID)
if err != nil {
return err
}
r.m[e.ID] = e
return nil
}
//Search books
func (r *InmemRepo) Search(query string) ([]*entity.Book, error) {
var d []*entity.Book
for _, j := range r.m {
if strings.Contains(strings.ToLower(j.Title), query) {
d = append(d, j)
}
}
return d, nil
}
//List books
func (r *InmemRepo) List() ([]*entity.Book, error) {
var d []*entity.Book
for _, j := range r.m {
d = append(d, j)
}
return d, nil
}
//Delete a book
func (r *InmemRepo) Delete(id entity.ID) error {
if r.m[id] == nil {
// return fmt.Errorf("not found")
return domain.ErrNotFound
}
r.m[id] = nil
return nil
}