mirror of
https://github.com/wgh136/nysoure.git
synced 2025-09-27 04:17:23 +00:00
78 lines
1.9 KiB
Go
78 lines
1.9 KiB
Go
package model
|
|
|
|
import (
|
|
"context"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/schema"
|
|
"reflect"
|
|
)
|
|
|
|
type UploadingFile struct {
|
|
gorm.Model
|
|
Filename string
|
|
Description string
|
|
TargetResourceID uint
|
|
TargetStorageID uint
|
|
UserID uint
|
|
BlockSize int64
|
|
TotalSize int64
|
|
Blocks UploadingFileBlocks `gorm:"type:blob"`
|
|
TempPath string
|
|
Resource Resource `gorm:"foreignKey:TargetResourceID"`
|
|
Storage Storage `gorm:"foreignKey:TargetStorageID"`
|
|
}
|
|
|
|
func (uf *UploadingFile) BlocksCount() int {
|
|
return int((uf.TotalSize + uf.BlockSize - 1) / uf.BlockSize)
|
|
}
|
|
|
|
type UploadingFileBlocks []bool
|
|
|
|
func (ufb *UploadingFileBlocks) Scan(ctx context.Context, field *schema.Field, dst reflect.Value, dbValue interface{}) (err error) {
|
|
data, ok := dbValue.([]byte)
|
|
if !ok {
|
|
return nil
|
|
}
|
|
*ufb = make([]bool, len(data)*8)
|
|
for i, b := range data {
|
|
for j := 0; j < 8; j++ {
|
|
(*ufb)[i*8+j] = (b>>j)&1 == 1
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (ufb UploadingFileBlocks) Value(ctx context.Context, field *schema.Field, dbValue reflect.Value) (value interface{}, err error) {
|
|
data := make([]byte, (len(ufb)+7)/8)
|
|
for i, b := range ufb {
|
|
if b {
|
|
data[i/8] |= 1 << (i % 8)
|
|
}
|
|
}
|
|
return data, nil
|
|
}
|
|
|
|
type UploadingFileView struct {
|
|
ID uint `json:"id"`
|
|
Filename string `json:"filename"`
|
|
Description string `json:"description"`
|
|
TotalSize int64 `json:"totalSize"`
|
|
BlockSize int64 `json:"blockSize"`
|
|
BlocksCount int `json:"blocksCount"`
|
|
StorageID uint `json:"storageId"`
|
|
ResourceID uint `json:"resourceId"`
|
|
}
|
|
|
|
func (uf *UploadingFile) ToView() *UploadingFileView {
|
|
return &UploadingFileView{
|
|
ID: uf.ID,
|
|
Filename: uf.Filename,
|
|
Description: uf.Description,
|
|
TotalSize: uf.TotalSize,
|
|
BlockSize: uf.BlockSize,
|
|
BlocksCount: uf.BlocksCount(),
|
|
StorageID: uf.TargetStorageID,
|
|
ResourceID: uf.TargetResourceID,
|
|
}
|
|
}
|