diff options
author | Lunny Xiao <xiaolunwen@gmail.com> | 2021-05-01 02:08:46 +0800 |
---|---|---|
committer | GitHub <noreply@github.com> | 2021-04-30 20:08:46 +0200 |
commit | f5eb33c354fc8e61cc1be221f3b7256fd7b66615 (patch) | |
tree | 2c9628831ac89d4049cdf71bd5ddd8eb4e7f7162 /models | |
parent | c80d7f33b67ad1beff7378bcba3aa44ac84669e9 (diff) | |
download | gitea-f5eb33c354fc8e61cc1be221f3b7256fd7b66615.tar.gz gitea-f5eb33c354fc8e61cc1be221f3b7256fd7b66615.zip |
Fix orphaned objects deletion bug (#15657)
* Fix orphaned objects deletion bug
* extend test
Co-authored-by: 6543 <6543@obermui.de>
Diffstat (limited to 'models')
-rw-r--r-- | models/consistency.go | 10 | ||||
-rw-r--r-- | models/consistency_test.go | 32 |
2 files changed, 39 insertions, 3 deletions
diff --git a/models/consistency.go b/models/consistency.go index a1a2bf7bb5..77a8018266 100644 --- a/models/consistency.go +++ b/models/consistency.go @@ -296,11 +296,15 @@ func CountOrphanedObjects(subject, refobject, joinCond string) (int64, error) { // DeleteOrphanedObjects delete subjects with have no existing refobject anymore func DeleteOrphanedObjects(subject, refobject, joinCond string) error { - _, err := x.In("id", builder.Select("`"+subject+"`.id"). + subQuery := builder.Select("`"+subject+"`.id"). From("`"+subject+"`"). Join("LEFT", "`"+refobject+"`", joinCond). - Where(builder.IsNull{"`" + refobject + "`.id"})). - Delete("`" + subject + "`") + Where(builder.IsNull{"`" + refobject + "`.id"}) + sql, args, err := builder.Delete(builder.In("id", subQuery)).From("`" + subject + "`").ToSQL() + if err != nil { + return err + } + _, err = x.Exec(append([]interface{}{sql}, args...)...) return err } diff --git a/models/consistency_test.go b/models/consistency_test.go new file mode 100644 index 0000000000..0a91d9f3da --- /dev/null +++ b/models/consistency_test.go @@ -0,0 +1,32 @@ +// Copyright 2021 Gitea. All rights reserved. +// Use of this source code is governed by a MIT-style +// license that can be found in the LICENSE file. + +package models + +import ( + "testing" + + "github.com/stretchr/testify/assert" +) + +func TestDeleteOrphanedObjects(t *testing.T) { + assert.NoError(t, PrepareTestDatabase()) + + countBefore, err := x.Count(&PullRequest{}) + assert.NoError(t, err) + + _, err = x.Insert(&PullRequest{IssueID: 1000}, &PullRequest{IssueID: 1001}, &PullRequest{IssueID: 1003}) + assert.NoError(t, err) + + orphaned, err := CountOrphanedObjects("pull_request", "issue", "pull_request.issue_id=issue.id") + assert.NoError(t, err) + assert.EqualValues(t, 3, orphaned) + + err = DeleteOrphanedObjects("pull_request", "issue", "pull_request.issue_id=issue.id") + assert.NoError(t, err) + + countAfter, err := x.Count(&PullRequest{}) + assert.NoError(t, err) + assert.EqualValues(t, countBefore, countAfter) +} |