Lee-WineList/utils/oss.go

package utils

import (
	"Lee-WineList/config"
	"fmt"
	"git.echol.cn/loser/logger/log"
	"github.com/aliyun/aliyun-oss-go-sdk/oss"
	"io"
)

type ossUtils struct{}

func OssUtils() *ossUtils {
	return &ossUtils{}
}

// UploadReader 上传文件 - io.Reader
func (o ossUtils) UploadReader(fileName string, data io.Reader) (string, error) {
	// 创建OSSClient实例。
	client, err := oss.New(config.Scd.Aliyun.Oss.Endpoint, config.Scd.Aliyun.Oss.AccessKeyId, config.Scd.Aliyun.Oss.AccessKeySecret)
	if err != nil {
		log.Errorf("创建OSSClient实例失败: %s", err)
		return "", err
	}
	// 获取存储空间。
	bucket, err := client.Bucket(config.Scd.Aliyun.Oss.Bucket)
	if err != nil {
		log.Errorf("获取存储空间失败: %s", err)
		return "", err
	}
	// 指定存储类型为标准存储，默认为标准存储。
	storageType := oss.ObjectStorageClass(oss.StorageStandard)
	// 指定访问权限为公共读，默认为继承bucket的权限。
	objectAcl := oss.ObjectACL(oss.ACLPublicRead)
	// 上传文件
	err = bucket.PutObject(fileName, data, storageType, objectAcl)
	if err != nil {
		log.Errorf("上传文件失败: %s", err)
		return "", err
	}
	fileUrl := fmt.Sprintf("https://%s.%s/%s", config.Scd.Aliyun.Oss.Bucket, config.Scd.Aliyun.Oss.Endpoint, fileName)
	return fileUrl, err
}

func (o ossUtils) GetFile(fileName, localFilePath string) error {
	client, err := oss.New(config.Scd.Aliyun.Oss.Endpoint, config.Scd.Aliyun.Oss.AccessKeyId, config.Scd.Aliyun.Oss.AccessKeySecret)
	if err != nil {
		log.Errorf("创建OSSClient实例失败: %s", err)
		return err
	}
	// 获取存储空间。
	bucket, err := client.Bucket(config.Scd.Aliyun.Oss.Bucket)
	if err != nil {
		log.Errorf("获取存储空间失败: %s", err)
		return err
	}
	err = bucket.GetObjectToFile(fileName, localFilePath)
	if err != nil {
		return err
	}
	return nil
}