diff options
author | 6543 | 2021-04-30 23:27:13 +0200 |
---|---|---|
committer | GitHub | 2021-04-30 22:27:13 +0100 |
commit | cefcc7613b0501a74c952edb02fbe884284fcd48 (patch) | |
tree | d28c683629ea4b38a98f2079a39d69bd61355657 | |
parent | 05f266c3315aec95f4cf18a9938decd707603e53 (diff) |
Fix orphaned objects deletion bug (#15657) (#15682)release/v1.13
* Fix orphaned objects deletion bug
* extend test
Co-authored-by: 6543 <6543@obermui.de>
Co-authored-by: Lunny Xiao <xiaolunwen@gmail.com>
-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 3a5c4cae5..d8abdd133 100644 --- a/models/consistency.go +++ b/models/consistency.go @@ -297,11 +297,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 000000000..0a91d9f3d --- /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) +} |