forked from Shiloh/githaven
74bd9691c6
Replace all calls to os.Remove/os.RemoveAll by retrying util.Remove/util.RemoveAll and remove circular dependencies from util. Fix #12339 Signed-off-by: Andrew Thornton <art27@cantab.net> Co-authored-by: silverwind <me@silverwind.io>
70 lines
1.6 KiB
Go
70 lines
1.6 KiB
Go
// 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 (
|
|
"path"
|
|
|
|
"code.gitea.io/gitea/modules/setting"
|
|
"code.gitea.io/gitea/modules/util"
|
|
|
|
"xorm.io/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
|
|
}
|
|
|
|
// AttachmentLocalPath returns where attachment is stored in local file
|
|
// system based on given UUID.
|
|
AttachmentLocalPath := func(uuid string) string {
|
|
return path.Join(setting.AttachmentPath, uuid[0:1], uuid[1:2], uuid)
|
|
}
|
|
|
|
sess := x.NewSession()
|
|
defer sess.Close()
|
|
|
|
var limit = setting.Database.IterateBufferSize
|
|
if limit <= 0 {
|
|
limit = 50
|
|
}
|
|
|
|
for {
|
|
attachements := make([]Attachment, 0, limit)
|
|
if err := sess.Where("`issue_id` = 0 and (`release_id` = 0 or `release_id` not in (select `id` from `release`))").
|
|
Cols("id, uuid").Limit(limit).
|
|
Asc("id").
|
|
Find(&attachements); err != nil {
|
|
return err
|
|
}
|
|
if len(attachements) == 0 {
|
|
return nil
|
|
}
|
|
|
|
var ids = make([]int64, 0, limit)
|
|
for _, attachment := range attachements {
|
|
ids = append(ids, attachment.ID)
|
|
}
|
|
if _, err := sess.In("id", ids).Delete(new(Attachment)); err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, attachment := range attachements {
|
|
if err := util.RemoveAll(AttachmentLocalPath(attachment.UUID)); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
if len(attachements) < limit {
|
|
return nil
|
|
}
|
|
}
|
|
}
|