minstrel/internal/db/dbq/loudness.sql.go

// Code generated by sqlc. DO NOT EDIT.
// versions:
//   sqlc v1.31.1
// source: loudness.sql

package dbq

import (
	"context"

	"github.com/jackc/pgx/v5/pgtype"
)

const deleteOrphanAlbumLoudness = `-- name: DeleteOrphanAlbumLoudness :execrows
DELETE FROM album_loudness a
 WHERE NOT EXISTS (
     SELECT 1 FROM tracks t WHERE t.album_id = a.album_id AND t.missing_since IS NULL
 )
`

// Album rows whose album has no present track left (every file missing). The
// album row itself survives a missing file, so its loudness would otherwise
// stay behind describing tracks that are gone; it is recomputed if they return.
func (q *Queries) DeleteOrphanAlbumLoudness(ctx context.Context) (int64, error) {
	result, err := q.db.Exec(ctx, deleteOrphanAlbumLoudness)
	if err != nil {
		return 0, err
	}
	return result.RowsAffected(), nil
}

const deleteTrackLoudness = `-- name: DeleteTrackLoudness :exec
DELETE FROM track_loudness WHERE track_id = $1
`

// The scan saw new bytes at this path. The stored measurement describes the old
// ones, so it goes, and the backfill measures the file again.
func (q *Queries) DeleteTrackLoudness(ctx context.Context, trackID pgtype.UUID) error {
	_, err := q.db.Exec(ctx, deleteTrackLoudness, trackID)
	return err
}

const getLoudnessCoverage = `-- name: GetLoudnessCoverage :one
SELECT count(*)::bigint AS total,
       count(*) FILTER (
           WHERE l.analysis_version >= $1
             AND l.integrated_lufs IS NOT NULL
       )::bigint AS measured,
       count(*) FILTER (
           WHERE l.analysis_version >= $1
             AND l.integrated_lufs IS NULL AND NOT l.unreadable
       )::bigint AS silent,
       count(*) FILTER (
           WHERE l.analysis_version >= $1
             AND l.unreadable
       )::bigint AS unreadable,
       count(*) FILTER (
           WHERE l.track_id IS NULL OR l.analysis_version < $1
       )::bigint AS pending
  FROM tracks t
  LEFT JOIN track_loudness l ON l.track_id = t.id
 WHERE t.missing_since IS NULL
`

type GetLoudnessCoverageRow struct {
	Total      int64
	Measured   int64
	Silent     int64
	Unreadable int64
	Pending    int64
}

// The admin gauge. measured + silent + unreadable + pending = total. Missing
// tracks are excluded, or the gauge could never reach the end.
func (q *Queries) GetLoudnessCoverage(ctx context.Context, currentVersion int16) (GetLoudnessCoverageRow, error) {
	row := q.db.QueryRow(ctx, getLoudnessCoverage, currentVersion)
	var i GetLoudnessCoverageRow
	err := row.Scan(
		&i.Total,
		&i.Measured,
		&i.Silent,
		&i.Unreadable,
		&i.Pending,
	)
	return i, err
}

const getLoudnessSettings = `-- name: GetLoudnessSettings :one
SELECT id, enabled, backfill_concurrency, updated_at FROM loudness_settings WHERE id = true
`

func (q *Queries) GetLoudnessSettings(ctx context.Context) (LoudnessSetting, error) {
	row := q.db.QueryRow(ctx, getLoudnessSettings)
	var i LoudnessSetting
	err := row.Scan(
		&i.ID,
		&i.Enabled,
		&i.BackfillConcurrency,
		&i.UpdatedAt,
	)
	return i, err
}

const listAlbumLoudnessInputs = `-- name: ListAlbumLoudnessInputs :many
SELECT t.id,
       (l.track_id IS NOT NULL)::boolean AS settled,
       l.true_peak_dbtp,
       l.block_hist_start,
       l.block_hist
  FROM tracks t
  LEFT JOIN track_loudness l
         ON l.track_id = t.id AND l.analysis_version >= $1
 WHERE t.album_id = $2
   AND t.missing_since IS NULL
`

type ListAlbumLoudnessInputsParams struct {
	CurrentVersion int16
	AlbumID        pgtype.UUID
}

type ListAlbumLoudnessInputsRow struct {
	ID             pgtype.UUID
	Settled        bool
	TruePeakDbtp   *float32
	BlockHistStart *int16
	BlockHist      []int32
}

// Every present track on one album with its current measurement, if any.
// settled is false for a track not yet measured at the current version.
func (q *Queries) ListAlbumLoudnessInputs(ctx context.Context, arg ListAlbumLoudnessInputsParams) ([]ListAlbumLoudnessInputsRow, error) {
	rows, err := q.db.Query(ctx, listAlbumLoudnessInputs, arg.CurrentVersion, arg.AlbumID)
	if err != nil {
		return nil, err
	}
	defer rows.Close()
	var items []ListAlbumLoudnessInputsRow
	for rows.Next() {
		var i ListAlbumLoudnessInputsRow
		if err := rows.Scan(
			&i.ID,
			&i.Settled,
			&i.TruePeakDbtp,
			&i.BlockHistStart,
			&i.BlockHist,
		); err != nil {
			return nil, err
		}
		items = append(items, i)
	}
	if err := rows.Err(); err != nil {
		return nil, err
	}
	return items, nil
}

const listAlbumsNeedingLoudness = `-- name: ListAlbumsNeedingLoudness :many
WITH present AS (
    SELECT t.album_id,
           md5(string_agg(
               t.id::text || ':' || coalesce(
                   l.analysis_version::text || '@' || l.analyzed_at::text, '-'),
               ',' ORDER BY t.id)) AS digest
      FROM tracks t
      LEFT JOIN track_loudness l
             ON l.track_id = t.id AND l.analysis_version >= $3::smallint
     WHERE t.missing_since IS NULL
     GROUP BY t.album_id
)
SELECT c.album_id, c.digest::text AS digest
  FROM present c
  LEFT JOIN album_loudness a ON a.album_id = c.album_id
 WHERE (a.album_id IS NULL OR a.inputs_digest <> c.digest)
   -- Casts: sqlc cannot infer a parameter's type through a CTE alias.
   AND c.album_id > $1::uuid
 ORDER BY c.album_id
 LIMIT $2::integer
`

type ListAlbumsNeedingLoudnessParams struct {
	AfterID        pgtype.UUID
	BatchLimit     int32
	CurrentVersion int16
}

type ListAlbumsNeedingLoudnessRow struct {
	AlbumID pgtype.UUID
	Digest  string
}

// The album pass's work queue (#4996): albums whose present tracks or their
// measurements have changed since album loudness was last computed, or that
// never had it. The digest is over every present track's id and the
// measurement it holds at the current version ('-' for none), so a track
// joining, leaving or being re-measured changes it. Keyset-paged on album id
// so a pass ends even if storing one album keeps failing.
func (q *Queries) ListAlbumsNeedingLoudness(ctx context.Context, arg ListAlbumsNeedingLoudnessParams) ([]ListAlbumsNeedingLoudnessRow, error) {
	rows, err := q.db.Query(ctx, listAlbumsNeedingLoudness, arg.AfterID, arg.BatchLimit, arg.CurrentVersion)
	if err != nil {
		return nil, err
	}
	defer rows.Close()
	var items []ListAlbumsNeedingLoudnessRow
	for rows.Next() {
		var i ListAlbumsNeedingLoudnessRow
		if err := rows.Scan(&i.AlbumID, &i.Digest); err != nil {
			return nil, err
		}
		items = append(items, i)
	}
	if err := rows.Err(); err != nil {
		return nil, err
	}
	return items, nil
}

const listTracksNeedingLoudness = `-- name: ListTracksNeedingLoudness :many
SELECT t.id, t.file_path, t.duration_ms
  FROM tracks t
  LEFT JOIN track_loudness l ON l.track_id = t.id
 WHERE t.missing_since IS NULL
   AND (l.track_id IS NULL OR l.analysis_version < $1)
   AND t.id > $2
 ORDER BY t.id
 LIMIT $3
`

type ListTracksNeedingLoudnessParams struct {
	CurrentVersion int16
	AfterID        pgtype.UUID
	BatchLimit     int32
}

type ListTracksNeedingLoudnessRow struct {
	ID         pgtype.UUID
	FilePath   string
	DurationMs int32
}

// The backfill's work queue: tracks with no measurement, or one taken by an
// older method. Keyset-paged on id so a pass visits each track at most once;
// an inconclusive attempt writes no row, and without the cursor a file that
// keeps timing out would be listed again straight away. Missing tracks are
// skipped: there is no file to read.
func (q *Queries) ListTracksNeedingLoudness(ctx context.Context, arg ListTracksNeedingLoudnessParams) ([]ListTracksNeedingLoudnessRow, error) {
	rows, err := q.db.Query(ctx, listTracksNeedingLoudness, arg.CurrentVersion, arg.AfterID, arg.BatchLimit)
	if err != nil {
		return nil, err
	}
	defer rows.Close()
	var items []ListTracksNeedingLoudnessRow
	for rows.Next() {
		var i ListTracksNeedingLoudnessRow
		if err := rows.Scan(&i.ID, &i.FilePath, &i.DurationMs); err != nil {
			return nil, err
		}
		items = append(items, i)
	}
	if err := rows.Err(); err != nil {
		return nil, err
	}
	return items, nil
}

const updateLoudnessSettings = `-- name: UpdateLoudnessSettings :one
UPDATE loudness_settings
   SET enabled              = $1,
       backfill_concurrency = $2,
       updated_at           = now()
 WHERE id = true
RETURNING id, enabled, backfill_concurrency, updated_at
`

type UpdateLoudnessSettingsParams struct {
	Enabled             bool
	BackfillConcurrency int32
}

// Whole-row write from the admin card; migration 0065's CHECK is the backstop
// behind the service's own validation.
func (q *Queries) UpdateLoudnessSettings(ctx context.Context, arg UpdateLoudnessSettingsParams) (LoudnessSetting, error) {
	row := q.db.QueryRow(ctx, updateLoudnessSettings, arg.Enabled, arg.BackfillConcurrency)
	var i LoudnessSetting
	err := row.Scan(
		&i.ID,
		&i.Enabled,
		&i.BackfillConcurrency,
		&i.UpdatedAt,
	)
	return i, err
}

const upsertAlbumLoudness = `-- name: UpsertAlbumLoudness :exec
INSERT INTO album_loudness (
    album_id, integrated_lufs, true_peak_dbtp, tracks_total, tracks_settled, inputs_digest
) VALUES (
    $1, $2, $3,
    $4, $5, $6
)
ON CONFLICT (album_id) DO UPDATE SET
    integrated_lufs = EXCLUDED.integrated_lufs,
    true_peak_dbtp  = EXCLUDED.true_peak_dbtp,
    tracks_total    = EXCLUDED.tracks_total,
    tracks_settled  = EXCLUDED.tracks_settled,
    inputs_digest   = EXCLUDED.inputs_digest,
    computed_at     = now()
`

type UpsertAlbumLoudnessParams struct {
	AlbumID        pgtype.UUID
	IntegratedLufs *float32
	TruePeakDbtp   *float32
	TracksTotal    int32
	TracksSettled  int32
	InputsDigest   string
}

func (q *Queries) UpsertAlbumLoudness(ctx context.Context, arg UpsertAlbumLoudnessParams) error {
	_, err := q.db.Exec(ctx, upsertAlbumLoudness,
		arg.AlbumID,
		arg.IntegratedLufs,
		arg.TruePeakDbtp,
		arg.TracksTotal,
		arg.TracksSettled,
		arg.InputsDigest,
	)
	return err
}

const upsertTrackLoudness = `-- name: UpsertTrackLoudness :exec
INSERT INTO track_loudness (
    track_id, integrated_lufs, true_peak_dbtp, loudness_range_lu,
    block_hist_start, block_hist, unreadable, analysis_version
) VALUES (
    $1, $2, $3,
    $4, $5, $6,
    $7, $8
)
ON CONFLICT (track_id) DO UPDATE SET
    integrated_lufs   = EXCLUDED.integrated_lufs,
    true_peak_dbtp    = EXCLUDED.true_peak_dbtp,
    loudness_range_lu = EXCLUDED.loudness_range_lu,
    block_hist_start  = EXCLUDED.block_hist_start,
    block_hist        = EXCLUDED.block_hist,
    unreadable        = EXCLUDED.unreadable,
    analysis_version  = EXCLUDED.analysis_version,
    analyzed_at       = now()
`

type UpsertTrackLoudnessParams struct {
	TrackID         pgtype.UUID
	IntegratedLufs  *float32
	TruePeakDbtp    *float32
	LoudnessRangeLu *float32
	BlockHistStart  *int16
	BlockHist       []int32
	Unreadable      bool
	AnalysisVersion int16
}

// Written when the backfill measures a track (#4995). Replaces the row
// wholesale: a measurement of the old bytes has no standing once the file has
// changed.
func (q *Queries) UpsertTrackLoudness(ctx context.Context, arg UpsertTrackLoudnessParams) error {
	_, err := q.db.Exec(ctx, upsertTrackLoudness,
		arg.TrackID,
		arg.IntegratedLufs,
		arg.TruePeakDbtp,
		arg.LoudnessRangeLu,
		arg.BlockHistStart,
		arg.BlockHist,
		arg.Unreadable,
		arg.AnalysisVersion,
	)
	return err
}