58 lines
1.7 KiB
Go
58 lines
1.7 KiB
Go
|
package dfs
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"bytes"
|
|||
|
|
"context"
|
|||
|
|
"fmt"
|
|||
|
|
"github.com/tencentyun/cos-go-sdk-v5"
|
|||
|
|
"io"
|
|||
|
|
"net/http"
|
|||
|
|
"net/url"
|
|||
|
|
"time"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type cosClient struct {
|
|||
|
|
client *cos.Client
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func newCos(secretId, secretKey, bucket string, region string) IDfs {
|
|||
|
|
c := new(cosClient)
|
|||
|
|
//u, _ := url.Parse(fmt.Sprintf("https://ttsj-1359624709.cos.ap-chengdu.myqcloud.com", bucket, region))
|
|||
|
|
u, _ := url.Parse(fmt.Sprintf("https://%v.cos.%v.myqcloud.com", bucket, region))
|
|||
|
|
b := &cos.BaseURL{BucketURL: u}
|
|||
|
|
client := cos.NewClient(b, &http.Client{
|
|||
|
|
Transport: &cos.AuthorizationTransport{
|
|||
|
|
SecretID: secretId, // 用户的 SecretId，建议使用子账号密钥，授权遵循最小权限指引，降低使用风险。子账号密钥获取可参考 https://cloud.tencent.com/document/product/598/37140
|
|||
|
|
SecretKey: secretKey, // 用户的 SecretKey，建议使用子账号密钥，授权遵循最小权限指引，降低使用风险。子账号密钥获取可参考 https://cloud.tencent.com/document/product/598/37140
|
|||
|
|
},
|
|||
|
|
})
|
|||
|
|
c.client = client
|
|||
|
|
|
|||
|
|
return c
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (c *cosClient) GetObj(path string) ([]byte, error) {
|
|||
|
|
ctx, cf := context.WithTimeout(context.Background(), time.Second*10)
|
|||
|
|
defer cf()
|
|||
|
|
rsp, err := c.client.Object.Get(ctx, path, nil)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, fmt.Errorf("get cos object %v error:%v", path, err)
|
|||
|
|
}
|
|||
|
|
rspBin, err := io.ReadAll(rsp.Body)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, fmt.Errorf("read response obj %v error:%v", path, err)
|
|||
|
|
}
|
|||
|
|
return rspBin, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (c *cosClient) PutObj(path string, content []byte) error {
|
|||
|
|
ctx, cf := context.WithTimeout(context.Background(), time.Second*10)
|
|||
|
|
defer cf()
|
|||
|
|
buffer := bytes.NewBuffer(content)
|
|||
|
|
_, err := c.client.Object.Put(ctx, path, buffer, nil)
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (c *cosClient) Close() {
|
|||
|
|
}
|