Платформа ЦРНП "Мирокод" для разработки проектов
https://git.mirocod.ru
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
45 lines
1.2 KiB
45 lines
1.2 KiB
// Copyright 2018 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 ( |
|
"fmt" |
|
"os" |
|
"path" |
|
|
|
"code.gitea.io/gitea/modules/log" |
|
"code.gitea.io/gitea/modules/setting" |
|
|
|
"github.com/go-xorm/xorm" |
|
) |
|
|
|
func addSizeToAttachment(x *xorm.Engine) error { |
|
type Attachment struct { |
|
ID int64 `xorm:"pk autoincr"` |
|
UUID string `xorm:"uuid UNIQUE"` |
|
Size int64 `xorm:"DEFAULT 0"` |
|
} |
|
if err := x.Sync2(new(Attachment)); err != nil { |
|
return fmt.Errorf("Sync2: %v", err) |
|
} |
|
|
|
attachments := make([]Attachment, 0, 100) |
|
if err := x.Find(&attachments); err != nil { |
|
return fmt.Errorf("query attachments: %v", err) |
|
} |
|
for _, attach := range attachments { |
|
localPath := path.Join(setting.AttachmentPath, attach.UUID[0:1], attach.UUID[1:2], attach.UUID) |
|
fi, err := os.Stat(localPath) |
|
if err != nil { |
|
log.Error("calculate file size of attachment[UUID: %s]: %v", attach.UUID, err) |
|
continue |
|
} |
|
attach.Size = fi.Size() |
|
if _, err := x.ID(attach.ID).Cols("size").Update(attach); err != nil { |
|
return fmt.Errorf("update size column: %v", err) |
|
} |
|
} |
|
return nil |
|
}
|
|
|