mirror of
https://github.com/idanoo/autobrr
synced 2025-07-23 08:49:13 +00:00
feat(indexers): update release baseurl on update (#1532)
This commit is contained in:
parent
19e129e55f
commit
3202c6043d
5 changed files with 94 additions and 9 deletions
|
@ -80,7 +80,17 @@ func (r *IndexerRepo) Update(ctx context.Context, indexer domain.Indexer) (*doma
|
|||
}
|
||||
|
||||
func (r *IndexerRepo) List(ctx context.Context) ([]domain.Indexer, error) {
|
||||
rows, err := r.db.handler.QueryContext(ctx, "SELECT id, enabled, name, identifier, implementation, base_url, settings FROM indexer ORDER BY name ASC")
|
||||
queryBuilder := r.db.squirrel.
|
||||
Select("id", "enabled", "name", "identifier", "implementation", "base_url", "settings").
|
||||
From("indexer").
|
||||
OrderBy("name ASC")
|
||||
|
||||
query, _, err := queryBuilder.ToSql()
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(err, "error building query")
|
||||
}
|
||||
|
||||
rows, err := r.db.handler.QueryContext(ctx, query)
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(err, "error executing query")
|
||||
}
|
||||
|
|
|
@ -734,3 +734,53 @@ func (repo *ReleaseRepo) CanDownloadShow(ctx context.Context, title string, seas
|
|||
|
||||
return true, nil
|
||||
}
|
||||
|
||||
func (repo *ReleaseRepo) UpdateBaseURL(ctx context.Context, indexer string, oldBaseURL, newBaseURL string) error {
|
||||
tx, err := repo.db.BeginTx(ctx, &sql.TxOptions{})
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
defer func() {
|
||||
var txErr error
|
||||
if p := recover(); p != nil {
|
||||
txErr = tx.Rollback()
|
||||
if txErr != nil {
|
||||
repo.log.Error().Err(txErr).Msg("error rolling back transaction")
|
||||
}
|
||||
repo.log.Error().Msgf("something went terribly wrong panic: %v", p)
|
||||
} else if err != nil {
|
||||
txErr = tx.Rollback()
|
||||
if txErr != nil {
|
||||
repo.log.Error().Err(txErr).Msg("error rolling back transaction")
|
||||
}
|
||||
} else {
|
||||
// All good, commit
|
||||
txErr = tx.Commit()
|
||||
if txErr != nil {
|
||||
repo.log.Error().Err(txErr).Msg("error committing transaction")
|
||||
}
|
||||
}
|
||||
}()
|
||||
|
||||
queryBuilder := repo.db.squirrel.
|
||||
RunWith(tx).
|
||||
Update("release").
|
||||
Set("download_url", sq.Expr("REPLACE(download_url, ?, ?)", oldBaseURL, newBaseURL)).
|
||||
Set("info_url", sq.Expr("REPLACE(info_url, ?, ?)", oldBaseURL, newBaseURL)).
|
||||
Where(sq.Eq{"indexer": indexer})
|
||||
|
||||
result, err := queryBuilder.ExecContext(ctx)
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "error executing query")
|
||||
}
|
||||
|
||||
rowsAffected, err := result.RowsAffected()
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "error getting rows affected")
|
||||
}
|
||||
|
||||
repo.log.Trace().Msgf("release updated (%d) base urls from %q to %q", rowsAffected, oldBaseURL, newBaseURL)
|
||||
|
||||
return nil
|
||||
}
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue