Delete references if repository gets deleted (#15681)

* Remove DeletedBranch and LFSLocks.

* Sort beans.

Co-authored-by: zeripath <art27@cantab.net>
This commit is contained in:
KN4CK3R 2021-04-30 21:10:39 +02:00 committed by GitHub
parent f5eb33c354
commit ee3fb92419
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 63 additions and 28 deletions

View File

@ -1455,24 +1455,26 @@ func DeleteRepository(doer *User, uid, repoID int64) error {
if err := deleteBeans(sess, if err := deleteBeans(sess,
&Access{RepoID: repo.ID}, &Access{RepoID: repo.ID},
&Action{RepoID: repo.ID}, &Action{RepoID: repo.ID},
&Watch{RepoID: repoID},
&Star{RepoID: repoID},
&Mirror{RepoID: repoID},
&Milestone{RepoID: repoID},
&Release{RepoID: repoID},
&Collaboration{RepoID: repoID}, &Collaboration{RepoID: repoID},
&PullRequest{BaseRepoID: repoID},
&RepoUnit{RepoID: repoID},
&RepoRedirect{RedirectRepoID: repoID},
&Webhook{RepoID: repoID},
&HookTask{RepoID: repoID},
&Notification{RepoID: repoID},
&CommitStatus{RepoID: repoID},
&RepoIndexerStatus{RepoID: repoID},
&LanguageStat{RepoID: repoID},
&Comment{RefRepoID: repoID}, &Comment{RefRepoID: repoID},
&Task{RepoID: repoID}, &CommitStatus{RepoID: repoID},
&DeletedBranch{RepoID: repoID},
&HookTask{RepoID: repoID},
&LFSLock{RepoID: repoID},
&LanguageStat{RepoID: repoID},
&Milestone{RepoID: repoID},
&Mirror{RepoID: repoID},
&Notification{RepoID: repoID},
&ProtectedBranch{RepoID: repoID}, &ProtectedBranch{RepoID: repoID},
&PullRequest{BaseRepoID: repoID},
&Release{RepoID: repoID},
&RepoIndexerStatus{RepoID: repoID},
&RepoRedirect{RedirectRepoID: repoID},
&RepoUnit{RepoID: repoID},
&Star{RepoID: repoID},
&Task{RepoID: repoID},
&Watch{RepoID: repoID},
&Webhook{RepoID: repoID},
); err != nil { ); err != nil {
return fmt.Errorf("deleteBeans: %v", err) return fmt.Errorf("deleteBeans: %v", err)
} }
@ -1488,10 +1490,6 @@ func DeleteRepository(doer *User, uid, repoID int64) error {
return err return err
} }
if _, err := sess.Where("repo_id = ?", repoID).Delete(new(RepoUnit)); err != nil {
return err
}
if repo.IsFork { if repo.IsFork {
if _, err := sess.Exec("UPDATE `repository` SET num_forks=num_forks-1 WHERE id=?", repo.ForkID); err != nil { if _, err := sess.Exec("UPDATE `repository` SET num_forks=num_forks-1 WHERE id=?", repo.ForkID); err != nil {
return fmt.Errorf("decrease fork count: %v", err) return fmt.Errorf("decrease fork count: %v", err)

View File

@ -23,7 +23,7 @@ func checkDBConsistency(logger log.Logger, autofix bool) error {
// find labels without existing repo or org // find labels without existing repo or org
count, err := models.CountOrphanedLabels() count, err := models.CountOrphanedLabels()
if err != nil { if err != nil {
logger.Critical("Error: %v whilst counting orphaned labels") logger.Critical("Error: %v whilst counting orphaned labels", err)
return err return err
} }
if count > 0 { if count > 0 {
@ -41,7 +41,7 @@ func checkDBConsistency(logger log.Logger, autofix bool) error {
// find IssueLabels without existing label // find IssueLabels without existing label
count, err = models.CountOrphanedIssueLabels() count, err = models.CountOrphanedIssueLabels()
if err != nil { if err != nil {
logger.Critical("Error: %v whilst counting orphaned issue_labels") logger.Critical("Error: %v whilst counting orphaned issue_labels", err)
return err return err
} }
if count > 0 { if count > 0 {
@ -59,7 +59,7 @@ func checkDBConsistency(logger log.Logger, autofix bool) error {
// find issues without existing repository // find issues without existing repository
count, err = models.CountOrphanedIssues() count, err = models.CountOrphanedIssues()
if err != nil { if err != nil {
logger.Critical("Error: %v whilst counting orphaned issues") logger.Critical("Error: %v whilst counting orphaned issues", err)
return err return err
} }
if count > 0 { if count > 0 {
@ -77,7 +77,7 @@ func checkDBConsistency(logger log.Logger, autofix bool) error {
// find pulls without existing issues // find pulls without existing issues
count, err = models.CountOrphanedObjects("pull_request", "issue", "pull_request.issue_id=issue.id") count, err = models.CountOrphanedObjects("pull_request", "issue", "pull_request.issue_id=issue.id")
if err != nil { if err != nil {
logger.Critical("Error: %v whilst counting orphaned objects") logger.Critical("Error: %v whilst counting orphaned objects", err)
return err return err
} }
if count > 0 { if count > 0 {
@ -95,7 +95,7 @@ func checkDBConsistency(logger log.Logger, autofix bool) error {
// find tracked times without existing issues/pulls // find tracked times without existing issues/pulls
count, err = models.CountOrphanedObjects("tracked_time", "issue", "tracked_time.issue_id=issue.id") count, err = models.CountOrphanedObjects("tracked_time", "issue", "tracked_time.issue_id=issue.id")
if err != nil { if err != nil {
logger.Critical("Error: %v whilst counting orphaned objects") logger.Critical("Error: %v whilst counting orphaned objects", err)
return err return err
} }
if count > 0 { if count > 0 {
@ -113,7 +113,7 @@ func checkDBConsistency(logger log.Logger, autofix bool) error {
// find null archived repositories // find null archived repositories
count, err = models.CountNullArchivedRepository() count, err = models.CountNullArchivedRepository()
if err != nil { if err != nil {
logger.Critical("Error: %v whilst counting null archived repositories") logger.Critical("Error: %v whilst counting null archived repositories", err)
return err return err
} }
if count > 0 { if count > 0 {
@ -132,7 +132,7 @@ func checkDBConsistency(logger log.Logger, autofix bool) error {
// find label comments with empty labels // find label comments with empty labels
count, err = models.CountCommentTypeLabelWithEmptyLabel() count, err = models.CountCommentTypeLabelWithEmptyLabel()
if err != nil { if err != nil {
logger.Critical("Error: %v whilst counting label comments with empty labels") logger.Critical("Error: %v whilst counting label comments with empty labels", err)
return err return err
} }
if count > 0 { if count > 0 {
@ -191,7 +191,8 @@ func checkDBConsistency(logger log.Logger, autofix bool) error {
if setting.Database.UsePostgreSQL { if setting.Database.UsePostgreSQL {
count, err = models.CountBadSequences() count, err = models.CountBadSequences()
if err != nil { if err != nil {
logger.Critical("Error: %v whilst checking sequence values") logger.Critical("Error: %v whilst checking sequence values", err)
return err
} }
if count > 0 { if count > 0 {
if autofix { if autofix {
@ -210,7 +211,7 @@ func checkDBConsistency(logger log.Logger, autofix bool) error {
// find protected branches without existing repository // find protected branches without existing repository
count, err = models.CountOrphanedObjects("protected_branch", "repository", "protected_branch.repo_id=repository.id") count, err = models.CountOrphanedObjects("protected_branch", "repository", "protected_branch.repo_id=repository.id")
if err != nil { if err != nil {
logger.Critical("Error: %v whilst counting orphaned objects") logger.Critical("Error: %v whilst counting orphaned objects", err)
return err return err
} }
if count > 0 { if count > 0 {
@ -225,6 +226,42 @@ func checkDBConsistency(logger log.Logger, autofix bool) error {
} }
} }
// find deleted branches without existing repository
count, err = models.CountOrphanedObjects("deleted_branch", "repository", "deleted_branch.repo_id=repository.id")
if err != nil {
logger.Critical("Error: %v whilst counting orphaned objects", err)
return err
}
if count > 0 {
if autofix {
if err = models.DeleteOrphanedObjects("deleted_branch", "repository", "deleted_branch.repo_id=repository.id"); err != nil {
logger.Critical("Error: %v whilst deleting orphaned objects", err)
return err
}
logger.Info("%d deleted branches without existing repository deleted", count)
} else {
logger.Warn("%d deleted branches without existing repository", count)
}
}
// find LFS locks without existing repository
count, err = models.CountOrphanedObjects("lfs_lock", "repository", "lfs_lock.repo_id=repository.id")
if err != nil {
logger.Critical("Error: %v whilst counting orphaned objects", err)
return err
}
if count > 0 {
if autofix {
if err = models.DeleteOrphanedObjects("lfs_lock", "repository", "lfs_lock.repo_id=repository.id"); err != nil {
logger.Critical("Error: %v whilst deleting orphaned objects", err)
return err
}
logger.Info("%d LFS locks without existing repository deleted", count)
} else {
logger.Warn("%d LFS locks without existing repository", count)
}
}
return nil return nil
} }