103 lines
3.0 KiB
Go
103 lines
3.0 KiB
Go
package genius
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
const (
|
|
artistImageBatch = 20
|
|
artistImageRecheck = 30 * 24 * time.Hour
|
|
)
|
|
|
|
// artistSource — то, что нужно проходу по аватаркам от Genius-клиента.
|
|
type artistSource interface {
|
|
Search(ctx context.Context, query string) ([]Song, error)
|
|
Artist(ctx context.Context, id int64) (Artist, error)
|
|
}
|
|
|
|
// artistImage ищет исполнителя по имени: берёт песню, у которой primary_artist совпал с именем,
|
|
// и аватарку из неё либо из отдельного запроса артиста. Пустая строка — не нашли.
|
|
func artistImage(ctx context.Context, src artistSource, name string) (string, error) {
|
|
songs, err := src.Search(ctx, name)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
for _, song := range songs {
|
|
if !sameArtist(song.Artist.Name, name) {
|
|
continue
|
|
}
|
|
if img := imageURL(song.Artist.ImageURL); img != "" {
|
|
return img, nil
|
|
}
|
|
if song.Artist.ID <= 0 {
|
|
continue
|
|
}
|
|
a, err := src.Artist(ctx, song.Artist.ID)
|
|
if err != nil {
|
|
var he *HTTPError
|
|
if errors.As(err, &he) && he.StatusCode == 404 {
|
|
continue
|
|
}
|
|
return "", err
|
|
}
|
|
return imageURL(a.ImageURL), nil
|
|
}
|
|
return "", nil
|
|
}
|
|
|
|
// fillArtistImages проходит по исполнителям без аватарки; каждого проверяет раз в artistImageRecheck.
|
|
func (s *Service) fillArtistImages(ctx context.Context) error {
|
|
var active bool
|
|
if err := s.pool.QueryRow(ctx, `SELECT EXISTS(SELECT 1 FROM genius_state WHERE NOT paused AND cooldown_until<=now())`).Scan(&active); err != nil || !active {
|
|
return err
|
|
}
|
|
rows, err := s.pool.Query(ctx, `SELECT id, name FROM artists
|
|
WHERE image_path IS NULL AND genius_image_url IS NULL
|
|
AND (genius_image_checked_at IS NULL OR genius_image_checked_at < now()-$1::interval)
|
|
ORDER BY genius_image_checked_at NULLS FIRST LIMIT $2`, artistImageRecheck.String(), artistImageBatch)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
type row struct {
|
|
id uuid.UUID
|
|
name string
|
|
}
|
|
var batch []row
|
|
for rows.Next() {
|
|
var r row
|
|
if err := rows.Scan(&r.id, &r.name); err != nil {
|
|
rows.Close()
|
|
return err
|
|
}
|
|
batch = append(batch, r)
|
|
}
|
|
rows.Close()
|
|
if err := rows.Err(); err != nil {
|
|
return err
|
|
}
|
|
for _, r := range batch {
|
|
if ctx.Err() != nil {
|
|
return ctx.Err()
|
|
}
|
|
reqCtx, cancel := context.WithTimeout(ctx, time.Minute)
|
|
img, err := artistImage(reqCtx, s.client, r.name)
|
|
cancel()
|
|
if err != nil {
|
|
var he *HTTPError
|
|
if errors.As(err, &he) && he.StatusCode == 429 {
|
|
_, _ = s.pool.Exec(ctx, `UPDATE genius_state SET cooldown_until=GREATEST(cooldown_until,$1)`, time.Now().Add(retryDelay(he.RetryAfter)))
|
|
}
|
|
return err
|
|
}
|
|
if _, err := s.pool.Exec(ctx, `UPDATE artists SET genius_image_checked_at=now(),
|
|
genius_image_url=COALESCE(genius_image_url, NULLIF($2,'')) WHERE id=$1`, r.id, img); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|