mirror of https://github.com/gabehf/Koito.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
110 lines
2.8 KiB
110 lines
2.8 KiB
package psql
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/gabehf/koito/internal/logger"
|
|
"github.com/gabehf/koito/internal/repository"
|
|
"github.com/jackc/pgx/v5"
|
|
)
|
|
|
|
func (d *Psql) MergeTracks(ctx context.Context, fromId, toId int32) error {
|
|
l := logger.FromContext(ctx)
|
|
l.Info().Msgf("Merging track %d into track %d", fromId, toId)
|
|
tx, err := d.conn.BeginTx(ctx, pgx.TxOptions{})
|
|
if err != nil {
|
|
l.Err(err).Msg("Failed to begin transaction")
|
|
return err
|
|
}
|
|
defer tx.Rollback(ctx)
|
|
qtx := d.q.WithTx(tx)
|
|
err = qtx.UpdateTrackIdForListens(ctx, repository.UpdateTrackIdForListensParams{
|
|
TrackID: fromId,
|
|
TrackID_2: toId,
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = qtx.CleanOrphanedEntries(ctx)
|
|
if err != nil {
|
|
l.Err(err).Msg("Failed to clean orphaned entries")
|
|
return err
|
|
}
|
|
return tx.Commit(ctx)
|
|
}
|
|
|
|
func (d *Psql) MergeAlbums(ctx context.Context, fromId, toId int32) error {
|
|
l := logger.FromContext(ctx)
|
|
l.Info().Msgf("Merging album %d into album %d", fromId, toId)
|
|
tx, err := d.conn.BeginTx(ctx, pgx.TxOptions{})
|
|
if err != nil {
|
|
l.Err(err).Msg("Failed to begin transaction")
|
|
return err
|
|
}
|
|
defer tx.Rollback(ctx)
|
|
qtx := d.q.WithTx(tx)
|
|
err = qtx.UpdateReleaseForAll(ctx, repository.UpdateReleaseForAllParams{
|
|
ReleaseID: fromId,
|
|
ReleaseID_2: toId,
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = qtx.CleanOrphanedEntries(ctx)
|
|
if err != nil {
|
|
l.Err(err).Msg("Failed to clean orphaned entries")
|
|
return err
|
|
}
|
|
return tx.Commit(ctx)
|
|
}
|
|
|
|
func (d *Psql) MergeArtists(ctx context.Context, fromId, toId int32) error {
|
|
l := logger.FromContext(ctx)
|
|
l.Info().Msgf("Merging artist %d into artist %d", fromId, toId)
|
|
tx, err := d.conn.BeginTx(ctx, pgx.TxOptions{})
|
|
if err != nil {
|
|
l.Err(err).Msg("Failed to begin transaction")
|
|
return err
|
|
}
|
|
defer tx.Rollback(ctx)
|
|
qtx := d.q.WithTx(tx)
|
|
err = qtx.DeleteConflictingArtistTracks(ctx, repository.DeleteConflictingArtistTracksParams{
|
|
ArtistID: fromId,
|
|
ArtistID_2: toId,
|
|
})
|
|
if err != nil {
|
|
l.Err(err).Msg("Failed to delete conflicting artist tracks")
|
|
return err
|
|
}
|
|
err = qtx.DeleteConflictingArtistReleases(ctx, repository.DeleteConflictingArtistReleasesParams{
|
|
ArtistID: fromId,
|
|
ArtistID_2: toId,
|
|
})
|
|
if err != nil {
|
|
l.Err(err).Msg("Failed to delete conflicting artist releases")
|
|
return err
|
|
}
|
|
err = qtx.UpdateArtistTracks(ctx, repository.UpdateArtistTracksParams{
|
|
ArtistID: fromId,
|
|
ArtistID_2: toId,
|
|
})
|
|
if err != nil {
|
|
l.Err(err).Msg("Failed to update artist tracks")
|
|
return err
|
|
}
|
|
err = qtx.UpdateArtistReleases(ctx, repository.UpdateArtistReleasesParams{
|
|
ArtistID: fromId,
|
|
ArtistID_2: toId,
|
|
})
|
|
if err != nil {
|
|
l.Err(err).Msg("Failed to update artist releases")
|
|
return err
|
|
}
|
|
err = qtx.CleanOrphanedEntries(ctx)
|
|
if err != nil {
|
|
l.Err(err).Msg("Failed to clean orphaned entries")
|
|
return err
|
|
}
|
|
return tx.Commit(ctx)
|
|
}
|