41 lines
879 B
Go
41 lines
879 B
Go
package ali
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
|
|
"github.com/aliyun/alibabacloud-oss-go-sdk-v2/oss"
|
|
)
|
|
|
|
type OSSClient struct {
|
|
client *oss.Client
|
|
internalClient *oss.Client
|
|
config *Config
|
|
}
|
|
|
|
func (c *OSSClient) GetClient() *oss.Client {
|
|
return c.client
|
|
}
|
|
|
|
func (c *OSSClient) PreSignUpload(ctx context.Context, path string) (*oss.PresignResult, error) {
|
|
request := &oss.PutObjectRequest{
|
|
Bucket: oss.Ptr(c.config.Bucket),
|
|
Key: oss.Ptr("quyun/" + strings.Trim(path, "/")),
|
|
ContentType: oss.Ptr("multipart/form-data"),
|
|
}
|
|
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
|
|
}
|