feat: improve repository structure

This commit is contained in:
Elton Minetto
2020-10-12 06:58:58 -03:00
parent d4496c2585
commit 2c0f46aff1
8 changed files with 64 additions and 66 deletions
+3 -4
View File
@@ -11,8 +11,7 @@ import (
"github.com/eminetto/clean-architecture-go-v2/domain/usecase/book" "github.com/eminetto/clean-architecture-go-v2/domain/usecase/book"
"github.com/eminetto/clean-architecture-go-v2/domain/usecase/user" "github.com/eminetto/clean-architecture-go-v2/domain/usecase/user"
brepo "github.com/eminetto/clean-architecture-go-v2/infra/repository/book" "github.com/eminetto/clean-architecture-go-v2/infra/repository"
urepo "github.com/eminetto/clean-architecture-go-v2/infra/repository/user"
"github.com/eminetto/clean-architecture-go-v2/domain/usecase/loan" "github.com/eminetto/clean-architecture-go-v2/domain/usecase/loan"
@@ -37,10 +36,10 @@ func main() {
} }
defer db.Close() defer db.Close()
bookRepo := brepo.NewMySQLRepository(db) bookRepo := repository.NewBookMySQL(db)
bookService := book.NewService(bookRepo) bookService := book.NewService(bookRepo)
userRepo := urepo.NewMySQLRepoRepository(db) userRepo := repository.NewUserMySQL(db)
userService := user.NewService(userRepo) userService := user.NewService(userRepo)
loanUseCase := loan.NewService(userService, bookService) loanUseCase := loan.NewService(userService, bookService)
+3 -3
View File
@@ -7,8 +7,8 @@ import (
"log" "log"
"os" "os"
book "github.com/eminetto/clean-architecture-go-v2/domain/usecase/book" "github.com/eminetto/clean-architecture-go-v2/domain/usecase/book"
repo "github.com/eminetto/clean-architecture-go-v2/infra/repository/book" "github.com/eminetto/clean-architecture-go-v2/infra/repository"
"github.com/eminetto/clean-architecture-go-v2/config" "github.com/eminetto/clean-architecture-go-v2/config"
_ "github.com/go-sql-driver/mysql" _ "github.com/go-sql-driver/mysql"
@@ -41,7 +41,7 @@ func main() {
log.Fatal(err.Error()) log.Fatal(err.Error())
} }
defer db.Close() defer db.Close()
repo := repo.NewMySQLRepository(db) repo := repository.NewBookMySQL(db)
service := book.NewService(repo) service := book.NewService(repo)
all, err := service.SearchBooks(query) all, err := service.SearchBooks(query)
if err != nil { if err != nil {
+5 -6
View File
@@ -5,8 +5,7 @@ import (
"time" "time"
"github.com/eminetto/clean-architecture-go-v2/domain/entity" "github.com/eminetto/clean-architecture-go-v2/domain/entity"
"github.com/eminetto/clean-architecture-go-v2/infra/repository"
"github.com/eminetto/clean-architecture-go-v2/infra/repository/book"
"github.com/eminetto/clean-architecture-go-v2/domain" "github.com/eminetto/clean-architecture-go-v2/domain"
@@ -24,7 +23,7 @@ func newFixtureBook() *entity.Book {
} }
func Test_Create(t *testing.T) { func Test_Create(t *testing.T) {
repo := book.NewInmemRepository() repo := repository.NewBookInmem()
m := NewService(repo) m := NewService(repo)
u := newFixtureBook() u := newFixtureBook()
_, err := m.CreateBook(u.Title, u.Author, u.Pages, u.Quantity) _, err := m.CreateBook(u.Title, u.Author, u.Pages, u.Quantity)
@@ -33,7 +32,7 @@ func Test_Create(t *testing.T) {
} }
func Test_SearchAndFind(t *testing.T) { func Test_SearchAndFind(t *testing.T) {
repo := book.NewInmemRepository() repo := repository.NewBookInmem()
m := NewService(repo) m := NewService(repo)
u1 := newFixtureBook() u1 := newFixtureBook()
u2 := newFixtureBook() u2 := newFixtureBook()
@@ -66,7 +65,7 @@ func Test_SearchAndFind(t *testing.T) {
} }
func Test_Update(t *testing.T) { func Test_Update(t *testing.T) {
repo := book.NewInmemRepository() repo := repository.NewBookInmem()
m := NewService(repo) m := NewService(repo)
u := newFixtureBook() u := newFixtureBook()
id, err := m.CreateBook(u.Title, u.Author, u.Pages, u.Quantity) id, err := m.CreateBook(u.Title, u.Author, u.Pages, u.Quantity)
@@ -80,7 +79,7 @@ func Test_Update(t *testing.T) {
} }
func TestDelete(t *testing.T) { func TestDelete(t *testing.T) {
repo := book.NewInmemRepository() repo := repository.NewBookInmem()
m := NewService(repo) m := NewService(repo)
u1 := newFixtureBook() u1 := newFixtureBook()
u2 := newFixtureBook() u2 := newFixtureBook()
+5 -5
View File
@@ -4,7 +4,7 @@ import (
"testing" "testing"
"time" "time"
"github.com/eminetto/clean-architecture-go-v2/infra/repository/user" "github.com/eminetto/clean-architecture-go-v2/infra/repository"
"github.com/eminetto/clean-architecture-go-v2/domain" "github.com/eminetto/clean-architecture-go-v2/domain"
"github.com/eminetto/clean-architecture-go-v2/domain/entity" "github.com/eminetto/clean-architecture-go-v2/domain/entity"
@@ -24,7 +24,7 @@ func newFixtureUser() *entity.User {
} }
func Test_Create(t *testing.T) { func Test_Create(t *testing.T) {
repo := user.NewInmemRepository() repo := repository.NewUserInmem()
m := NewService(repo) m := NewService(repo)
u := newFixtureUser() u := newFixtureUser()
_, err := m.CreateUser(u.Email, u.Password, u.FirstName, u.LastName) _, err := m.CreateUser(u.Email, u.Password, u.FirstName, u.LastName)
@@ -34,7 +34,7 @@ func Test_Create(t *testing.T) {
} }
func Test_SearchAndFind(t *testing.T) { func Test_SearchAndFind(t *testing.T) {
repo := user.NewInmemRepository() repo := repository.NewUserInmem()
m := NewService(repo) m := NewService(repo)
u1 := newFixtureUser() u1 := newFixtureUser()
u2 := newFixtureUser() u2 := newFixtureUser()
@@ -67,7 +67,7 @@ func Test_SearchAndFind(t *testing.T) {
} }
func Test_Update(t *testing.T) { func Test_Update(t *testing.T) {
repo := user.NewInmemRepository() repo := repository.NewUserInmem()
m := NewService(repo) m := NewService(repo)
u := newFixtureUser() u := newFixtureUser()
id, err := m.CreateUser(u.Email, u.Password, u.FirstName, u.LastName) id, err := m.CreateUser(u.Email, u.Password, u.FirstName, u.LastName)
@@ -84,7 +84,7 @@ func Test_Update(t *testing.T) {
} }
func TestDelete(t *testing.T) { func TestDelete(t *testing.T) {
repo := user.NewInmemRepository() repo := repository.NewUserInmem()
m := NewService(repo) m := NewService(repo)
u1 := newFixtureUser() u1 := newFixtureUser()
u2 := newFixtureUser() u2 := newFixtureUser()
@@ -1,4 +1,4 @@
package book package repository
import ( import (
"strings" "strings"
@@ -7,27 +7,27 @@ import (
"github.com/eminetto/clean-architecture-go-v2/domain/entity" "github.com/eminetto/clean-architecture-go-v2/domain/entity"
) )
//InmemRepo in memory repo //BookInmem in memory repo
type InmemRepo struct { type BookInmem struct {
m map[entity.ID]*entity.Book m map[entity.ID]*entity.Book
} }
//NewInmemRepository create new repository //NewBookInmem create new repository
func NewInmemRepository() *InmemRepo { func NewBookInmem() *BookInmem {
var m = map[entity.ID]*entity.Book{} var m = map[entity.ID]*entity.Book{}
return &InmemRepo{ return &BookInmem{
m: m, m: m,
} }
} }
//Create a book //Create a book
func (r *InmemRepo) Create(e *entity.Book) (entity.ID, error) { func (r *BookInmem) Create(e *entity.Book) (entity.ID, error) {
r.m[e.ID] = e r.m[e.ID] = e
return e.ID, nil return e.ID, nil
} }
//Get a book //Get a book
func (r *InmemRepo) Get(id entity.ID) (*entity.Book, error) { func (r *BookInmem) Get(id entity.ID) (*entity.Book, error) {
if r.m[id] == nil { if r.m[id] == nil {
// return nil, fmt.Errorf("not found") // return nil, fmt.Errorf("not found")
return nil, domain.ErrNotFound return nil, domain.ErrNotFound
@@ -36,7 +36,7 @@ func (r *InmemRepo) Get(id entity.ID) (*entity.Book, error) {
} }
//Update a book //Update a book
func (r *InmemRepo) Update(e *entity.Book) error { func (r *BookInmem) Update(e *entity.Book) error {
_, err := r.Get(e.ID) _, err := r.Get(e.ID)
if err != nil { if err != nil {
return err return err
@@ -46,7 +46,7 @@ func (r *InmemRepo) Update(e *entity.Book) error {
} }
//Search books //Search books
func (r *InmemRepo) Search(query string) ([]*entity.Book, error) { func (r *BookInmem) Search(query string) ([]*entity.Book, error) {
var d []*entity.Book var d []*entity.Book
for _, j := range r.m { for _, j := range r.m {
if strings.Contains(strings.ToLower(j.Title), query) { if strings.Contains(strings.ToLower(j.Title), query) {
@@ -57,7 +57,7 @@ func (r *InmemRepo) Search(query string) ([]*entity.Book, error) {
} }
//List books //List books
func (r *InmemRepo) List() ([]*entity.Book, error) { func (r *BookInmem) List() ([]*entity.Book, error) {
var d []*entity.Book var d []*entity.Book
for _, j := range r.m { for _, j := range r.m {
d = append(d, j) d = append(d, j)
@@ -66,7 +66,7 @@ func (r *InmemRepo) List() ([]*entity.Book, error) {
} }
//Delete a book //Delete a book
func (r *InmemRepo) Delete(id entity.ID) error { func (r *BookInmem) Delete(id entity.ID) error {
if r.m[id] == nil { if r.m[id] == nil {
// return fmt.Errorf("not found") // return fmt.Errorf("not found")
return domain.ErrNotFound return domain.ErrNotFound
@@ -1,4 +1,4 @@
package book package repository
import ( import (
"database/sql" "database/sql"
@@ -7,20 +7,20 @@ import (
"github.com/eminetto/clean-architecture-go-v2/domain/entity" "github.com/eminetto/clean-architecture-go-v2/domain/entity"
) )
//MySQLRepo mysql repo //BookMySQL mysql repo
type MySQLRepo struct { type BookMySQL struct {
db *sql.DB db *sql.DB
} }
//NewMySQLRepository create new repository //NewBookMySQL create new repository
func NewMySQLRepository(db *sql.DB) *MySQLRepo { func NewBookMySQL(db *sql.DB) *BookMySQL {
return &MySQLRepo{ return &BookMySQL{
db: db, db: db,
} }
} }
//Create a book //Create a book
func (r *MySQLRepo) Create(e *entity.Book) (entity.ID, error) { func (r *BookMySQL) Create(e *entity.Book) (entity.ID, error) {
stmt, err := r.db.Prepare(` stmt, err := r.db.Prepare(`
insert into book (id, title, author, pages, quantity, created_at) insert into book (id, title, author, pages, quantity, created_at)
values(?,?,?,?,?,?)`) values(?,?,?,?,?,?)`)
@@ -46,7 +46,7 @@ func (r *MySQLRepo) Create(e *entity.Book) (entity.ID, error) {
} }
//Get a book //Get a book
func (r *MySQLRepo) Get(id entity.ID) (*entity.Book, error) { func (r *BookMySQL) Get(id entity.ID) (*entity.Book, error) {
stmt, err := r.db.Prepare(`select id, title, author, pages, quantity, created_at from book where id = ?`) stmt, err := r.db.Prepare(`select id, title, author, pages, quantity, created_at from book where id = ?`)
if err != nil { if err != nil {
return nil, err return nil, err
@@ -63,7 +63,7 @@ func (r *MySQLRepo) Get(id entity.ID) (*entity.Book, error) {
} }
//Update a book //Update a book
func (r *MySQLRepo) Update(e *entity.Book) error { func (r *BookMySQL) Update(e *entity.Book) error {
e.UpdatedAt = time.Now() e.UpdatedAt = time.Now()
_, err := r.db.Exec("update book set title = ?, author = ?, pages = ?, quantity = ?, updated_at = ? where id = ?", e.Title, e.Author, e.Pages, e.Quantity, e.UpdatedAt.Format("2006-01-02"), e.ID) _, err := r.db.Exec("update book set title = ?, author = ?, pages = ?, quantity = ?, updated_at = ? where id = ?", e.Title, e.Author, e.Pages, e.Quantity, e.UpdatedAt.Format("2006-01-02"), e.ID)
if err != nil { if err != nil {
@@ -73,7 +73,7 @@ func (r *MySQLRepo) Update(e *entity.Book) error {
} }
//Search books //Search books
func (r *MySQLRepo) Search(query string) ([]*entity.Book, error) { func (r *BookMySQL) Search(query string) ([]*entity.Book, error) {
stmt, err := r.db.Prepare(`select id, title, author, pages, quantity, created_at from book where title like ?`) stmt, err := r.db.Prepare(`select id, title, author, pages, quantity, created_at from book where title like ?`)
if err != nil { if err != nil {
return nil, err return nil, err
@@ -96,7 +96,7 @@ func (r *MySQLRepo) Search(query string) ([]*entity.Book, error) {
} }
//List books //List books
func (r *MySQLRepo) List() ([]*entity.Book, error) { func (r *BookMySQL) List() ([]*entity.Book, error) {
stmt, err := r.db.Prepare(`select id, title, author, pages, quantity, created_at from book`) stmt, err := r.db.Prepare(`select id, title, author, pages, quantity, created_at from book`)
if err != nil { if err != nil {
return nil, err return nil, err
@@ -118,7 +118,7 @@ func (r *MySQLRepo) List() ([]*entity.Book, error) {
} }
//Delete a book //Delete a book
func (r *MySQLRepo) Delete(id entity.ID) error { func (r *BookMySQL) Delete(id entity.ID) error {
_, err := r.db.Exec("delete from book where id = ?", id) _, err := r.db.Exec("delete from book where id = ?", id)
if err != nil { if err != nil {
return err return err
@@ -1,4 +1,4 @@
package user package repository
import ( import (
"fmt" "fmt"
@@ -8,27 +8,27 @@ import (
"github.com/eminetto/clean-architecture-go-v2/domain/entity" "github.com/eminetto/clean-architecture-go-v2/domain/entity"
) )
//IRepo in memory repo //UserInmem in memory repo
type IRepo struct { type UserInmem struct {
m map[entity.ID]*entity.User m map[entity.ID]*entity.User
} }
//NewInmemRepository create new repository //NewUserInmem create new repository
func NewInmemRepository() *IRepo { func NewUserInmem() *UserInmem {
var m = map[entity.ID]*entity.User{} var m = map[entity.ID]*entity.User{}
return &IRepo{ return &UserInmem{
m: m, m: m,
} }
} }
//Create an user //Create an user
func (r *IRepo) Create(e *entity.User) (entity.ID, error) { func (r *UserInmem) Create(e *entity.User) (entity.ID, error) {
r.m[e.ID] = e r.m[e.ID] = e
return e.ID, nil return e.ID, nil
} }
//Get an user //Get an user
func (r *IRepo) Get(id entity.ID) (*entity.User, error) { func (r *UserInmem) Get(id entity.ID) (*entity.User, error) {
if r.m[id] == nil { if r.m[id] == nil {
// return nil, fmt.Errorf("not found") // return nil, fmt.Errorf("not found")
return nil, domain.ErrNotFound return nil, domain.ErrNotFound
@@ -37,7 +37,7 @@ func (r *IRepo) Get(id entity.ID) (*entity.User, error) {
} }
//Update an user //Update an user
func (r *IRepo) Update(e *entity.User) error { func (r *UserInmem) Update(e *entity.User) error {
_, err := r.Get(e.ID) _, err := r.Get(e.ID)
if err != nil { if err != nil {
return err return err
@@ -47,7 +47,7 @@ func (r *IRepo) Update(e *entity.User) error {
} }
//Search users //Search users
func (r *IRepo) Search(query string) ([]*entity.User, error) { func (r *UserInmem) Search(query string) ([]*entity.User, error) {
var d []*entity.User var d []*entity.User
for _, j := range r.m { for _, j := range r.m {
if strings.Contains(strings.ToLower(j.FirstName), query) { if strings.Contains(strings.ToLower(j.FirstName), query) {
@@ -63,7 +63,7 @@ func (r *IRepo) Search(query string) ([]*entity.User, error) {
} }
//List users //List users
func (r *IRepo) List() ([]*entity.User, error) { func (r *UserInmem) List() ([]*entity.User, error) {
var d []*entity.User var d []*entity.User
for _, j := range r.m { for _, j := range r.m {
d = append(d, j) d = append(d, j)
@@ -72,7 +72,7 @@ func (r *IRepo) List() ([]*entity.User, error) {
} }
//Delete an user //Delete an user
func (r *IRepo) Delete(id entity.ID) error { func (r *UserInmem) Delete(id entity.ID) error {
if r.m[id] == nil { if r.m[id] == nil {
return fmt.Errorf("not found") return fmt.Errorf("not found")
} }
@@ -1,4 +1,4 @@
package user package repository
import ( import (
"database/sql" "database/sql"
@@ -8,20 +8,20 @@ import (
"github.com/eminetto/clean-architecture-go-v2/domain/entity" "github.com/eminetto/clean-architecture-go-v2/domain/entity"
) )
//MySQLRepo mysql repo //UserMySQL mysql repo
type MySQLRepo struct { type UserMySQL struct {
db *sql.DB db *sql.DB
} }
//NewMySQLRepoRepository create new repository //NewUserMySQL create new repository
func NewMySQLRepoRepository(db *sql.DB) *MySQLRepo { func NewUserMySQL(db *sql.DB) *UserMySQL {
return &MySQLRepo{ return &UserMySQL{
db: db, db: db,
} }
} }
//Create an user //Create an user
func (r *MySQLRepo) Create(e *entity.User) (entity.ID, error) { func (r *UserMySQL) Create(e *entity.User) (entity.ID, error) {
stmt, err := r.db.Prepare(` stmt, err := r.db.Prepare(`
insert into user (id, email, password, first_name, last_name, created_at) insert into user (id, email, password, first_name, last_name, created_at)
values(?,?,?,?,?,?)`) values(?,?,?,?,?,?)`)
@@ -47,7 +47,7 @@ func (r *MySQLRepo) Create(e *entity.User) (entity.ID, error) {
} }
//Get an user //Get an user
func (r *MySQLRepo) Get(id entity.ID) (*entity.User, error) { func (r *UserMySQL) Get(id entity.ID) (*entity.User, error) {
return getUser(id, r.db) return getUser(id, r.db)
} }
@@ -81,7 +81,7 @@ func getUser(id entity.ID, db *sql.DB) (*entity.User, error) {
} }
//Update an user //Update an user
func (r *MySQLRepo) Update(e *entity.User) error { func (r *UserMySQL) Update(e *entity.User) error {
e.UpdatedAt = time.Now() e.UpdatedAt = time.Now()
_, err := r.db.Exec("update user set email = ?, password = ?, first_name = ?, last_name = ?, updated_at = ? where id = ?", e.Email, e.Password, e.FirstName, e.LastName, e.UpdatedAt.Format("2006-01-02"), e.ID) _, err := r.db.Exec("update user set email = ?, password = ?, first_name = ?, last_name = ?, updated_at = ? where id = ?", e.Email, e.Password, e.FirstName, e.LastName, e.UpdatedAt.Format("2006-01-02"), e.ID)
if err != nil { if err != nil {
@@ -101,7 +101,7 @@ func (r *MySQLRepo) Update(e *entity.User) error {
} }
//Search users //Search users
func (r *MySQLRepo) Search(query string) ([]*entity.User, error) { func (r *UserMySQL) Search(query string) ([]*entity.User, error) {
stmt, err := r.db.Prepare(`select id from user where name like ?`) stmt, err := r.db.Prepare(`select id from user where name like ?`)
if err != nil { if err != nil {
return nil, err return nil, err
@@ -135,7 +135,7 @@ func (r *MySQLRepo) Search(query string) ([]*entity.User, error) {
} }
//List users //List users
func (r *MySQLRepo) List() ([]*entity.User, error) { func (r *UserMySQL) List() ([]*entity.User, error) {
stmt, err := r.db.Prepare(`select id from user`) stmt, err := r.db.Prepare(`select id from user`)
if err != nil { if err != nil {
return nil, err return nil, err
@@ -169,7 +169,7 @@ func (r *MySQLRepo) List() ([]*entity.User, error) {
} }
//Delete an user //Delete an user
func (r *MySQLRepo) Delete(id entity.ID) error { func (r *UserMySQL) Delete(id entity.ID) error {
_, err := r.db.Exec("delete from user where id = ?", id) _, err := r.db.Exec("delete from user where id = ?", id)
if err != nil { if err != nil {
return err return err