Elton Minetto
74 lines
1.2 KiB
Go
74 lines
1.2 KiB
Go
package book
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/eminetto/clean-architecture-go-v2/entity"
|
|
)
|
|
|
|
//inmem in memory repo
|
|
type inmem struct {
|
|
m map[entity.ID]*entity.Book
|
|
}
|
|
|
|
//newInmem create new repository
|
|
func newInmem() *inmem {
|
|
var m = map[entity.ID]*entity.Book{}
|
|
return &inmem{
|
|
m: m,
|
|
}
|
|
}
|
|
|
|
//Create a book
|
|
func (r *inmem) Create(e *entity.Book) (entity.ID, error) {
|
|
r.m[e.ID] = e
|
|
return e.ID, nil
|
|
}
|
|
|
|
//Get a book
|
|
func (r *inmem) Get(id entity.ID) (*entity.Book, error) {
|
|
if r.m[id] == nil {
|
|
return nil, entity.ErrNotFound
|
|
}
|
|
return r.m[id], nil
|
|
}
|
|
|
|
//Update a book
|
|
func (r *inmem) 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 *inmem) 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 *inmem) 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 *inmem) Delete(id entity.ID) error {
|
|
if r.m[id] == nil {
|
|
return entity.ErrNotFound
|
|
}
|
|
r.m[id] = nil
|
|
return nil
|
|
}
|