86 lines
1.9 KiB
Go
86 lines
1.9 KiB
Go
package models
|
|
|
|
import (
|
|
"database/sql"
|
|
"log"
|
|
"net/url"
|
|
"strconv"
|
|
"time"
|
|
)
|
|
|
|
type SameBookingError struct{}
|
|
|
|
func (e SameBookingError) Error() string {
|
|
return "the same booking already exists!"
|
|
}
|
|
|
|
type Booking struct {
|
|
CardID string `json:"cradID"`
|
|
ReaderID string `json:"readerID"`
|
|
BookingType int `json:"bookingTyp"`
|
|
LoggedTime time.Time `json:"loggedTime"`
|
|
Id int `json:"id"`
|
|
}
|
|
|
|
func (b Booking) New(card_id string, reader_id string, booking_type int) Booking {
|
|
return Booking{
|
|
CardID: card_id,
|
|
ReaderID: reader_id,
|
|
BookingType: booking_type,
|
|
}
|
|
}
|
|
|
|
func (b *Booking) FromUrlParams(params url.Values) Booking {
|
|
bookingType, err := strconv.Atoi(params.Get("bookingType"))
|
|
if err != nil {
|
|
log.Println("Error parsing bookingType: ", err)
|
|
return b.New("", "", 0)
|
|
}
|
|
cardID := params.Get("cardID")
|
|
readerID := params.Get("readerID")
|
|
return Booking{BookingType: bookingType, CardID: cardID, ReaderID: readerID}
|
|
}
|
|
|
|
func (b Booking) Verify() bool {
|
|
if b.CardID == "" || b.ReaderID == "" || b.BookingType == 0 {
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
func (b Booking) Insert(db *sql.DB) error {
|
|
if !checkLastBooking(b, db) {
|
|
return SameBookingError{}
|
|
}
|
|
stmt, err := db.Prepare((`INSERT INTO zeiten (card_id, reader_id, booking_type) VALUES ($1, $2, $3)`))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = stmt.Query(b.CardID, b.ReaderID, b.BookingType)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func checkLastBooking(b Booking, db *sql.DB) bool {
|
|
var booking_type int
|
|
stmt, err := db.Prepare((`SELECT booking_type FROM "zeiten" WHERE "card_id" = $1 ORDER BY "logged_time" DESC LIMIT 1;`))
|
|
if err != nil {
|
|
log.Fatalf("Error preparing query: %v", err)
|
|
return false
|
|
}
|
|
err = stmt.QueryRow(b.CardID).Scan(&booking_type)
|
|
if err == sql.ErrNoRows {
|
|
return true
|
|
}
|
|
if err != nil {
|
|
log.Println("Error checking last booking: ", err)
|
|
return false
|
|
}
|
|
if booking_type == b.BookingType {
|
|
return false
|
|
}
|
|
return true
|
|
}
|