Fix repository deletion when there is large number of issues in it (#5426)

This commit is contained in:
Lunny Xiao 2018-11-30 20:59:12 +08:00 committed by Lauris BH
parent c8a9384a70
commit f80b4f46ae
1 changed files with 40 additions and 39 deletions

View File

@ -32,6 +32,7 @@ import (
"github.com/Unknwon/cae/zip" "github.com/Unknwon/cae/zip"
"github.com/Unknwon/com" "github.com/Unknwon/com"
"github.com/go-xorm/builder"
"github.com/go-xorm/xorm" "github.com/go-xorm/xorm"
"github.com/mcuadros/go-version" "github.com/mcuadros/go-version"
"gopkg.in/ini.v1" "gopkg.in/ini.v1"
@ -1774,51 +1775,51 @@ func DeleteRepository(doer *User, uid, repoID int64) error {
return fmt.Errorf("deleteBeans: %v", err) return fmt.Errorf("deleteBeans: %v", err)
} }
// Delete comments and attachments. deleteCond := builder.Select("id").From("issue").Where(builder.Eq{"repo_id": repoID})
issueIDs := make([]int64, 0, 25) // Delete comments and attachments
attachmentPaths := make([]string, 0, len(issueIDs)) if _, err = sess.In("issue_id", deleteCond).
if err = sess. Delete(&Comment{}); err != nil {
Table("issue").
Cols("id").
Where("repo_id=?", repoID).
Find(&issueIDs); err != nil {
return err return err
} }
if len(issueIDs) > 0 { if _, err = sess.In("issue_id", deleteCond).
if _, err = sess.In("issue_id", issueIDs).Delete(&Comment{}); err != nil { Delete(&IssueUser{}); err != nil {
return err 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
}
attachments := make([]*Attachment, 0, 5) if _, err = sess.In("issue_id", deleteCond).
if err = sess. Delete(&Reaction{}); err != nil {
In("issue_id", issueIDs). return err
Find(&attachments); err != nil { }
return err
}
for j := range attachments {
attachmentPaths = append(attachmentPaths, attachments[j].LocalPath())
}
if _, err = sess.In("issue_id", issueIDs).Delete(&Attachment{}); err != nil { if _, err = sess.In("issue_id", deleteCond).
return err Delete(&IssueWatch{}); err != nil {
} return err
}
if _, err = sess.Delete(&Issue{RepoID: repoID}); err != nil { if _, err = sess.In("issue_id", deleteCond).
return err 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 { if _, err = sess.Where("repo_id = ?", repoID).Delete(new(RepoUnit)); err != nil {