Alassane
49177d742f
Co-authored-by: ayattara <alassane.yattara@savoirfairelinux.com> Co-authored-by: Carlos Florencio <carlosmflorencio@gmail.com>
78 lines
2.2 KiB
Go
78 lines
2.2 KiB
Go
package gitlab
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"errors"
|
|
"fmt"
|
|
)
|
|
|
|
var gitlabTaskOwnedTables = []string{
|
|
"gitlab_mr_watches",
|
|
"gitlab_task_mrs",
|
|
"gitlab_task_mr_options",
|
|
"gitlab_task_mr_state",
|
|
}
|
|
|
|
// DeleteTaskMRsByTaskID removes every contribution association owned by a
|
|
// hard-deleted task. It is safe to replay after a missed event.
|
|
func (s *Store) DeleteTaskMRsByTaskID(ctx context.Context, taskID string) (int64, error) {
|
|
result, err := s.db.ExecContext(ctx, `DELETE FROM gitlab_task_mrs WHERE task_id = ?`, taskID)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return result.RowsAffected()
|
|
}
|
|
|
|
// DeleteTaskOwnedStateByTaskID removes all GitLab rows owned by a hard-deleted
|
|
// task in one transaction. It is safe to replay after a missed event.
|
|
func (s *Store) DeleteTaskOwnedStateByTaskID(ctx context.Context, taskID string) (int64, error) {
|
|
tx, err := s.db.BeginTxx(ctx, nil)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
defer func() { _ = tx.Rollback() }()
|
|
|
|
var deleted int64
|
|
for _, table := range gitlabTaskOwnedTables {
|
|
result, err := tx.ExecContext(ctx, `DELETE FROM `+table+` WHERE task_id = ?`, taskID)
|
|
if err != nil {
|
|
return 0, fmt.Errorf("delete task-owned rows from %s: %w", table, err)
|
|
}
|
|
rows, err := result.RowsAffected()
|
|
if err != nil {
|
|
return 0, fmt.Errorf("count deleted rows from %s: %w", table, err)
|
|
}
|
|
deleted += rows
|
|
}
|
|
if err := tx.Commit(); err != nil {
|
|
return 0, err
|
|
}
|
|
return deleted, nil
|
|
}
|
|
|
|
func (s *Store) healTaskOwnedOrphans() error {
|
|
var exists int
|
|
err := s.db.Get(&exists, `SELECT 1 FROM sqlite_master WHERE type = 'table' AND name = 'tasks'`)
|
|
if errors.Is(err, sql.ErrNoRows) {
|
|
return nil // tasks table not yet created — skip sweep
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
tx, err := s.db.Beginx()
|
|
if err != nil {
|
|
return fmt.Errorf("begin orphan sweep: %w", err)
|
|
}
|
|
defer func() { _ = tx.Rollback() }()
|
|
for _, table := range gitlabTaskOwnedTables {
|
|
if _, err := tx.Exec(`DELETE FROM ` + table + ` WHERE NOT EXISTS (SELECT 1 FROM tasks WHERE tasks.id = ` + table + `.task_id)`); err != nil {
|
|
return fmt.Errorf("remove orphaned %s: %w", table, err)
|
|
}
|
|
}
|
|
if err := tx.Commit(); err != nil {
|
|
return fmt.Errorf("commit orphan sweep: %w", err)
|
|
}
|
|
return nil
|
|
}
|