You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
56 lines
1.5 KiB
56 lines
1.5 KiB
package gostorage
|
|
|
|
import (
|
|
"context"
|
|
"github.com/baidubce/bce-sdk-go/bce"
|
|
"github.com/baidubce/bce-sdk-go/services/bos"
|
|
"io"
|
|
)
|
|
|
|
// Baidu 百度云
|
|
type Baidu struct {
|
|
AccessKey string
|
|
SecretAccessKey string
|
|
Endpoint string
|
|
BucketName string
|
|
error error // 错误信息
|
|
client *bos.Client // 驱动
|
|
}
|
|
|
|
// NewBaidu 初始化
|
|
// https://cloud.baidu.com/doc/BOS/s/4jwvyry1p
|
|
func NewBaidu(ctx context.Context, accessKey string, secretAccessKey, endpoint, bucketName string) *Baidu {
|
|
app := &Baidu{AccessKey: accessKey, SecretAccessKey: secretAccessKey, Endpoint: endpoint, BucketName: bucketName}
|
|
clientConfig := bos.BosClientConfiguration{
|
|
Ak: accessKey,
|
|
Sk: secretAccessKey,
|
|
Endpoint: endpoint,
|
|
RedirectDisabled: false,
|
|
}
|
|
app.client, app.error = bos.NewClientWithConfig(&clientConfig)
|
|
return app
|
|
}
|
|
|
|
// Bucket 存储空间
|
|
func (c *Baidu) Bucket(name string) *Baidu {
|
|
c.BucketName = name
|
|
return c
|
|
}
|
|
|
|
// PutObject 上传文件流
|
|
// @param file 文件流
|
|
// @param filePath 文件路径
|
|
// @param fileName 文件名称
|
|
func (c *Baidu) PutObject(ctx context.Context, file io.Reader, filePath, fileName string) (resp FileInfo, err error) {
|
|
objectKey := filePath
|
|
if fileName != "" {
|
|
objectKey = filePath + "/" + fileName
|
|
}
|
|
bodyStream, err := bce.NewBodyFromSizedReader(file, -1)
|
|
_, err = c.client.PutObject(c.BucketName, objectKey, bodyStream, nil)
|
|
resp.Path = filePath
|
|
resp.Name = fileName
|
|
resp.Url = objectKey
|
|
return
|
|
}
|