阿里云oss
This commit is contained in:
lzh
35
common_fun/file_func.go
Normal file
35
common_fun/file_func.go
Normal file
@@ -0,0 +1,35 @@
|
||||
package cf
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
)
|
||||
|
||||
// EnsureDirExists 确保目录存在，如果不存在则创建
|
||||
func EnsureDirExists(dirPath string) error {
|
||||
// 判断路径是否存在
|
||||
_, err := os.Stat(dirPath)
|
||||
if os.IsNotExist(err) {
|
||||
// 目录不存在，递归创建
|
||||
err = os.MkdirAll(dirPath, os.ModePerm)
|
||||
if err != nil {
|
||||
return fmt.Errorf("创建目录失败，目录：%s，err： %w", dirPath, err)
|
||||
}
|
||||
} else if err != nil {
|
||||
// 其他错误（如权限问题）
|
||||
return fmt.Errorf("检查目录失败，目录：%s，err： %w", dirPath, err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// EnsureParentDirExists 确保文件的目录存在，如果不存在则创建
|
||||
func EnsureParentDirExists(filePath string) error {
|
||||
// 提取父级目录路径
|
||||
dir := filepath.Dir(filePath)
|
||||
err := EnsureDirExists(dir)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
19
common_fun/file_func_test.go
Normal file
19
common_fun/file_func_test.go
Normal file
@@ -0,0 +1,19 @@
|
||||
package cf
|
||||
|
||||
import "testing"
|
||||
|
||||
func TestEnsureDirExists(t *testing.T) {
|
||||
url := "/test"
|
||||
err := EnsureDirExists(url)
|
||||
if err != nil {
|
||||
t.Errorf("EnsureDirExists() error = %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestEnsureParentDirExists(t *testing.T) {
|
||||
filePath := "/test/test1/test.txt"
|
||||
err := EnsureParentDirExists(filePath)
|
||||
if err != nil {
|
||||
t.Errorf("EnsureDirExists() error = %v", err)
|
||||
}
|
||||
}
|
||||
2
go.mod
2
go.mod
@@ -9,6 +9,7 @@ require (
|
||||
github.com/alibabacloud-go/dysmsapi-20170525/v5 v5.1.1
|
||||
github.com/alibabacloud-go/tea v1.3.8
|
||||
github.com/alibabacloud-go/tea-utils/v2 v2.0.7
|
||||
github.com/aliyun/alibabacloud-oss-go-sdk-v2 v1.2.3
|
||||
github.com/pkg/errors v0.9.1
|
||||
github.com/redis/go-redis/v9 v9.10.0
|
||||
github.com/stretchr/testify v1.10.0
|
||||
@@ -30,6 +31,7 @@ require (
|
||||
github.com/pmezard/go-difflib v1.0.0 // indirect
|
||||
github.com/tjfoc/gmsm v1.4.1 // indirect
|
||||
golang.org/x/net v0.41.0 // indirect
|
||||
golang.org/x/time v0.4.0 // indirect
|
||||
gopkg.in/ini.v1 v1.67.0 // indirect
|
||||
gopkg.in/yaml.v3 v3.0.1 // indirect
|
||||
)
|
||||
|
||||
9
go.sum
9
go.sum
@@ -44,13 +44,17 @@ github.com/alibabacloud-go/tea-utils/v2 v2.0.6/go.mod h1:qxn986l+q33J5VkialKMqT/
|
||||
github.com/alibabacloud-go/tea-utils/v2 v2.0.7 h1:WDx5qW3Xa5ZgJ1c8NfqJkF6w+AU5wB8835UdhPr6Ax0=
|
||||
github.com/alibabacloud-go/tea-utils/v2 v2.0.7/go.mod h1:qxn986l+q33J5VkialKMqT/TTs3E+U9MJpd001iWQ9I=
|
||||
github.com/alibabacloud-go/tea-xml v1.1.3/go.mod h1:Rq08vgCcCAjHyRi/M7xlHKUykZCEtyBy9+DPF6GgEu8=
|
||||
github.com/aliyun/alibabacloud-oss-go-sdk-v2 v1.2.3 h1:LyeTJauAchnWdre3sAyterGrzaAtZ4dSNoIvDvaWfo4=
|
||||
github.com/aliyun/alibabacloud-oss-go-sdk-v2 v1.2.3/go.mod h1:FTzydeQVmR24FI0D6XWUOMKckjXehM/jgMn1xC+DA9M=
|
||||
github.com/aliyun/credentials-go v1.1.2/go.mod h1:ozcZaMR5kLM7pwtCMEpVmQ242suV6qTJya2bDq4X1Tw=
|
||||
github.com/aliyun/credentials-go v1.3.1/go.mod h1:8jKYhQuDawt8x2+fusqa1Y6mPxemTsBEN04dgcAcYz0=
|
||||
github.com/aliyun/credentials-go v1.3.6/go.mod h1:1LxUuX7L5YrZUWzBrRyk0SwSdH4OmPrib8NVePL3fxM=
|
||||
github.com/aliyun/credentials-go v1.4.5 h1:O76WYKgdy1oQYYiJkERjlA2dxGuvLRrzuO2ScrtGWSk=
|
||||
github.com/aliyun/credentials-go v1.4.5/go.mod h1:Jm6d+xIgwJVLVWT561vy67ZRP4lPTQxMbEYRuT2Ti1U=
|
||||
github.com/bsm/ginkgo/v2 v2.12.0 h1:Ny8MWAHyOepLGlLKYmXG4IEkioBysk6GpaRTLC8zwWs=
|
||||
github.com/bsm/ginkgo/v2 v2.12.0/go.mod h1:SwYbGRRDovPVboqFv0tPTcG1sN61LM1Z4ARdbAV9g4c=
|
||||
github.com/bsm/gomega v1.27.10 h1:yeMWxP2pV2fG3FgAODIY8EiRE3dy0aeFYt4l7wh6yKA=
|
||||
github.com/bsm/gomega v1.27.10/go.mod h1:JyEr/xRbxbtgWNi8tIEVPUYZ5Dzef52k01W3YH0H+O0=
|
||||
github.com/census-instrumentation/opencensus-proto v0.2.1/go.mod h1:f6KPmirojxKA12rnyqOA5BBL4O983OfeGPqjHWSTneU=
|
||||
github.com/cespare/xxhash/v2 v2.3.0 h1:UL815xU9SqsFlibzuggzjXhog7bL6oX9BbNZnL2UFvs=
|
||||
github.com/cespare/xxhash/v2 v2.3.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
|
||||
@@ -169,10 +173,7 @@ golang.org/x/net v0.20.0/go.mod h1:z8BVo6PvndSri0LbOE3hAn0apkU+1YvI6E70E9jsnvY=
|
||||
golang.org/x/net v0.21.0/go.mod h1:bIjVDfnllIU7BJ2DNgfnXvpSvtn8VRwhlsaeUTyUS44=
|
||||
golang.org/x/net v0.23.0/go.mod h1:JKghWKKOSdJwpW2GEx0Ja7fmaKnMsbu+MWVZTokSYmg=
|
||||
golang.org/x/net v0.25.0/go.mod h1:JkAGAh7GEvH74S6FOH42FLoXpXbE/aqXSrIQjXgsiwM=
|
||||
golang.org/x/net v0.26.0 h1:soB7SVo0PWrY4vPW/+ay0jKDNScG2X9wFeYlXIvJsOQ=
|
||||
golang.org/x/net v0.26.0/go.mod h1:5YKkiSynbBIh3p6iOc/vibscux0x38BZDkn8sCUPxHE=
|
||||
golang.org/x/net v0.37.1-0.20250305215238-2914f4677317 h1:wneCP+2d9NUmndnyTmY7VwUNYiP26xiN/AtdcojQ1lI=
|
||||
golang.org/x/net v0.37.1-0.20250305215238-2914f4677317/go.mod h1:ivrbrMbzFq5J41QOQh0siUuly180yBYtLp+CKbEaFx8=
|
||||
golang.org/x/net v0.41.0 h1:vBTly1HeNPEn3wtREYfy4GZ/NECgw2Cnl+nK6Nz3uvw=
|
||||
golang.org/x/net v0.41.0/go.mod h1:B/K4NNqkfmg07DQYrbwvSluqCJOOXwUjeb/5lOisjbA=
|
||||
golang.org/x/oauth2 v0.0.0-20180821212333-d2e6202438be/go.mod h1:N/0e6XlmueqKjAGxoOufVs8QHGRruUQn6yWY3a++T0U=
|
||||
@@ -227,6 +228,8 @@ golang.org/x/text v0.13.0/go.mod h1:TvPlkZtksWOMsz7fbANvkp4WM8x/WCo/om8BMLbz+aE=
|
||||
golang.org/x/text v0.14.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU=
|
||||
golang.org/x/text v0.15.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU=
|
||||
golang.org/x/text v0.16.0/go.mod h1:GhwF1Be+LQoKShO3cGOHzqOgRrGaYc9AvblQOmPVHnI=
|
||||
golang.org/x/time v0.4.0 h1:Z81tqI5ddIoXDPvVQ7/7CC9TnLM7ubaFG2qXYd5BbYY=
|
||||
golang.org/x/time v0.4.0/go.mod h1:3BpzKBy/shNhVucY/MWOyx10tF3SFh9QdLuxbVysPQM=
|
||||
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
||||
golang.org/x/tools v0.0.0-20190114222345-bf090417da8b/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
||||
golang.org/x/tools v0.0.0-20190226205152-f727befe758c/go.mod h1:9Yl7xja0Znq3iFh3HoIrodX9oNMXvdceNzlUR8zjMvY=
|
||||
|
||||
147
oss_tool/aliyun_oss.go
Normal file
147
oss_tool/aliyun_oss.go
Normal file
@@ -0,0 +1,147 @@
|
||||
package oss_tool
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"fmt"
|
||||
cf "git.ssgfgtfy.com/public/ssgf_utils/common_fun"
|
||||
"github.com/aliyun/alibabacloud-oss-go-sdk-v2/oss"
|
||||
"github.com/aliyun/alibabacloud-oss-go-sdk-v2/oss/credentials"
|
||||
"image"
|
||||
"image/jpeg"
|
||||
"image/png"
|
||||
"io"
|
||||
"os"
|
||||
"time"
|
||||
)
|
||||
|
||||
type ALiYunOSSClient struct {
|
||||
AccessKeyID string
|
||||
AccessKeySecret string
|
||||
Region string
|
||||
ossClient *oss.Client
|
||||
}
|
||||
|
||||
func (c *ALiYunOSSClient) NewAliYunOSS() (err error) {
|
||||
if c.AccessKeyID == "" || c.AccessKeySecret == "" {
|
||||
return errors.New("请配置 oss accessKeyID accessKeySecret")
|
||||
}
|
||||
if c.Region == "" {
|
||||
return errors.New("请配置 oss region")
|
||||
}
|
||||
provider := credentials.NewStaticCredentialsProvider(c.AccessKeyID, c.AccessKeySecret)
|
||||
cfg := oss.LoadDefaultConfig().WithSignatureVersion(oss.SignatureVersionV4).WithCredentialsProvider(provider).WithRegion(c.Region)
|
||||
c.ossClient = oss.NewClient(cfg)
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c *ALiYunOSSClient) GetSignUrl(bucket string, key string, expires time.Duration) (result *oss.PresignResult, err error) {
|
||||
// 生成PutObject的预签名URL
|
||||
result, err = c.ossClient.Presign(
|
||||
context.Background(),
|
||||
&oss.PutObjectRequest{
|
||||
Bucket: oss.Ptr(bucket),
|
||||
Key: oss.Ptr(key),
|
||||
//ContentType: oss.Ptr("application/octet-stream"),
|
||||
},
|
||||
oss.PresignExpires(expires),
|
||||
)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// PutForLocalFile 上传本地文件
|
||||
func (c *ALiYunOSSClient) PutForLocalFile(bucket, key, path string) (result *oss.PutObjectResult, err error) {
|
||||
// 创建上传对象的请求
|
||||
putRequest := &oss.PutObjectRequest{
|
||||
Bucket: oss.Ptr(bucket), // 存储空间名称
|
||||
Key: oss.Ptr(key), // 对象名称
|
||||
StorageClass: oss.StorageClassStandard, // 指定对象的存储类型为标准存储
|
||||
ContentType: oss.Ptr("application/octet-stream"),
|
||||
}
|
||||
// 执行上传对象的请求
|
||||
result, err = c.ossClient.PutObjectFromFile(context.Background(), putRequest, path)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// GetObject 下载文件,返回的是一个*oss.GetObjectResult （不推荐使用，如果使用的话，需要手动调用object.Body.Close()手动关闭资源）
|
||||
func (c *ALiYunOSSClient) GetObject(bucket string, key string) (object *oss.GetObjectResult, body io.ReadCloser, err error) {
|
||||
getRequest := &oss.GetObjectRequest{
|
||||
Bucket: oss.Ptr(bucket), // 存储空间名称
|
||||
Key: oss.Ptr(key), // 对象名称
|
||||
}
|
||||
// 执行获取对象的操作并处理结果
|
||||
result, err := c.ossClient.GetObject(context.TODO(), getRequest)
|
||||
if err != nil {
|
||||
return nil, nil, err
|
||||
}
|
||||
//defer func() {
|
||||
// _ = result.Body.Close() // 确保在函数结束时关闭响应体
|
||||
//}()
|
||||
return result, result.Body, nil
|
||||
}
|
||||
|
||||
// GetObjectToFile 获取对象并保存到本地
|
||||
func (c *ALiYunOSSClient) GetObjectToFile(bucket string, key string, path string) (err error) {
|
||||
getRequest := &oss.GetObjectRequest{
|
||||
Bucket: oss.Ptr(bucket), // 存储空间名称
|
||||
Key: oss.Ptr(key), // 对象名称
|
||||
}
|
||||
// 执行获取对象的操作并处理结果
|
||||
result, err := c.ossClient.GetObject(context.TODO(), getRequest)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer func() {
|
||||
_ = result.Body.Close() // 确保在函数结束时关闭响应体
|
||||
}()
|
||||
// 确保文件目录存在
|
||||
err = cf.EnsureParentDirExists(path)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
// 创建目标文件
|
||||
file, err := os.Create(path)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer file.Close()
|
||||
|
||||
// 将 reader 中的内容复制到文件中
|
||||
_, err = io.Copy(file, result.Body)
|
||||
return nil
|
||||
}
|
||||
|
||||
// GetObjectToImage 获取对象并保存为图片对象
|
||||
func (c *ALiYunOSSClient) GetObjectToImage(bucket string, key string) (img image.Image, err error) {
|
||||
getRequest := &oss.GetObjectRequest{
|
||||
Bucket: oss.Ptr(bucket), // 存储空间名称
|
||||
Key: oss.Ptr(key), // 对象名称
|
||||
}
|
||||
// 执行获取对象的操作并处理结果
|
||||
result, err := c.ossClient.GetObject(context.TODO(), getRequest)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer func() {
|
||||
_ = result.Body.Close() // 确保在函数结束时关闭响应体
|
||||
}()
|
||||
img, _, err = image.Decode(result.Body)
|
||||
if err != nil {
|
||||
// 尝试 JPEG 解码
|
||||
img, err = jpeg.Decode(result.Body)
|
||||
if err != nil {
|
||||
// 尝试 PNG 解码
|
||||
img, err = png.Decode(result.Body)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("图片解码失败（支持的格式：JPEG/PNG）: %v", err)
|
||||
}
|
||||
}
|
||||
}
|
||||
return img, nil
|
||||
}
|
||||
75
oss_tool/aliyun_oss_test.go
Normal file
75
oss_tool/aliyun_oss_test.go
Normal file
@@ -0,0 +1,75 @@
|
||||
package oss_tool
|
||||
|
||||
import (
|
||||
"os"
|
||||
"testing"
|
||||
)
|
||||
|
||||
var (
|
||||
accessKeyId = os.Getenv("OSS_ACCESS_KEY_ID")
|
||||
accessKeySecret = os.Getenv("OSS_SECRET_ACCESS_KEY")
|
||||
region = os.Getenv("OSS_REGION")
|
||||
client = &ALiYunOSSClient{
|
||||
AccessKeyID: accessKeyId,
|
||||
AccessKeySecret: accessKeySecret,
|
||||
Region: region,
|
||||
}
|
||||
)
|
||||
|
||||
func TestALiYunOSSClient_NewAliYunOSS(t *testing.T) {
|
||||
err := client.NewAliYunOSS()
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
t.Log(client.ossClient)
|
||||
}
|
||||
|
||||
func TestALiYunOSSClient_GetSignUrl(t *testing.T) {
|
||||
err := client.NewAliYunOSS()
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
sign, err := client.GetSignUrl("", "test/upload/bizhi1.jpg", 0)
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
t.Log(sign)
|
||||
}
|
||||
|
||||
func TestALiYunOSSClient_PutForLocalFile(t *testing.T) {
|
||||
err := client.NewAliYunOSS()
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
result, err := client.PutForLocalFile("", "test/upload/bizhi2.jpg", "C:\\Users\\Administrator\\Desktop\\壁纸1.jpg")
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
t.Log(result)
|
||||
}
|
||||
|
||||
func TestALiYunOSSClient_GetObjectToFile(t *testing.T) {
|
||||
err := client.NewAliYunOSS()
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
err = client.GetObjectToFile("", "test/upload/bizhi1.jpg", "D:/bizhi1.jpg")
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
} else {
|
||||
t.Log("成功")
|
||||
}
|
||||
}
|
||||
|
||||
func TestALiYunOSSClient_GetObjectToImage(t *testing.T) {
|
||||
err := client.NewAliYunOSS()
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
img, err := client.GetObjectToImage("", "test/upload/bizhi1.jpg")
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
} else {
|
||||
t.Log(img)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user