Library/models/book.go

91 lines
2.2 KiB
Go
Raw Normal View History

package models
import (
"fmt"
)
2017-11-08 09:55:17 +00:00
// Book holds a book
type Book struct {
2017-11-07 15:35:10 +00:00
ID int64 `xorm:"int(11) autoincr not null unique pk"`
Title string `xorm:"varchar(250) not null"`
Isbn string `xorm:"varchar(30)"`
Year int64 `xorm:"int(11)"`
Price float64 `xorm:"double"`
Status int64 `xorm:"int(11)"`
Publisher int64 `xorm:"int(11)"`
Created int64 `xorm:"created"`
Updated int64 `xorm:"updated"`
Quantity int64 `xorm:"-"`
PublisherFull Publisher `xorm:"-"`
2017-11-07 15:35:10 +00:00
Authors []Author `xorm:"-"`
}
2017-11-08 09:55:17 +00:00
// TableName returns the name for the books table
2017-11-07 15:35:10 +00:00
func (Book) TableName() string {
return "books"
}
2017-11-16 11:51:23 +00:00
// Quantity is the quantity for a book
type Quantity struct {
ID int64 `xorm:"int(11) autoincr not null unique pk"`
BookID int64 `xorm:"int(11) not null"`
Quantity int64 `xorm:"int(11) not null"`
Created int64 `xorm:"created"`
}
// TableName returns the name for the quantites table
func (Quantity) TableName() string {
return "books_quantities"
}
// GetQuantityByBook returns the current quantity for a book
func GetQuantityByBook(bookID int64) (quantity int64, err error) {
bq := Quantity{BookID: bookID}
has, err := x.Desc("id").Get(&bq)
if err != nil {
return 0, err
}
if has {
quantity = bq.Quantity
}
return quantity, nil
}
// SetBookQuantity sets a new quantity for a book
func (book Book) setBookQuantity(quantity int64) (err error) {
q := Quantity{BookID: book.ID, Quantity: quantity}
_, err = x.Insert(q)
return err
}
2017-11-08 09:55:17 +00:00
// GetBookByID gets a Book by its ID
func GetBookByID(ID int64) (book Book, exists bool, err error) {
// Get the Book
has, err := x.ID(ID).Get(&book)
if has {
// Get the books quantity. We can't join it because xorm ignores the Quantity option in struct
book.Quantity, err = GetQuantityByBook(ID)
if err != nil {
fmt.Println("Error getting quantity:", err)
}
// Get publisher. We can't join it because xorm ignores the PublisherID option in struct
book.PublisherFull, _, err = GetPublisherByID(book.Publisher)
if err != nil {
fmt.Println("Error getting publisher:", err)
}
// Get all authors
book.Authors, err = GetAuthorsByBook(book.ID)
if err != nil {
fmt.Println("Error getting authors:", err)
}
}
return book, has, err
2017-11-07 15:35:10 +00:00
}