aliyun_oss.go

package sc

import (
	"fmt"
	"github.com/aliyun/aliyun-oss-go-sdk/oss"
	slog "github.com/vearne/simplelog"
	"os"
	"time"
)

type AliOSS struct {
	AccessKeyId     string `mapstructure:"access_key_id"`
	AccessKeySecret string `mapstructure:"access_key_secret"`
	BucketName      string `mapstructure:"bucket_name"`
	Endpoint        string `mapstructure:"endpoint"`
	DirPath         string `mapstructure:"dir_path"`
	Bucket          *oss.Bucket
}

func (s *AliOSS) Init() error {
	client, err := oss.New(s.Endpoint, s.AccessKeyId, s.AccessKeySecret)
	if err != nil {
		return err
	}
	s.Bucket, err = client.Bucket(s.BucketName)
	if err != nil {
		return err
	}
	return nil
}

func (s *AliOSS) GetDirPath() string {
	return s.DirPath
}

func (s *AliOSS) UploadFile(key string, localFilePath string) bool {
	err := s.Bucket.PutObjectFromFile(key, localFilePath)
	if err != nil {
		slog.Error("AliOSS.UploadFile, error:%v", err)
		return false
	}
	return true
}

func (s *AliOSS) ListKeys(prefix string) ([]string, error) {
	lsRes, err := s.Bucket.ListObjects(oss.Prefix(prefix))
	if err != nil {
		slog.Error("AliOSS.ListKeys, error:%v", err)
		return nil, err
	}

	result := make([]string, 0)
	for _, object := range lsRes.Objects {
		result = append(result, object.Key)
	}
	return result, nil
}

func (s *AliOSS) DownloadFile(key string, localFilePath string) bool {
	err := s.Bucket.GetObjectToFile(key, localFilePath)
	if err != nil {
		slog.Error("AliOSS.DownloadFile, error:%v", err)
		return false
	}
	return true
}

func (s *AliOSS) Compare(key string, localFilePath string) (int, error) {
	lsRes, err := s.Bucket.ListObjects(oss.Prefix(key))
	if err != nil {
		slog.Error("AliOSS.ListKeys, error:%v", err)
		return -1, err
	}

	if len(lsRes.Objects) <= 0 {
		return -1, fmt.Errorf("AliOSS.ListKeys, len(lsRes.Objects) <= 0")
	}

	var obj oss.ObjectProperties
	for _, object := range lsRes.Objects {
		if key == object.Key {
			obj = object
			break
		}
	}

	info, err := os.Stat(localFilePath)
	if err != nil {
		slog.Debug("os.Stat error, %v", err)
		return 1, nil
	}

	localLastModified := info.ModTime()
	if obj.LastModified.Unix() > localLastModified.Unix() {
		return 1, nil
	} else if obj.LastModified.Unix() == localLastModified.Unix() {
		return 0, nil
	} else {
		return -1, nil
	}

}

func (s *AliOSS) AdjustMTime(key string, localFilePath string) error {
	lsRes, err := s.Bucket.ListObjects(oss.Prefix(key))
	if err != nil {
		return fmt.Errorf("AliOSS.ListKeys, error:%v", err)
	}

	if len(lsRes.Objects) <= 0 {
		return fmt.Errorf("AliOSS.ListKeys, len(lsRes.Objects) <= 0")
	}

	var obj oss.ObjectProperties
	for _, object := range lsRes.Objects {
		if key == object.Key {
			obj = object
			break
		}
	}

	mTime := time.Unix(obj.LastModified.Unix(), 0)
	err = os.Chtimes(localFilePath, time.Now(), mTime)

	return err
}

func (s *AliOSS) Delete(key string) error {
	err := s.Bucket.DeleteObject(key)
	return err
}