git/watch-tool
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: include transaktions for writing in db to improve security and prevent data loss

Browse source
This commit is contained in:
Patryk Hegenberg 2025-09-24 11:43:53 +02:00
parent 2f20c58400
commit 553a85562b
1 changed files with 16 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

17
local_storageV2.go
View file

@ -251,6 +251,11 @@ func (s *SQLiteStorage) MarkAsExported(ctx context.Context, ids []int64) error {
return nil return nil
} }
tx, err := s.db.BeginTx(ctx, nil)
if err != nil {
return err
}
placeholders := strings.Repeat("?,", len(ids)) placeholders := strings.Repeat("?,", len(ids))
placeholders = placeholders[:len(placeholders)-1] placeholders = placeholders[:len(placeholders)-1]
@ -261,7 +266,17 @@ func (s *SQLiteStorage) MarkAsExported(ctx context.Context, ids []int64) error {
args[i] = id args[i] = id
} }
_, err := s.db.ExecContext(ctx, sqlQuery, args...) _, err = tx.ExecContext(ctx, sqlQuery, args...)
if err != nil {
tx.Rollback()
return err
}
err = tx.Commit()
if err != nil {
return err
}
return err return err
} }