hrms_api/services/resource/oss.go

package resource

import (
	"context"
	"encoding/json"
	"errors"
	"github.com/aliyun/alibaba-cloud-sdk-go/services/sts"
	"github.com/aliyun/aliyun-oss-go-sdk/oss"
	"hongze/hrms_api/global"
	"hongze/hrms_api/services/alarm_msg"
	"hongze/hrms_api/utils"
	"os"
	"time"
)

var (
	RoleArn            = "acs:ram::1884217364581072:role/hzossrole"
	RoleSessionName    = "hzossRole"
	RAMAccessKeyId     = "LTAI5t9S36LXduhTnECVY6Hn"
	RAMAccessKeySecret = "V6FG5bdzjKKqMpqEqxeWijJCdzDCuL"
	STSTokenCacheKey   = "hrms_api:oss:sts_token"
)


// UploadAliyun 图片上传到阿里云
func UploadAliyun(filename, filepath string) (string, error) {
	client, err := oss.New(global.CONFIG.AliOss.EndPoint, global.CONFIG.AliOss.AccessKeyId, global.CONFIG.AliOss.AccessKeySecret)
	if err != nil {
		return "1", err
	}
	bucket, err := client.Bucket(global.CONFIG.AliOss.BucketName)
	if err != nil {
		return "2", err
	}
	path := global.CONFIG.AliOss.UploadDir + time.Now().Format("200601/20060102/")
	path += filename
	err = bucket.PutObjectFromFile(path, filepath)
	if err != nil {
		return "3", err
	}
	path = global.CONFIG.AliOss.ImgHost + path
	return path, err
}

// UploadAudioAliyun 音频上传到阿里云
func UploadAudioAliyun(filename, filepath string) (string, error) {
	client, err := oss.New(global.CONFIG.AliOss.EndPoint, global.CONFIG.AliOss.AccessKeyId, global.CONFIG.AliOss.AccessKeySecret)
	if err != nil {
		return "1", err
	}
	bucket, err := client.Bucket(global.CONFIG.AliOss.BucketName)
	if err != nil {
		return "2", err
	}
	path := global.CONFIG.AliOss.UploadAudioDir + time.Now().Format("200601/20060102/")
	path += filename
	err = bucket.PutObjectFromFile(path, filepath)
	if err != nil {
		return "3", err
	}
	path = global.CONFIG.AliOss.ImgHost + path
	return path, err
}

// UploadVideoAliyun 视频上传到阿里云
func UploadVideoAliyun(filename, filepath, savePath string) error {
	defer func() {
		_ = os.Remove(filepath)
	}()
	client, err := oss.New(global.CONFIG.AliOss.EndPoint, global.CONFIG.AliOss.AccessKeyId, global.CONFIG.AliOss.AccessKeySecret)
	if err != nil {
		return err
	}
	bucket, err := client.Bucket(global.CONFIG.AliOss.BucketName)
	if err != nil {
		return err
	}
	//path := global.CONFIG.AliOss.UploadAudioDir + time.Now().Format("200601/20060102/")
	//path += filename
	err = bucket.PutObjectFromFile(savePath, filepath)
	if err != nil {
		return err
	}
	//path = global.CONFIG.AliOss.ImgHost + path
	//return path,err
	return err
}

var (
	Bucketname       = "hzchart"
	Endpoint         = "oss-cn-shanghai.aliyuncs.com"
	ResourceHost     = "https://hzstatic.hzinsights.com/"
	StaticDir        = "static/"
	AccessKeyId      = "LTAIFMZYQhS2BTvW"
	AccessKeySecret  = "12kk1ptCHoGWedhBnKRVW5hRJzq9Fq"
	UploadDir        = "static/images/"
)

// UploadAliyunToDir
func UploadAliyunToDir(filename, filepath, fileDir string) (string, error) {
	client, err := oss.New(Endpoint, AccessKeyId, AccessKeySecret)
	if err != nil {
		return "1", err
	}
	bucket, err := client.Bucket(Bucketname)
	if err != nil {
		return "2", err
	}
	if fileDir == "" {
		fileDir = time.Now().Format("200601/20060102/")
	}
	path := StaticDir + fileDir
	path += filename
	err = bucket.PutObjectFromFile(path, filepath)
	if err != nil {
		return "3", err
	}
	path = ResourceHost + path
	return path, err
}


// UploadAliyunV2 图片上传到阿里云
func UploadAliyunV2(filename, filepath string) (string, error) {
	client, err := oss.New(Endpoint, AccessKeyId, AccessKeySecret)
	if err != nil {
		return "1", err
	}
	bucket, err := client.Bucket(Bucketname)
	if err != nil {
		return "2", err
	}
	path := UploadDir + time.Now().Format("200601/20060102/")
	path += filename
	err = bucket.PutObjectFromFile(path, filepath)
	if err != nil {
		return "3", err
	}
	path = ResourceHost + path
	return path, err
}

const (
	HzEndpoint          = "oss-cn-shanghai.aliyuncs.com"
	// todo 修改bucket空间
	HzBucketName string = "hzchart"
)

//上传文件到阿里云
func UploadFileToAliyun(filename, filepath, savePath string) error {
	defer func() {
		os.Remove(filepath)
	}()
	client, err := oss.New(HzEndpoint, AccessKeyId, AccessKeySecret)
	if err != nil {
		return err
	}
	bucket, err := client.Bucket(HzBucketName)
	if err != nil {
		return err
	}
	//path := utils.Upload_Audio_Dir + time.Now().Format("200601/20060102/")
	//path += filename
	err = bucket.PutObjectFromFile(savePath, filepath)
	if err != nil {
		return err
	}
	return err
}
type STSToken struct {
	AccessKeyId     string `json:"access_key_id"`
	AccessKeySecret string `json:"access_key_secret"`
	SecurityToken   string `json:"security_token"`
	ExpiredTime     string `json:"expired_time"`
}

// GetOssSTSToken 获取STSToken
func GetOssSTSToken() (item *STSToken, err error) {
	defer func() {
		if err != nil {
			global.LOG.Info("GetOssSTSToken Err:"+err.Error())
			go alarm_msg.SendAlarmMsg("获取STSToken失败, ErrMsg: "+err.Error(), 3)
		}
	}()
	item = new(STSToken)
	// 获取缓存中的Token
	recent, _ := global.Redis.Get(context.TODO(), STSTokenCacheKey).Result()
	if recent != "" {
		lastToken := new(STSToken)
		if e := json.Unmarshal([]byte(recent), &lastToken); e != nil {
			err = errors.New("GetOssSTSToken lastToken Unmarshal Err: " + e.Error())
			return
		}
		// 未防止正在上传大文件时Token过期, 将判定的过期时间提前10分钟
		afterTime := time.Now().Local().Add(10 * time.Minute)
		expired, e := time.ParseInLocation(utils.FormatDateTime, lastToken.ExpiredTime, time.Local)
		if e != nil {
			err = errors.New("GetOssSTSToken expiredTime Parse Err: " + e.Error())
			return
		}
		if expired.After(afterTime) {
			item.AccessKeyId = lastToken.AccessKeyId
			item.AccessKeySecret = lastToken.AccessKeySecret
			item.SecurityToken = lastToken.SecurityToken
			item.ExpiredTime = lastToken.ExpiredTime
			return
		}
	}
	// 已过期则获取新的token
	newToken, e := NewSTSToken()
	if e != nil {
		err = errors.New("GetOssSTSToken NewSTSToken Err: " + e.Error())
		return
	}
	newTokenJson, e := json.Marshal(newToken)
	if e != nil {
		err = errors.New("GetOssSTSToken NewToken JSON Err: " + e.Error())
		return
	}
	// 覆盖缓存

	if statusCmd := global.Redis.SetEX(context.TODO(), STSTokenCacheKey, newTokenJson, time.Hour); statusCmd.Err() != nil {
		err = errors.New("GetOssSTSToken SetRedis Err: " + statusCmd.Err().Error())
		return
	}
	item = newToken
	return
}

// NewSTSToken 获取一个新的STSToken
func NewSTSToken() (item *STSToken, err error) {
	defer func() {
		if err != nil {
			global.LOG.Info("NewSTSToken Err:"+err.Error())
		}
	}()
	item = new(STSToken)
	client, e := sts.NewClientWithAccessKey("cn-shanghai", RAMAccessKeyId, RAMAccessKeySecret)
	if e != nil {
		err = errors.New("NewSTSToken NewClient Err: " + e.Error())
		return
	}
	request := sts.CreateAssumeRoleRequest()
	request.Scheme = "https"
	request.RegionId = "cn-shanghai"
	request.RoleArn = RoleArn
	now := time.Now().Format(utils.FormatDateTimeUnSpace)
	request.RoleSessionName = RoleSessionName + now
	request.DurationSeconds = "3600"

	response, e := client.AssumeRole(request)
	if e != nil {
		err = errors.New("NewSTSToken AssumeRole Err: " + e.Error())
		return
	}
	if response != nil {
		item.AccessKeyId = response.Credentials.AccessKeyId
		item.AccessKeySecret = response.Credentials.AccessKeySecret
		item.SecurityToken = response.Credentials.SecurityToken
		t, _ := time.Parse(time.RFC3339, response.Credentials.Expiration)
		expiration := t.In(time.Local)
		item.ExpiredTime = expiration.Format(utils.FormatDateTime)
	}
	return
}