fix(downloadclients): remove from filter actions on delete (#891)

fix(downloadclients): properly delete from filter actions
This commit is contained in:
ze0s 2023-05-01 01:18:46 +02:00 committed by GitHub
parent 91e4f5b435
commit 64900c4d56
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 59 additions and 6 deletions

View file

@ -41,12 +41,15 @@ func (r *ActionRepo) FindByFilterID(ctx context.Context, filterID int) ([]*domai
}
for _, action := range actions {
if action.ClientID != 0 {
if action.ClientID > 0 {
client, err := r.attachDownloadClient(ctx, tx, action.ClientID)
if err != nil {
return nil, err
}
action.Client = *client
if client != nil {
action.Client = client
}
}
}
@ -180,6 +183,10 @@ func (r *ActionRepo) attachDownloadClient(ctx context.Context, tx *Tx, clientID
var settingsJsonStr string
if err := row.Scan(&client.ID, &client.Name, &client.Type, &client.Enabled, &client.Host, &client.Port, &client.TLS, &client.TLSSkipVerify, &client.Username, &client.Password, &settingsJsonStr); err != nil {
if err == sql.ErrNoRows {
r.log.Warn().Msgf("no download client with id %d", clientID)
return nil, nil
}
return nil, errors.Wrap(err, "error scanning row")
}
@ -608,7 +615,6 @@ func (r *ActionRepo) StoreFilterActions(ctx context.Context, actions []*domain.A
if err := tx.Commit(); err != nil {
return nil, errors.Wrap(err, "error updating filter actions")
}
return actions, nil

View file

@ -254,6 +254,31 @@ func (r *DownloadClientRepo) Update(ctx context.Context, client domain.DownloadC
}
func (r *DownloadClientRepo) Delete(ctx context.Context, clientID int) error {
tx, err := r.db.BeginTx(ctx, &sql.TxOptions{Isolation: sql.LevelWriteCommitted})
if err != nil {
return err
}
defer tx.Rollback()
if err := r.delete(ctx, tx, clientID); err != nil {
return errors.Wrap(err, "error deleting download client: %d", clientID)
}
if err := r.deleteClientFromAction(ctx, tx, clientID); err != nil {
return errors.Wrap(err, "error deleting download client: %d", clientID)
}
if err := tx.Commit(); err != nil {
return errors.Wrap(err, "error deleting download client: %d", clientID)
}
r.log.Info().Msgf("delete download client: %d", clientID)
return nil
}
func (r *DownloadClientRepo) delete(ctx context.Context, tx *Tx, clientID int) error {
queryBuilder := r.db.squirrel.
Delete("client").
Where(sq.Eq{"id": clientID})
@ -263,7 +288,7 @@ func (r *DownloadClientRepo) Delete(ctx context.Context, clientID int) error {
return errors.Wrap(err, "error building query")
}
res, err := r.db.handler.ExecContext(ctx, query, args...)
res, err := tx.ExecContext(ctx, query, args...)
if err != nil {
return errors.Wrap(err, "error executing query")
}
@ -276,7 +301,29 @@ func (r *DownloadClientRepo) Delete(ctx context.Context, clientID int) error {
return errors.New("no rows affected")
}
r.log.Info().Msgf("delete download client: %d", clientID)
r.log.Debug().Msgf("delete download client: %d", clientID)
return nil
}
func (r *DownloadClientRepo) deleteClientFromAction(ctx context.Context, tx *Tx, clientID int) error {
var err error
queryBuilder := r.db.squirrel.
Update("action").
Set("enabled", false).
Set("client_id", 0).
Where(sq.Eq{"client_id": clientID}).
Suffix("RETURNING filter_id").RunWith(tx)
// return values
var filterID int
if err = queryBuilder.QueryRowContext(ctx).Scan(&filterID); err != nil {
return errors.Wrap(err, "error executing query")
}
r.log.Debug().Msgf("deleting download client %d from action for filter %d", clientID, filterID)
return nil
}

View file

@ -49,7 +49,7 @@ type Action struct {
WebhookHeaders []string `json:"webhook_headers,omitempty"`
FilterID int `json:"filter_id,omitempty"`
ClientID int32 `json:"client_id,omitempty"`
Client DownloadClient `json:"client,omitempty"`
Client *DownloadClient `json:"client,omitempty"`
}
// ParseMacros parse all macros on action