From 084797b4dc56b6f9186bb20f6b09ae902a20a26a Mon Sep 17 00:00:00 2001 From: Lunny Xiao Date: Tue, 6 Sep 2022 06:48:57 +0800 Subject: [PATCH] Fix delete user missed some comments (#21067) (#21068) --- models/packages/package_test.go | 2 +- models/user.go | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/models/packages/package_test.go b/models/packages/package_test.go index 3d3a2333b..0bfe6a398 100644 --- a/models/packages/package_test.go +++ b/models/packages/package_test.go @@ -27,7 +27,7 @@ func TestMain(m *testing.M) { func TestHasOwnerPackages(t *testing.T) { assert.NoError(t, unittest.PrepareTestDatabase()) - owner := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1}) + owner := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1}).(*user_model.User) p, err := packages_model.TryInsertPackage(db.DefaultContext, &packages_model.Package{ OwnerID: owner.ID, diff --git a/models/user.go b/models/user.go index 49374014a..fb9246070 100644 --- a/models/user.go +++ b/models/user.go @@ -100,9 +100,9 @@ func DeleteUser(ctx context.Context, u *user_model.User) (err error) { // Delete Comments const batchSize = 50 - for start := 0; ; start += batchSize { + for { comments := make([]*issues_model.Comment, 0, batchSize) - if err = e.Where("type=? AND poster_id=?", issues_model.CommentTypeComment, u.ID).Limit(batchSize, start).Find(&comments); err != nil { + if err = e.Where("type=? AND poster_id=?", issues_model.CommentTypeComment, u.ID).Limit(batchSize, 0).Find(&comments); err != nil { return err } if len(comments) == 0 { @@ -200,7 +200,7 @@ func DeleteUser(ctx context.Context, u *user_model.User) (err error) { // ***** END: ExternalLoginUser ***** if _, err = e.ID(u.ID).Delete(new(user_model.User)); err != nil { - return fmt.Errorf("Delete: %v", err) + return fmt.Errorf("delete: %v", err) } return nil