|
|
package storage
|
|
|
|
|
|
import (
|
|
|
"errors"
|
|
|
"fmt"
|
|
|
"io"
|
|
|
"os"
|
|
|
"path/filepath"
|
|
|
"strings"
|
|
|
"time"
|
|
|
|
|
|
"github.com/Masterminds/semver/v3"
|
|
|
"github.com/rocboss/paopao-ce/internal/core"
|
|
|
)
|
|
|
|
|
|
var (
|
|
|
_ core.ObjectStorageService = (*localossServant)(nil)
|
|
|
_ core.VersionInfo = (*localossServant)(nil)
|
|
|
)
|
|
|
|
|
|
type localossServant struct {
|
|
|
savePath string
|
|
|
domain string
|
|
|
}
|
|
|
|
|
|
func (s *localossServant) Name() string {
|
|
|
return "LocalOSS"
|
|
|
}
|
|
|
|
|
|
func (s *localossServant) Version() *semver.Version {
|
|
|
return semver.MustParse("v0.1.0")
|
|
|
}
|
|
|
|
|
|
func (s *localossServant) PutObject(objectKey string, reader io.Reader, objectSize int64, contentType string) (string, error) {
|
|
|
saveDir := s.savePath + filepath.Dir(objectKey)
|
|
|
err := os.MkdirAll(saveDir, 0750)
|
|
|
if err != nil && !os.IsExist(err) {
|
|
|
return "", err
|
|
|
}
|
|
|
|
|
|
savePath := s.savePath + objectKey
|
|
|
writer, err := os.Create(savePath)
|
|
|
if err != nil {
|
|
|
return "", err
|
|
|
}
|
|
|
defer writer.Close()
|
|
|
|
|
|
written, err := io.Copy(writer, reader)
|
|
|
if err != nil {
|
|
|
return "", err
|
|
|
}
|
|
|
if written != objectSize {
|
|
|
os.Remove(savePath)
|
|
|
return "", errors.New("put object not complete")
|
|
|
}
|
|
|
|
|
|
return s.domain + objectKey, nil
|
|
|
}
|
|
|
|
|
|
func (s *localossServant) DeleteObject(objectKey string) error {
|
|
|
return os.Remove(s.savePath + objectKey)
|
|
|
}
|
|
|
|
|
|
func (s *localossServant) DeleteObjcets(objectKeys []string) (err error) {
|
|
|
// 宽松处理删除动作，尽可能删除所有objectKey，如果出错，只返回最后一个错误
|
|
|
for _, objectKey := range objectKeys {
|
|
|
if e := os.Remove(s.savePath + objectKey); e != nil {
|
|
|
err = e
|
|
|
}
|
|
|
}
|
|
|
return
|
|
|
}
|
|
|
|
|
|
func (s *localossServant) IsObjectExist(objectKey string) (bool, error) {
|
|
|
fi, err := os.Stat(s.savePath + objectKey)
|
|
|
if err != nil {
|
|
|
return false, err
|
|
|
}
|
|
|
return !fi.IsDir(), nil
|
|
|
}
|
|
|
|
|
|
func (s *localossServant) SignURL(objectKey string, expiredInSec int64) (string, error) {
|
|
|
if expiredInSec < 0 {
|
|
|
return "", fmt.Errorf("invalid expires: %d, expires must bigger than 0", expiredInSec)
|
|
|
}
|
|
|
expiration := time.Now().Unix() + expiredInSec
|
|
|
|
|
|
// Fixed: Just make things simple and simple now so return an veiry simple sign url.
|
|
|
// Maybe make another process logic for sign url in future but not now.
|
|
|
uri := fmt.Sprintf("%s%s?expired=%d", s.domain, objectKey, expiration)
|
|
|
return uri, nil
|
|
|
}
|
|
|
|
|
|
func (s *localossServant) ObjectURL(objetKey string) string {
|
|
|
return s.domain + objetKey
|
|
|
}
|
|
|
|
|
|
func (s *localossServant) ObjectKey(objectUrl string) string {
|
|
|
return strings.Replace(objectUrl, s.domain, "", -1)
|
|
|
}
|
