fix(files): refactor upload mgr as upload info store

src/db/fileinfostore/file_info_store.go

@@ -54,6 +54,12 @@ type IFileInfoStore interface {
 	SetSha1(itemPath, sign string) error
 	GetInfos(itemPaths []string) (map[string]*FileInfo, error)
 	GetSharingDir(hashID string) (string, error)
+	// upload info
+	AddUploadInfo(user, filePath, tmpPath string, fileSize int64) error
+	SetUploadInfo(user, filePath string, newUploaded int64) error
+	GetUploadInfo(user, filePath string) (string, int64, int64, error)
+	DelUploadInfo(user, filePath string) error
+	ListUploadInfo(user string) ([]*UploadInfo, error)
 }
 
 type FileInfoStore struct {

src/db/fileinfostore/upload_info_store.go

@@ -0,0 +1,114 @@
+package fileinfostore
+
+import (
+	"encoding/json"
+	"errors"
+	"fmt"
+)
+
+var (
+	ErrCreateExisting  = errors.New("create upload info which already exists")
+	ErrGreaterThanSize = errors.New("uploaded is greater than file size")
+	ErrUploadNotFound  = errors.New("upload info not found")
+
+	uploadsPrefix = "uploads"
+)
+
+type UploadInfo struct {
+	RealFilePath string `json:"realFilePath"`
+	Size         int64  `json:"size"`
+	Uploaded     int64  `json:"uploaded"`
+}
+
+func UploadNS(user string) string {
+	return fmt.Sprintf("%s/%s", uploadsPrefix, user)
+}
+
+func (fi *FileInfoStore) AddUploadInfo(user, filePath, tmpPath string, fileSize int64) error {
+	ns := UploadNS(user)
+	err := fi.store.AddNamespace(ns)
+	if err != nil {
+		return err
+	}
+
+	_, ok := fi.store.GetStringIn(ns, tmpPath)
+	if ok {
+		return ErrCreateExisting
+	}
+
+	info := &UploadInfo{
+		RealFilePath: filePath,
+		Size:         fileSize,
+		Uploaded:     0,
+	}
+	infoBytes, err := json.Marshal(info)
+	if err != nil {
+		return err
+	}
+
+	return fi.store.SetStringIn(ns, tmpPath, string(infoBytes))
+}
+
+func (fi *FileInfoStore) SetUploadInfo(user, filePath string, newUploaded int64) error {
+	realFilePath, fileSize, _, err := fi.GetUploadInfo(user, filePath)
+	if err != nil {
+		return err
+	} else if newUploaded > fileSize {
+		return ErrGreaterThanSize
+	}
+
+	newInfo := &UploadInfo{
+		RealFilePath: realFilePath,
+		Size:         fileSize,
+		Uploaded:     newUploaded,
+	}
+	newInfoBytes, err := json.Marshal(newInfo)
+	if err != nil {
+		return err
+	}
+	return fi.store.SetStringIn(UploadNS(user), filePath, string(newInfoBytes))
+}
+
+func (fi *FileInfoStore) GetUploadInfo(user, filePath string) (string, int64, int64, error) {
+	ns := UploadNS(user)
+	infoBytes, ok := fi.store.GetStringIn(ns, filePath)
+	if !ok {
+		return "", 0, 0, ErrUploadNotFound
+	}
+
+	info := &UploadInfo{}
+	err := json.Unmarshal([]byte(infoBytes), info)
+	if err != nil {
+		return "", 0, 0, err
+	}
+
+	return info.RealFilePath, info.Size, info.Uploaded, nil
+}
+
+func (fi *FileInfoStore) DelUploadInfo(user, filePath string) error {
+	return fi.store.DelInt64In(UploadNS(user), filePath)
+}
+
+func (fi *FileInfoStore) ListUploadInfo(user string) ([]*UploadInfo, error) {
+	ns := UploadNS(user)
+	if !fi.store.HasNamespace(ns) {
+		return nil, nil
+	}
+
+	infoMap, err := fi.store.ListStringsIn(ns)
+	if err != nil {
+		return nil, err
+	}
+
+	infos := []*UploadInfo{}
+	for _, infoStr := range infoMap {
+		info := &UploadInfo{}
+		err = json.Unmarshal([]byte(infoStr), info)
+		if err != nil {
+			return nil, err
+		}
+		infos = append(infos, info)
+	}
+
+	return infos, nil
+}