Browse Source
* Add size column to attachment Migrate attachments by calculating file sizes Signed-off-by: Jonas Franz <info@jonasfranz.software> * Calculate attachment size on creation Signed-off-by: Jonas Franz <info@jonasfranz.software> * Log error instead of returning error Signed-off-by: Jonas Franz <info@jonasfranz.software>aj-second-factor
Jonas Franz
6 years ago
committed by
Bo-Yi Wu
4 changed files with 57 additions and 23 deletions
@ -0,0 +1,45 @@ |
|||
// 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(4, "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 |
|||
} |
Loading…
Reference in new issue