feat: improvements in structures
This commit is contained in:
@@ -0,0 +1,127 @@
|
||||
package book
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
"time"
|
||||
|
||||
"github.com/eminetto/clean-architecture-go-v2/domain/entity"
|
||||
)
|
||||
|
||||
//MySQLRepo mysql repo
|
||||
type MySQLRepo struct {
|
||||
db *sql.DB
|
||||
}
|
||||
|
||||
//NewMySQLRepository create new repository
|
||||
func NewMySQLRepository(db *sql.DB) *MySQLRepo {
|
||||
return &MySQLRepo{
|
||||
db: db,
|
||||
}
|
||||
}
|
||||
|
||||
//Create a book
|
||||
func (r *MySQLRepo) 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 *MySQLRepo) 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 *MySQLRepo) 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 *MySQLRepo) 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 *MySQLRepo) 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 *MySQLRepo) Delete(id entity.ID) error {
|
||||
_, err := r.db.Exec("delete from book where id = ?", id)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user