Initial commit

server/model/uploading_file.go

@@ -0,0 +1,77 @@
+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,
+	}
+}