es_index_data/createEsIndex/oss/ossclient.go

package oss

import (
	"esindex/config"
	"fmt"
	"github.com/aliyun/aliyun-oss-go-sdk/oss"
	"go.uber.org/zap"
	"io/ioutil"
	util "jygit.jydev.jianyu360.cn/data_processing/common_utils"
	"jygit.jydev.jianyu360.cn/data_processing/common_utils/log"
	"os"
	"time"
)

var (
	//ossEndpoint        = "oss-cn-beijing-internal.aliyuncs.com" //正式环境用：oss-cn-beijing-internal.aliyuncs.com 测试：oss-cn-beijing.aliyuncs.com
	//ossAccessKeyId     = "LTAI4G5x9aoZx8dDamQ7vfZi"
	//ossAccessKeySecret = "Bk98FsbPYXcJe72n1bG3Ssf73acuNh"
	//ossBucketName      = "topjy"
	ossclient *oss.Client
)

func InitOss() {
	now := time.Now()
	if config.Conf.DB.Oss.Endpoint == "" || config.Conf.DB.Oss.AccessKey == "" || config.Conf.DB.Oss.AccessSecret == "" {
		log.Info("InitOss", zap.String("OSS:", "配置信息为空"))
		os.Exit(-1)
	}
	client, err := oss.New(config.Conf.DB.Oss.Endpoint, config.Conf.DB.Oss.AccessKey, config.Conf.DB.Oss.AccessSecret)
	if err != nil {
		fmt.Println("Error:", err)
		os.Exit(-1)
	}
	ossclient = client

	log.Info("InitOss", zap.Any("duration", time.Since(now).Seconds()))
}

func OssGetObject(objectName, id string) string {
	util.Catch()
	// 获取存储空间。
	bucket, err := ossclient.Bucket(config.Conf.DB.Oss.BucketName)
	if err != nil {
		return ""
	}

	// 下载文件到流。
	body, err := bucket.GetObject(objectName)
	if err != nil {
		//log.Info("OssGetObject", zap.String("key", objectName), zap.String("id", id))
		//log.Info("OssGetObject", zap.Error(err))
		return ""
	}
	defer body.Close()
	data, err := ioutil.ReadAll(body)
	if err != nil {
		log.Info("OssGetObject", zap.Error(err))
		return ""
	}
	return string(data)
}

func OssObjExists(bname, fid string) bool {
	util.Catch()
	// 获取存储空间。
	bucket, err := ossclient.Bucket(bname)
	if err != nil {
		fmt.Println("Error:", err)
	}
	// 判断文件是否存在。
	isExist, err := bucket.IsObjectExist(fid)
	if err != nil {
		fmt.Println("Error:", err)
	}
	return isExist
}