77 lines
2.9 KiB
Go
77 lines
2.9 KiB
Go
package data
|
|
|
|
import (
|
|
"database/sql"
|
|
"time"
|
|
|
|
"github.com/lib/pq"
|
|
"greenlight.debuggingjon.dev/internal/validator"
|
|
)
|
|
|
|
// Define a MovieModel struct type which wraps a sql.DB connection pool.
|
|
type MovieModel struct {
|
|
DB *sql.DB
|
|
}
|
|
|
|
// Add a placeholder method for inserting a new record in the movies table.
|
|
func (m MovieModel) Insert(movie *Movie) error {
|
|
// Define the SQL query for inserting a new record in the movies table and returning
|
|
// the system-generated data.
|
|
query := `
|
|
INSERT INTO movies (title, year, runtime, genres)
|
|
VALUES ($1, $2, $3, $4)
|
|
RETURNING id, created_at, version`
|
|
|
|
// Create an args slice containing the values for the placeholder parameters from
|
|
// the movie struct. Declaring this slice immediately next to our SQL query helps to
|
|
// make it nice and clear *what values are being used where* in the query.
|
|
args := []interface{}{movie.Title, movie.Year, movie.Runtime, pq.Array(movie.Genres)}
|
|
|
|
// Use the QueryRow() method to execute the SQL query on our connection pool,
|
|
// passing in the args slice as a variadic parameter and scanning the system-
|
|
// generated id, created_at and version values into the movie struct.
|
|
return m.DB.QueryRow(query, args...).Scan(&movie.ID, &movie.CreatedAt, &movie.Version)
|
|
}
|
|
|
|
// Add a placeholder method for fetching a specific record from the movies table.
|
|
func (m MovieModel) Get(id int64) (*Movie, error) {
|
|
return nil, nil
|
|
}
|
|
|
|
// Add a placeholder method for updating a specific record in the “movies table.
|
|
func (m MovieModel) Update(movie *Movie) error {
|
|
return nil
|
|
}
|
|
|
|
// Add a placeholder method for deleting a specific record from the movies table.
|
|
func (m MovieModel) Delete(id int64) error {
|
|
return nil
|
|
}
|
|
|
|
type Movie struct {
|
|
ID int64 `json:"id"`
|
|
CreatedAt time.Time `json:"-"` // Use the - directive
|
|
Title string `json:"title"`
|
|
Year int32 `json:"year,omitempty"` // Add the omitempty directive
|
|
Runtime Runtime `json:"runtime,omitempty"` // Add the omitempty directive
|
|
Genres []string `json:"genres,omitempty"` // Add the omitempty directive
|
|
Version int32 `json:"version"`
|
|
}
|
|
|
|
func ValidateMovie(v *validator.Validator, movie *Movie) {
|
|
v.Check(movie.Title != "", "title", "must be provided")
|
|
v.Check(len(movie.Title) <= 500, "title", "must not be more than 500 bytes long")
|
|
|
|
v.Check(movie.Year != 0, "year", "must be provided")
|
|
v.Check(movie.Year >= 1888, "year", "must be greater than 1888")
|
|
v.Check(movie.Year <= int32(time.Now().Year()), "year", "must not be in the future")
|
|
|
|
v.Check(movie.Runtime != 0, "runtime", "must be provided")
|
|
v.Check(movie.Runtime > 0, "runtime", "must be a positive integer")
|
|
|
|
v.Check(movie.Genres != nil, "genres", "must be provided")
|
|
v.Check(len(movie.Genres) >= 1, "genres", "must contain at least 1 genre")
|
|
v.Check(len(movie.Genres) <= 5, "genres", "must not contain more than 5 genres")
|
|
v.Check(validator.Unique(movie.Genres), "genres", "must not contain duplicate values")
|
|
}
|