diff options
Diffstat (limited to 'models/migrations/v96.go')
-rw-r--r-- | models/migrations/v96.go | 48 |
1 files changed, 48 insertions, 0 deletions
diff --git a/models/migrations/v96.go b/models/migrations/v96.go new file mode 100644 index 0000000000..5c2135ffc6 --- /dev/null +++ b/models/migrations/v96.go @@ -0,0 +1,48 @@ +// Copyright 2019 The Gitea Authors. All rights reserved. +// Use of this source code is governed by a MIT-style +// license that can be found in the LICENSE file. + +package migrations + +import ( + "os" + + "code.gitea.io/gitea/models" + "code.gitea.io/gitea/modules/setting" + + "github.com/go-xorm/xorm" +) + +func deleteOrphanedAttachments(x *xorm.Engine) error { + + type Attachment struct { + ID int64 `xorm:"pk autoincr"` + UUID string `xorm:"uuid UNIQUE"` + IssueID int64 `xorm:"INDEX"` + ReleaseID int64 `xorm:"INDEX"` + CommentID int64 + } + + sess := x.NewSession() + defer sess.Close() + + err := sess.BufferSize(setting.Database.IterateBufferSize). + Where("`comment_id` = 0 and (`release_id` = 0 or `release_id` not in (select `id` from `release`))").Cols("uuid"). + Iterate(new(Attachment), + func(idx int, bean interface{}) error { + attachment := bean.(*Attachment) + + if err := os.RemoveAll(models.AttachmentLocalPath(attachment.UUID)); err != nil { + return err + } + + _, err := sess.ID(attachment.ID).NoAutoCondition().Delete(attachment) + return err + }) + + if err != nil { + return err + } + + return sess.Commit() +} |