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) { // TODO return false, 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) }