1
Fork 0
photoview/api/graphql/models/photo.go

64 lines
1.1 KiB
Go

package models
import (
"database/sql"
"strconv"
)
type Photo struct {
PhotoID int
Title string
Path string
AlbumId int
ExifId *int
}
type PhotoPurpose string
const (
PhotoThumbnail PhotoPurpose = "thumbnail"
PhotoHighRes PhotoPurpose = "high-res"
PhotoOriginal PhotoPurpose = "original"
)
type PhotoURL struct {
UrlID int
PhotoId int
PhotoName string
Width int
Height int
purpose PhotoPurpose
}
func (p *Photo) ID() string {
return strconv.Itoa(p.PhotoID)
}
func NewPhotoFromRow(row *sql.Row) (*Photo, error) {
photo := Photo{}
if err := row.Scan(&photo.PhotoID, &photo.Title, &photo.Path, &photo.AlbumId, &photo.ExifId); err != nil {
return nil, err
}
return &photo, nil
}
func NewPhotosFromRows(rows *sql.Rows) ([]*Photo, error) {
photos := make([]*Photo, 0)
for rows.Next() {
var photo Photo
if err := rows.Scan(&photo.PhotoID, &photo.Title, &photo.Path, &photo.AlbumId, &photo.ExifId); err != nil {
return nil, err
}
photos = append(photos, &photo)
}
return photos, nil
}
func (p *PhotoURL) URL() string {
return "URL:" + p.PhotoName
}