feat: change folder structure

2020-10-24 11:36:58 -03:00
parent d755aabe11
commit 62f7fb4dd4
4 changed files with 2 additions and 2 deletions
@@ -0,0 +1,127 @@
+package repository
+
+import (
+	"database/sql"
+	"time"
+
+	"github.com/eminetto/clean-architecture-go-v2/entity"
+)
+
+//BookMySQL mysql repo
+type BookMySQL struct {
+	db *sql.DB
+}
+
+//NewBookMySQL create new repository
+func NewBookMySQL(db *sql.DB) *BookMySQL {
+	return &BookMySQL{
+		db: db,
+	}
+}
+
+//Create a book
+func (r *BookMySQL) Create(e *entity.Book) (entity.ID, error) {
+	stmt, err := r.db.Prepare(`
+		insert into book (id, title, author, pages, quantity, created_at) 
+		values(?,?,?,?,?,?)`)
+	if err != nil {
+		return e.ID, err
+	}
+	_, err = stmt.Exec(
+		e.ID,
+		e.Title,
+		e.Author,
+		e.Pages,
+		e.Quantity,
+		time.Now().Format("2006-01-02"),
+	)
+	if err != nil {
+		return e.ID, err
+	}
+	err = stmt.Close()
+	if err != nil {
+		return e.ID, err
+	}
+	return e.ID, nil
+}
+
+//Get a book
+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 = ?`)
+	if err != nil {
+		return nil, err
+	}
+	var b entity.Book
+	rows, err := stmt.Query(id)
+	if err != nil {
+		return nil, err
+	}
+	for rows.Next() {
+		err = rows.Scan(&b.ID, &b.Title, &b.Author, &b.Pages, &b.Quantity, &b.CreatedAt)
+	}
+	return &b, nil
+}
+
+//Update a book
+func (r *BookMySQL) Update(e *entity.Book) error {
+	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)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+//Search books
+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 ?`)
+	if err != nil {
+		return nil, err
+	}
+	var books []*entity.Book
+	rows, err := stmt.Query("%" + query + "%")
+	if err != nil {
+		return nil, err
+	}
+	for rows.Next() {
+		var b entity.Book
+		err = rows.Scan(&b.ID, &b.Title, &b.Author, &b.Pages, &b.Quantity, &b.CreatedAt)
+		if err != nil {
+			return nil, err
+		}
+		books = append(books, &b)
+	}
+
+	return books, nil
+}
+
+//List books
+func (r *BookMySQL) List() ([]*entity.Book, error) {
+	stmt, err := r.db.Prepare(`select id, title, author, pages, quantity, created_at from book`)
+	if err != nil {
+		return nil, err
+	}
+	var books []*entity.Book
+	rows, err := stmt.Query()
+	if err != nil {
+		return nil, err
+	}
+	for rows.Next() {
+		var b entity.Book
+		err = rows.Scan(&b.ID, &b.Title, &b.Author, &b.Pages, &b.Quantity, &b.CreatedAt)
+		if err != nil {
+			return nil, err
+		}
+		books = append(books, &b)
+	}
+	return books, nil
+}
+
+//Delete a book
+func (r *BookMySQL) Delete(id entity.ID) error {
+	_, err := r.db.Exec("delete from book where id = ?", id)
+	if err != nil {
+		return err
+	}
+	return nil
+}
@@ -0,0 +1,178 @@
+package repository
+
+import (
+	"database/sql"
+	"fmt"
+	"time"
+
+	"github.com/eminetto/clean-architecture-go-v2/entity"
+)
+
+//UserMySQL mysql repo
+type UserMySQL struct {
+	db *sql.DB
+}
+
+//NewUserMySQL create new repository
+func NewUserMySQL(db *sql.DB) *UserMySQL {
+	return &UserMySQL{
+		db: db,
+	}
+}
+
+//Create an user
+func (r *UserMySQL) Create(e *entity.User) (entity.ID, error) {
+	stmt, err := r.db.Prepare(`
+		insert into user (id, email, password, first_name, last_name, created_at) 
+		values(?,?,?,?,?,?)`)
+	if err != nil {
+		return e.ID, err
+	}
+	_, err = stmt.Exec(
+		e.ID,
+		e.Email,
+		e.Password,
+		e.FirstName,
+		e.LastName,
+		time.Now().Format("2006-01-02"),
+	)
+	if err != nil {
+		return e.ID, err
+	}
+	err = stmt.Close()
+	if err != nil {
+		return e.ID, err
+	}
+	return e.ID, nil
+}
+
+//Get an user
+func (r *UserMySQL) Get(id entity.ID) (*entity.User, error) {
+	return getUser(id, r.db)
+}
+
+func getUser(id entity.ID, db *sql.DB) (*entity.User, error) {
+	stmt, err := db.Prepare(`select id, email, first_name, last_name, created_at from user where id = ?`)
+	if err != nil {
+		return nil, err
+	}
+	var u entity.User
+	rows, err := stmt.Query(id)
+	if err != nil {
+		return nil, err
+	}
+	for rows.Next() {
+		err = rows.Scan(&u.ID, &u.Email, &u.FirstName, &u.LastName, &u.CreatedAt)
+	}
+	stmt, err = db.Prepare(`select book_id from book_user where user_id = ?`)
+	if err != nil {
+		return nil, err
+	}
+	rows, err = stmt.Query(id)
+	if err != nil {
+		return nil, err
+	}
+	for rows.Next() {
+		var i entity.ID
+		err = rows.Scan(&i)
+		u.Books = append(u.Books, i)
+	}
+	return &u, nil
+}
+
+//Update an user
+func (r *UserMySQL) Update(e *entity.User) error {
+	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)
+	if err != nil {
+		return err
+	}
+	_, err = r.db.Exec("delete from book_user where user_id = ?", e.ID)
+	if err != nil {
+		return err
+	}
+	for _, b := range e.Books {
+		_, err := r.db.Exec("insert into book_user values(?,?,?)", e.ID, b, time.Now().Format("2006-01-02"))
+		if err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
+//Search users
+func (r *UserMySQL) Search(query string) ([]*entity.User, error) {
+	stmt, err := r.db.Prepare(`select id from user where name like ?`)
+	if err != nil {
+		return nil, err
+	}
+	defer stmt.Close()
+	var ids []entity.ID
+	rows, err := stmt.Query()
+	if err != nil {
+		return nil, err
+	}
+	for rows.Next() {
+		var i entity.ID
+		err = rows.Scan(&i)
+		if err != nil {
+			return nil, err
+		}
+		ids = append(ids, i)
+	}
+	if len(ids) == 0 {
+		return nil, fmt.Errorf("not found")
+	}
+	var users []*entity.User
+	for _, id := range ids {
+		u, err := getUser(id, r.db)
+		if err != nil {
+			return nil, err
+		}
+		users = append(users, u)
+	}
+	return users, nil
+}
+
+//List users
+func (r *UserMySQL) List() ([]*entity.User, error) {
+	stmt, err := r.db.Prepare(`select id from user`)
+	if err != nil {
+		return nil, err
+	}
+	defer stmt.Close()
+	var ids []entity.ID
+	rows, err := stmt.Query()
+	if err != nil {
+		return nil, err
+	}
+	for rows.Next() {
+		var i entity.ID
+		err = rows.Scan(&i)
+		if err != nil {
+			return nil, err
+		}
+		ids = append(ids, i)
+	}
+	if len(ids) == 0 {
+		return nil, fmt.Errorf("not found")
+	}
+	var users []*entity.User
+	for _, id := range ids {
+		u, err := getUser(id, r.db)
+		if err != nil {
+			return nil, err
+		}
+		users = append(users, u)
+	}
+	return users, nil
+}
+
+//Delete an user
+func (r *UserMySQL) Delete(id entity.ID) error {
+	_, err := r.db.Exec("delete from user where id = ?", id)
+	if err != nil {
+		return err
+	}
+	return nil
+}