feat(db): add LoadRadioCandidates query

Single SELECT that joins tracks with general_likes (for is_liked) and an aggregated LATERAL subquery on play_events (for last_played_at, play_count, skip_count). Excludes seed + tracks played in the last N hours. Drives the M3 weighted shuffle scoring.
2026-04-27 08:01:50 -04:00
parent 8c3d06c0a1
commit 3ef560e414
2 changed files with 127 additions and 0 deletions
@@ -0,0 +1,98 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+// source: recommendation.sql
+
+package dbq
+
+import (
+	"context"
+
+	"github.com/jackc/pgx/v5/pgtype"
+)
+
+const loadRadioCandidates = `-- name: LoadRadioCandidates :many
+SELECT
+    t.id, t.title, t.album_id, t.artist_id, t.track_number, t.disc_number, t.duration_ms, t.file_path, t.file_size, t.file_format, t.bitrate, t.mbid, t.genre, t.added_at, t.updated_at,
+    (l.user_id IS NOT NULL)::bool AS is_liked,
+    pe.last_played_at::timestamptz AS last_played_at,
+    pe.play_count,
+    pe.skip_count
+FROM tracks t
+LEFT JOIN general_likes l ON l.user_id = $1 AND l.track_id = t.id
+LEFT JOIN LATERAL (
+    SELECT
+        max(started_at) AS last_played_at,
+        count(*) AS play_count,
+        count(*) FILTER (WHERE was_skipped) AS skip_count
+    FROM play_events
+    WHERE user_id = $1 AND track_id = t.id
+) pe ON true
+WHERE t.id <> $2
+  AND NOT EXISTS (
+    SELECT 1 FROM play_events
+    WHERE user_id = $1 AND track_id = t.id
+      AND started_at > now() - $3 * interval '1 hour'
+  )
+`
+
+type LoadRadioCandidatesParams struct {
+	UserID  pgtype.UUID
+	ID      pgtype.UUID
+	Column3 interface{}
+}
+
+type LoadRadioCandidatesRow struct {
+	Track        Track
+	IsLiked      bool
+	LastPlayedAt pgtype.Timestamptz
+	PlayCount    int64
+	SkipCount    int64
+}
+
+// Returns all tracks except the seed and any played by the user within
+// the last $3 hours, joined with the stats needed for scoring:
+//
+//	is_liked     — boolean from general_likes for this user
+//	last_played_at — max(play_events.started_at) for this user/track
+//	play_count   — count of play_events for this user/track
+//	skip_count   — play_events where was_skipped=true
+func (q *Queries) LoadRadioCandidates(ctx context.Context, arg LoadRadioCandidatesParams) ([]LoadRadioCandidatesRow, error) {
+	rows, err := q.db.Query(ctx, loadRadioCandidates, arg.UserID, arg.ID, arg.Column3)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+	var items []LoadRadioCandidatesRow
+	for rows.Next() {
+		var i LoadRadioCandidatesRow
+		if err := rows.Scan(
+			&i.Track.ID,
+			&i.Track.Title,
+			&i.Track.AlbumID,
+			&i.Track.ArtistID,
+			&i.Track.TrackNumber,
+			&i.Track.DiscNumber,
+			&i.Track.DurationMs,
+			&i.Track.FilePath,
+			&i.Track.FileSize,
+			&i.Track.FileFormat,
+			&i.Track.Bitrate,
+			&i.Track.Mbid,
+			&i.Track.Genre,
+			&i.Track.AddedAt,
+			&i.Track.UpdatedAt,
+			&i.IsLiked,
+			&i.LastPlayedAt,
+			&i.PlayCount,
+			&i.SkipCount,
+		); err != nil {
+			return nil, err
+		}
+		items = append(items, i)
+	}
+	if err := rows.Err(); err != nil {
+		return nil, err
+	}
+	return items, nil
+}
@@ -0,0 +1,29 @@
+-- name: LoadRadioCandidates :many
+-- Returns all tracks except the seed and any played by the user within
+-- the last $3 hours, joined with the stats needed for scoring:
+--   is_liked     — boolean from general_likes for this user
+--   last_played_at — max(play_events.started_at) for this user/track
+--   play_count   — count of play_events for this user/track
+--   skip_count   — play_events where was_skipped=true
+SELECT
+    sqlc.embed(t),
+    (l.user_id IS NOT NULL)::bool AS is_liked,
+    pe.last_played_at::timestamptz AS last_played_at,
+    pe.play_count,
+    pe.skip_count
+FROM tracks t
+LEFT JOIN general_likes l ON l.user_id = $1 AND l.track_id = t.id
+LEFT JOIN LATERAL (
+    SELECT
+        max(started_at) AS last_played_at,
+        count(*) AS play_count,
+        count(*) FILTER (WHERE was_skipped) AS skip_count
+    FROM play_events
+    WHERE user_id = $1 AND track_id = t.id
+) pe ON true
+WHERE t.id <> $2
+  AND NOT EXISTS (
+    SELECT 1 FROM play_events
+    WHERE user_id = $1 AND track_id = t.id
+      AND started_at > now() - $3 * interval '1 hour'
+  );