30 lines
712 B
Go
30 lines
712 B
Go
package data
|
|
|
|
import (
|
|
"database/sql"
|
|
"errors"
|
|
)
|
|
|
|
// Define a custom ErrRecordNotFound error. We'll return this from our Get() method when
|
|
// looking up a movie that doesn't exist in our database.
|
|
var (
|
|
ErrRecordNotFound = errors.New("record not found")
|
|
ErrEditConflict = errors.New("edit conflict")
|
|
)
|
|
|
|
type Models struct {
|
|
Movies MovieModel
|
|
Permissions PermissionModel // Add a new Permissions field.
|
|
Tokens TokenModel
|
|
Users UserModel
|
|
}
|
|
|
|
func NewModels(db *sql.DB) Models {
|
|
return Models{
|
|
Movies: MovieModel{DB: db},
|
|
Permissions: PermissionModel{DB: db}, // Initialize a new PermissionModel instance.
|
|
Tokens: TokenModel{DB: db},
|
|
Users: UserModel{DB: db},
|
|
}
|
|
}
|