88 lines
1.9 KiB
Go
88 lines
1.9 KiB
Go
package ali
|
|
|
|
import (
|
|
"context"
|
|
"path/filepath"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/aliyun/alibabacloud-oss-go-sdk-v2/oss"
|
|
)
|
|
|
|
type OSSClient struct {
|
|
client *oss.Client
|
|
internalClient *oss.Client
|
|
config *Config
|
|
}
|
|
|
|
func (c *OSSClient) GetSavePath(path string) string {
|
|
return filepath.Join("quyun", strings.Trim(path, "/"))
|
|
}
|
|
|
|
func (c *OSSClient) GetClient() *oss.Client {
|
|
return c.client
|
|
}
|
|
|
|
func (c *OSSClient) PreSignUpload(ctx context.Context, path, mimeType string) (*oss.PresignResult, error) {
|
|
request := &oss.PutObjectRequest{
|
|
Bucket: oss.Ptr(c.config.Bucket),
|
|
Key: oss.Ptr(c.GetSavePath(path)),
|
|
ContentType: oss.Ptr(mimeType),
|
|
}
|
|
return c.client.Presign(ctx, request)
|
|
}
|
|
|
|
func (c *OSSClient) Download(ctx context.Context, path, dest string) error {
|
|
request := &oss.GetObjectRequest{
|
|
Bucket: oss.Ptr(c.config.Bucket),
|
|
Key: oss.Ptr(path),
|
|
}
|
|
|
|
_, err := c.client.GetObjectToFile(ctx, request, dest)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// GetSignedUrl
|
|
func (c *OSSClient) GetSignedUrl(ctx context.Context, path string) (string, error) {
|
|
request := &oss.GetObjectRequest{
|
|
Bucket: oss.Ptr(c.config.Bucket),
|
|
Key: oss.Ptr(path),
|
|
}
|
|
|
|
preSign, err := c.client.Presign(ctx, request, oss.PresignExpires(time.Minute*5))
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return preSign.URL, nil
|
|
}
|
|
|
|
// Delete
|
|
func (c *OSSClient) Delete(ctx context.Context, path string) error {
|
|
request := &oss.DeleteObjectRequest{
|
|
Bucket: oss.Ptr(c.config.Bucket),
|
|
Key: oss.Ptr(path),
|
|
}
|
|
|
|
_, err := c.internalClient.DeleteObject(ctx, request)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Upload
|
|
func (c *OSSClient) Upload(ctx context.Context, input, dst string) error {
|
|
request := &oss.PutObjectRequest{
|
|
Bucket: oss.Ptr(c.config.Bucket),
|
|
Key: oss.Ptr(c.GetSavePath(dst)),
|
|
}
|
|
|
|
if _, err := c.internalClient.PutObjectFromFile(ctx, request, input); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|