diff --git a/common_fun/file_func.go b/common_fun/file_func.go new file mode 100644 index 0000000..f6cd885 --- /dev/null +++ b/common_fun/file_func.go @@ -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 +} diff --git a/common_fun/file_func_test.go b/common_fun/file_func_test.go new file mode 100644 index 0000000..1d2bda2 --- /dev/null +++ b/common_fun/file_func_test.go @@ -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) + } +} diff --git a/go.mod b/go.mod index 816fc7d..2b93510 100644 --- a/go.mod +++ b/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 ) diff --git a/go.sum b/go.sum index b725020..cb80986 100644 --- a/go.sum +++ b/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= diff --git a/oss_tool/aliyun_oss.go b/oss_tool/aliyun_oss.go new file mode 100644 index 0000000..06e6b49 --- /dev/null +++ b/oss_tool/aliyun_oss.go @@ -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 +} diff --git a/oss_tool/aliyun_oss_test.go b/oss_tool/aliyun_oss_test.go new file mode 100644 index 0000000..b94302e --- /dev/null +++ b/oss_tool/aliyun_oss_test.go @@ -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) + } +}