49 lines
884 B
Go
49 lines
884 B
Go
package models
|
|
|
|
import (
|
|
"database/sql"
|
|
)
|
|
|
|
type Album struct {
|
|
AlbumID int
|
|
Title string
|
|
ParentAlbum *int
|
|
OwnerID int
|
|
Path string
|
|
PathHash string
|
|
}
|
|
|
|
func (a *Album) ID() int {
|
|
return a.AlbumID
|
|
}
|
|
|
|
func (a *Album) FilePath() string {
|
|
return a.Path
|
|
}
|
|
|
|
func NewAlbumFromRow(row *sql.Row) (*Album, error) {
|
|
album := Album{}
|
|
|
|
if err := row.Scan(&album.AlbumID, &album.Title, &album.ParentAlbum, &album.OwnerID, &album.Path, &album.PathHash); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &album, nil
|
|
}
|
|
|
|
func NewAlbumsFromRows(rows *sql.Rows) ([]*Album, error) {
|
|
albums := make([]*Album, 0)
|
|
|
|
for rows.Next() {
|
|
var album Album
|
|
if err := rows.Scan(&album.AlbumID, &album.Title, &album.ParentAlbum, &album.OwnerID, &album.Path, &album.PathHash); err != nil {
|
|
return nil, err
|
|
}
|
|
albums = append(albums, &album)
|
|
}
|
|
|
|
rows.Close()
|
|
|
|
return albums, nil
|
|
}
|