gabehf/Koito
1
0
Fork 0
mirror of https://github.com/gabehf/Koito.git synced 2026-04-22 20:11:50 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: backfill durations from musicbrainz

Browse source
This commit is contained in:
Gabe Farrell 2026-01-14 00:01:31 -05:00
parent 288d04d714
commit 2a73856f39
6 changed files with 207 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

1
internal/db/db.go
View file

@ -16,6 +16,7 @@ type DB interface {
GetAlbum(ctx context.Context, opts GetAlbumOpts) (*models.Album, error)
GetAlbumWithNoMbzIDByTitles(ctx context.Context, artistId int32, titles []string) (*models.Album, error)
GetTrack(ctx context.Context, opts GetTrackOpts) (*models.Track, error)
GetTracksWithNoDurationButHaveMbzID(ctx context.Context, from int32) ([]*models.Track, error)
GetArtistsForAlbum(ctx context.Context, id int32) ([]*models.Artist, error)
GetArtistsForTrack(ctx context.Context, id int32) ([]*models.Artist, error)
GetTopTracksPaginated(ctx context.Context, opts GetItemsOpts) (*PaginatedResponse[*models.Track], error)

26
internal/db/psql/track.go
View file

@ -375,3 +375,29 @@ func (d *Psql) SetPrimaryTrackArtist(ctx context.Context, id int32, artistId int
}
return tx.Commit(ctx)
}
// returns nil, nil when no results
func (d *Psql) GetTracksWithNoDurationButHaveMbzID(ctx context.Context, from int32) ([]*models.Track, error) {
results, err := d.q.GetTracksWithNoDurationButHaveMbzID(ctx, repository.GetTracksWithNoDurationButHaveMbzIDParams{
Limit: 20,
ID: 0,
})
if errors.Is(err, pgx.ErrNoRows) {
return nil, nil
} else if err != nil {
return nil, fmt.Errorf("GetTracksWithNoDurationButHaveMbzID: %w", err)
}
ret := make([]*models.Track, 0)
for _, v := range results {
ret = append(ret, &models.Track{
ID: v.ID,
Duration: v.Duration,
MbzID: v.MusicBrainzID,
Title: v.Title,
})
}
return ret, nil
}