Carlos Florêncio
79 lines
2.2 KiB
Go
79 lines
2.2 KiB
Go
package github
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
dbutil "github.com/kandev/kandev/internal/db"
|
|
)
|
|
|
|
var githubTaskOwnedTables = []string{
|
|
"github_pr_watches",
|
|
"github_task_prs",
|
|
"github_task_ci_options",
|
|
"github_task_pr_automation_options",
|
|
"github_task_ci_pr_state",
|
|
}
|
|
|
|
// DeleteTaskPRsByTaskID removes every contribution association owned by a
|
|
// hard-deleted task. It is safe to replay after a missed event.
|
|
func (s *Store) DeleteTaskPRsByTaskID(ctx context.Context, taskID string) (int64, error) {
|
|
result, err := s.db.ExecContext(ctx, s.db.Rebind(`DELETE FROM github_task_prs WHERE task_id = ?`), taskID)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return result.RowsAffected()
|
|
}
|
|
|
|
// DeleteTaskOwnedStateByTaskID removes all GitHub 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 githubTaskOwnedTables {
|
|
result, err := tx.ExecContext(ctx, tx.Rebind(`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 {
|
|
exists, err := dbutil.TableExists(s.db, "tasks")
|
|
if err != nil {
|
|
return fmt.Errorf("check tasks table for orphan sweep: %w", err)
|
|
}
|
|
if !exists {
|
|
return nil
|
|
}
|
|
|
|
tx, err := s.db.Beginx()
|
|
if err != nil {
|
|
return fmt.Errorf("begin orphan sweep: %w", err)
|
|
}
|
|
defer func() { _ = tx.Rollback() }()
|
|
for _, table := range githubTaskOwnedTables {
|
|
if _, err := tx.Exec(tx.Rebind(`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
|
|
}
|