Alexander
69 lines
1.8 KiB
Go
69 lines
1.8 KiB
Go
package database
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/jackc/pgx/v5/pgxpool"
|
|
)
|
|
|
|
type Track struct {
|
|
ID string
|
|
ExternalID string
|
|
AlbumID string
|
|
Title string
|
|
DurationMS int
|
|
ISRC string
|
|
DiscNumber int
|
|
TrackNumber int
|
|
CreatedAt time.Time
|
|
}
|
|
|
|
type TrackRepository struct {
|
|
pool *pgxpool.Pool
|
|
}
|
|
|
|
func NewTrackRepository(pool *pgxpool.Pool) *TrackRepository {
|
|
return &TrackRepository{pool: pool}
|
|
}
|
|
|
|
func (r *TrackRepository) Create(ctx context.Context, t *Track) error {
|
|
_, err := r.pool.Exec(ctx,
|
|
`INSERT INTO tracks (external_id, album_id, title, duration_ms, isrc, disc_number, track_number)
|
|
VALUES ($1, $2, $3, $4, $5, $6, $7)
|
|
ON CONFLICT (external_id) DO UPDATE SET
|
|
title = EXCLUDED.title,
|
|
duration_ms = EXCLUDED.duration_ms,
|
|
isrc = EXCLUDED.isrc,
|
|
disc_number = EXCLUDED.disc_number,
|
|
track_number = EXCLUDED.track_number`,
|
|
t.ExternalID, t.AlbumID, t.Title, t.DurationMS, t.ISRC, t.DiscNumber, t.TrackNumber,
|
|
)
|
|
if err != nil {
|
|
return fmt.Errorf("creating track: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (r *TrackRepository) GetByAlbumID(ctx context.Context, albumID string) ([]*Track, error) {
|
|
rows, err := r.pool.Query(ctx,
|
|
`SELECT id, external_id, album_id, title, duration_ms, isrc, disc_number, track_number, created_at
|
|
FROM tracks WHERE album_id = $1 ORDER BY disc_number, track_number`, albumID,
|
|
)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("listing tracks: %w", err)
|
|
}
|
|
defer rows.Close()
|
|
|
|
var tracks []*Track
|
|
for rows.Next() {
|
|
t := &Track{}
|
|
if err := rows.Scan(&t.ID, &t.ExternalID, &t.AlbumID, &t.Title, &t.DurationMS, &t.ISRC, &t.DiscNumber, &t.TrackNumber, &t.CreatedAt); err != nil {
|
|
return nil, fmt.Errorf("scanning track: %w", err)
|
|
}
|
|
tracks = append(tracks, t)
|
|
}
|
|
return tracks, nil
|
|
}
|