diff options
author | Lunny Xiao <xiaolunwen@gmail.com> | 2018-11-30 20:59:12 +0800 |
---|---|---|
committer | Lauris BH <lauris@nix.lv> | 2018-11-30 14:59:12 +0200 |
commit | f80b4f46ae67b9382f973cdd5ef64c9f53e793e2 (patch) | |
tree | 11672e68224ec10eb65192120229221f56296235 /models | |
parent | c8a9384a70219fbc85473404f2ac2747387d73ea (diff) | |
download | gitea-f80b4f46ae67b9382f973cdd5ef64c9f53e793e2.tar.gz gitea-f80b4f46ae67b9382f973cdd5ef64c9f53e793e2.zip |
Fix repository deletion when there is large number of issues in it (#5426)
Diffstat (limited to 'models')
-rw-r--r-- | models/repo.go | 79 |
1 files changed, 40 insertions, 39 deletions
diff --git a/models/repo.go b/models/repo.go index b86226ec82..a3e3838186 100644 --- a/models/repo.go +++ b/models/repo.go @@ -32,6 +32,7 @@ import ( "github.com/Unknwon/cae/zip" "github.com/Unknwon/com" + "github.com/go-xorm/builder" "github.com/go-xorm/xorm" "github.com/mcuadros/go-version" "gopkg.in/ini.v1" @@ -1774,51 +1775,51 @@ func DeleteRepository(doer *User, uid, repoID int64) error { return fmt.Errorf("deleteBeans: %v", err) } - // Delete comments and attachments. - issueIDs := make([]int64, 0, 25) - attachmentPaths := make([]string, 0, len(issueIDs)) - if err = sess. - Table("issue"). - Cols("id"). - Where("repo_id=?", repoID). - Find(&issueIDs); err != nil { + deleteCond := builder.Select("id").From("issue").Where(builder.Eq{"repo_id": repoID}) + // Delete comments and attachments + if _, err = sess.In("issue_id", deleteCond). + Delete(&Comment{}); err != nil { return err } - if len(issueIDs) > 0 { - if _, err = sess.In("issue_id", issueIDs).Delete(&Comment{}); err != nil { - return err - } - if _, err = sess.In("issue_id", issueIDs).Delete(&IssueUser{}); err != nil { - return err - } - if _, err = sess.In("issue_id", issueIDs).Delete(&Reaction{}); err != nil { - return err - } - if _, err = sess.In("issue_id", issueIDs).Delete(&IssueWatch{}); err != nil { - return err - } - if _, err = sess.In("issue_id", issueIDs).Delete(&Stopwatch{}); err != nil { - return err - } + if _, err = sess.In("issue_id", deleteCond). + Delete(&IssueUser{}); err != nil { + return err + } - attachments := make([]*Attachment, 0, 5) - if err = sess. - In("issue_id", issueIDs). - Find(&attachments); err != nil { - return err - } - for j := range attachments { - attachmentPaths = append(attachmentPaths, attachments[j].LocalPath()) - } + if _, err = sess.In("issue_id", deleteCond). + Delete(&Reaction{}); err != nil { + return err + } - if _, err = sess.In("issue_id", issueIDs).Delete(&Attachment{}); err != nil { - return err - } + if _, err = sess.In("issue_id", deleteCond). + Delete(&IssueWatch{}); err != nil { + return err + } - if _, err = sess.Delete(&Issue{RepoID: repoID}); err != nil { - return err - } + if _, err = sess.In("issue_id", deleteCond). + Delete(&Stopwatch{}); err != nil { + return err + } + + attachmentPaths := make([]string, 0, 20) + attachments := make([]*Attachment, 0, len(attachmentPaths)) + if err = sess.Join("INNER", "issue", "issue.id = attachment.issue_id"). + Where("issue.repo_id = ?", repoID). + Find(&attachments); err != nil { + return err + } + for j := range attachments { + attachmentPaths = append(attachmentPaths, attachments[j].LocalPath()) + } + + if _, err = sess.In("issue_id", deleteCond). + Delete(&Attachment{}); err != nil { + return err + } + + if _, err = sess.Delete(&Issue{RepoID: repoID}); err != nil { + return err } if _, err = sess.Where("repo_id = ?", repoID).Delete(new(RepoUnit)); err != nil { |