data_ai/ul/oss.go

package ul

import (
	"bytes"
	"fmt"
	"github.com/aliyun/aliyun-oss-go-sdk/oss"
	"io"
	util "jygit.jydev.jianyu360.cn/data_processing/common_utils"
	"log"
	"os"
)

var (
	ossEndpoint        = "oss-cn-beijing-internal.aliyuncs.com" //http://oss-cn-beijing.aliyuncs.com"
	ossEndpointTest    = "oss-cn-beijing.aliyuncs.com"
	ossAccessKeyId     = "LTAI4G5x9aoZx8dDamQ7vfZi"
	ossAccessKeySecret = "Bk98FsbPYXcJe72n1bG3Ssf73acuNh"
	ossBucketName      = "jy-datadetail"
	ossclient          *oss.Client
)

func InitOss(isTest bool) {
	if isTest {
		ossEndpoint = ossEndpointTest //测试服务器~阿里云
	}
	client, err := oss.New(ossEndpoint, ossAccessKeyId, ossAccessKeySecret)
	if err != nil {
		log.Println("Error:", err)
		os.Exit(-1)
	}
	ossclient = client
}

func OssGetObject(objectName string) string {
	util.Catch()
	// 获取存储空间。
	bucket, err := ossclient.Bucket(ossBucketName)
	if err != nil {
		fmt.Println("Error:", err)
		return ""
	}
	// 下载文件到流。
	body, err := bucket.GetObject(objectName)
	if err != nil {
		fmt.Println("Error:", err)
		return ""
	}
	defer body.Close()
	data, err := io.ReadAll(body)
	if err != nil {
		fmt.Println("Error:", err)
		return ""
	}
	return string(data)
}

func OssPutObject(bucketName, objectKey, text string) (bool, error) {
	defer util.Catch()
	// 获取存储空间。
	bucket, err := ossclient.Bucket(bucketName)
	if err != nil {
		log.Println("Error:", err)
		return false, err
	}
	// 上传文件流。
	err = bucket.PutObject(objectKey, bytes.NewReader([]byte(text)))
	if err != nil {
		log.Println("Error:", err)
		return false, err
	}
	return true, nil
}