82 lines
2.1 KiB
Go
82 lines
2.1 KiB
Go
package mysql
|
|
|
|
import (
|
|
"database/sql"
|
|
"errors"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/davidlick/bookish/bookish-server/models"
|
|
"github.com/gobuffalo/nulls"
|
|
"github.com/gobuffalo/uuid"
|
|
)
|
|
|
|
// IsBookAvailable queries the database for available books.
|
|
func (s *store) IsBookAvailable(title string) (available bool, err error) {
|
|
rr := []models.Rental{}
|
|
err = s.Where(fmt.Sprintf("book_title = '%s'", title)).Where("return_date IS NULL").All(&rr)
|
|
if err != nil {
|
|
// If no records are returned then the book is available for rent.
|
|
if errors.Is(sql.ErrNoRows, err) {
|
|
return true, nil
|
|
}
|
|
|
|
return false, err
|
|
}
|
|
|
|
// If 5 books are rented out this book is not available.
|
|
if len(rr) > 4 {
|
|
return false, nil
|
|
}
|
|
|
|
return true, nil
|
|
}
|
|
|
|
// RenterAlreadyCheckedOut queries the database for the given renterId and title. If a record is returned checkedOut will be true.
|
|
func (s *store) RenterAlreadyCheckedOut(renterId uuid.UUID, title string) (checkedOut bool, err error) {
|
|
rr := []models.Rental{}
|
|
err = s.Where(fmt.Sprintf("renter_id = '%s'", renterId)).
|
|
Where(fmt.Sprintf("book_title = '%s'", title)).
|
|
Where("return_date IS NULL").
|
|
All(&rr)
|
|
if err != nil {
|
|
// If no records are returned then the book is available for rent.
|
|
if errors.Is(sql.ErrNoRows, err) {
|
|
return false, nil
|
|
}
|
|
|
|
return false, err
|
|
}
|
|
|
|
// If 5 books are rented out this book is not available.
|
|
if len(rr) == 0 {
|
|
return false, nil
|
|
}
|
|
|
|
return true, nil
|
|
}
|
|
|
|
// CheckoutBook inserts a record indicating the book is checked out to the renter.
|
|
func (s *store) CheckoutBook(renterId uuid.UUID, title string) error {
|
|
r := models.Rental{BookTitle: title, RenterID: renterId, RentalDate: time.Now()}
|
|
err := s.Create(&r)
|
|
return err
|
|
}
|
|
|
|
// ReturnBook sets the return_date field to now indicating the book has been returned.
|
|
func (s *store) ReturnBook(renterId uuid.UUID, title string) error {
|
|
r := models.Rental{}
|
|
err := s.Where(fmt.Sprintf("renter_id = '%s'", renterId)).
|
|
Where(fmt.Sprintf("book_title = '%s'", title)).
|
|
Where("return_date IS NULL").
|
|
First(&r)
|
|
if err != nil {
|
|
return ErrNotFound
|
|
}
|
|
|
|
r.ReturnDate = nulls.Time{Time: time.Now(), Valid: true}
|
|
|
|
err = s.Save(&r)
|
|
return err
|
|
}
|