agin719
4 years ago
committed by
GitHub
GitHub
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
19 changed files with 845 additions and 105 deletions
-
4batch.go
-
102bucket_test.go
-
17ci.go
-
271ci_doc.go
-
2cos.go
-
2costesting/ci_test.go
-
14example/bucket/intelligenttiering.go
-
132example/object/ci_doc_process.go
-
25example/object/get.go
-
17example/object/put.go
-
59example/object/put_with_timeout.go
-
21example/object/upload.go
-
40example/object/uploadPart.go
-
28helper.go
-
61object.go
-
10object_part.go
-
2object_test.go
-
135progress.go
-
8service_test.go
@ -0,0 +1,271 @@ |
|||||
|
package cos |
||||
|
|
||||
|
import ( |
||||
|
"context" |
||||
|
"encoding/xml" |
||||
|
"net/http" |
||||
|
) |
||||
|
|
||||
|
type DocProcessJobInput struct { |
||||
|
Object string `xml:"Object,omitempty"` |
||||
|
} |
||||
|
|
||||
|
type DocProcessJobOutput struct { |
||||
|
Region string `xml:"Region,omitempty"` |
||||
|
Bucket string `xml:"Bucket,omitempty"` |
||||
|
Object string `xml:"Object,omitempty"` |
||||
|
} |
||||
|
|
||||
|
type DocProcessJobDocProcess struct { |
||||
|
SrcType string `xml:"SrcType,omitempty"` |
||||
|
TgtType string `xml:"TgtType,omitempty"` |
||||
|
SheetId int `xml:"SheetId,omitempty"` |
||||
|
StartPage int `xml:"StartPage,omitempty"` |
||||
|
EndPage int `xml:"EndPage,omitempty"` |
||||
|
ImageParams string `xml:"ImageParams,omitempty"` |
||||
|
DocPassword string `xml:"DocPassword,omitempty"` |
||||
|
Comments int `xml:"Comments,omitempty"` |
||||
|
PaperDirection int `xml:"PaperDirection,omitempty"` |
||||
|
Quality int `xml:"Quality,omitempty"` |
||||
|
Zoom int `xml:"Zoom,omitempty"` |
||||
|
} |
||||
|
|
||||
|
type DocProcessJobDocProcessResult struct { |
||||
|
FailPageCount int `xml:",omitempty"` |
||||
|
SuccPageCount int `xml:"SuccPageCount,omitempty"` |
||||
|
TaskId string `xml:"TaskId,omitempty"` |
||||
|
TgtType string `xml:"TgtType,omitempty"` |
||||
|
TotalPageCount int `xml:"TotalPageCount,omitempty"` |
||||
|
PageInfo struct { |
||||
|
PageNo int `xml:"PageNo,omitempty"` |
||||
|
TgtUri string `xml:"TgtUri,omitempty"` |
||||
|
} `xml:"PageInfo,omitempty"` |
||||
|
} |
||||
|
|
||||
|
type DocProcessJobOperation struct { |
||||
|
Output *DocProcessJobOutput `xml:"Output,omitempty"` |
||||
|
DocProcess *DocProcessJobDocProcess `xml:"DocProcess,omitempty"` |
||||
|
DocProcessResult *DocProcessJobDocProcessResult `xml:"DocProcessResult,omitempty"` |
||||
|
} |
||||
|
|
||||
|
type DocProcessJobDetail struct { |
||||
|
Code string `xml:"Code,omitempty"` |
||||
|
Message string `xml:"Message,omitempty"` |
||||
|
JobId string `xml:"JobId,omitempty"` |
||||
|
Tag string `xml:"Tag,omitempty"` |
||||
|
State string `xml:"State,omitempty"` |
||||
|
CreationTime string `xml:"CreationTime,omitempty"` |
||||
|
QueueId string `xml:"QueueId,omitempty"` |
||||
|
Input *DocProcessJobInput `xml:"Input,omitempty"` |
||||
|
Operation *DocProcessJobOperation `xml:"Operation,omitempty"` |
||||
|
} |
||||
|
|
||||
|
type CreateDocProcessJobsOptions struct { |
||||
|
XMLName xml.Name `xml:"Request"` |
||||
|
Tag string `xml:"Tag,omitempty"` |
||||
|
Input *DocProcessJobInput `xml:"Input,omitempty"` |
||||
|
Operation *DocProcessJobOperation `xml:"Operation,omitempty"` |
||||
|
QueueId string `xml:"QueueId,omitempty"` |
||||
|
} |
||||
|
|
||||
|
type CreateDocProcessJobsResult struct { |
||||
|
XMLName xml.Name `xml:"Response"` |
||||
|
JobsDetail DocProcessJobDetail `xml:"JobsDetail,omitempty"` |
||||
|
} |
||||
|
|
||||
|
func (s *CIService) CreateDocProcessJobs(ctx context.Context, opt *CreateDocProcessJobsOptions) (*CreateDocProcessJobsResult, *Response, error) { |
||||
|
var res CreateDocProcessJobsResult |
||||
|
sendOpt := sendOptions{ |
||||
|
baseURL: s.client.BaseURL.CIURL, |
||||
|
uri: "/doc_jobs", |
||||
|
method: http.MethodPost, |
||||
|
body: opt, |
||||
|
result: &res, |
||||
|
} |
||||
|
resp, err := s.client.send(ctx, &sendOpt) |
||||
|
return &res, resp, err |
||||
|
} |
||||
|
|
||||
|
type DescribeDocProcessJobResult struct { |
||||
|
XMLName xml.Name `xml:"Response"` |
||||
|
JobsDetail *DocProcessJobDetail `xml:"JobsDetail,omitempty"` |
||||
|
NonExistJobIds string `xml:"NonExistJobIds,omitempty"` |
||||
|
} |
||||
|
|
||||
|
func (s *CIService) DescribeDocProcessJob(ctx context.Context, jobid string) (*DescribeDocProcessJobResult, *Response, error) { |
||||
|
var res DescribeDocProcessJobResult |
||||
|
sendOpt := sendOptions{ |
||||
|
baseURL: s.client.BaseURL.CIURL, |
||||
|
uri: "/doc_jobs/" + jobid, |
||||
|
method: http.MethodGet, |
||||
|
result: &res, |
||||
|
} |
||||
|
resp, err := s.client.send(ctx, &sendOpt) |
||||
|
return &res, resp, err |
||||
|
} |
||||
|
|
||||
|
type DescribeDocProcessJobsOptions struct { |
||||
|
QueueId string `url:"queueId,omitempty"` |
||||
|
Tag string `url:"tag,omitempty"` |
||||
|
OrderByTime string `url:"orderByTime,omitempty"` |
||||
|
NextToken string `url:"nextToken,omitempty"` |
||||
|
Size int `url:"size,omitempty"` |
||||
|
States string `url:"states,omitempty"` |
||||
|
StartCreationTime string `url:"startCreationTime,omitempty"` |
||||
|
EndCreationTime string `url:"endCreationTime,omitempty"` |
||||
|
} |
||||
|
|
||||
|
type DescribeDocProcessJobsResult struct { |
||||
|
XMLName xml.Name `xml:"Response"` |
||||
|
JobsDetail []DocProcessJobDetail `xml:"JobsDetail,omitempty"` |
||||
|
NextToken string `xml:"NextToken,omitempty"` |
||||
|
} |
||||
|
|
||||
|
func (s *CIService) DescribeDocProcessJobs(ctx context.Context, opt *DescribeDocProcessJobsOptions) (*DescribeDocProcessJobsResult, *Response, error) { |
||||
|
var res DescribeDocProcessJobsResult |
||||
|
sendOpt := sendOptions{ |
||||
|
baseURL: s.client.BaseURL.CIURL, |
||||
|
uri: "/doc_jobs", |
||||
|
optQuery: opt, |
||||
|
method: http.MethodGet, |
||||
|
result: &res, |
||||
|
} |
||||
|
resp, err := s.client.send(ctx, &sendOpt) |
||||
|
return &res, resp, err |
||||
|
} |
||||
|
|
||||
|
type DescribeDocProcessQueuesOptions struct { |
||||
|
QueueIds string `url:"queueIds,omitempty"` |
||||
|
State string `url:"state,omitempty"` |
||||
|
PageNumber int `url:"pageNumber,omitempty"` |
||||
|
PageSize int `url:"pageSize,omitempty"` |
||||
|
} |
||||
|
|
||||
|
type DescribeDocProcessQueuesResult struct { |
||||
|
XMLName xml.Name `xml:"Response"` |
||||
|
RequestId string `xml:"RequestId,omitempty"` |
||||
|
TotalCount int `xml:"TotalCount,omitempty"` |
||||
|
PageNumber int `xml:"PageNumber,omitempty"` |
||||
|
PageSize int `xml:"PageSize,omitempty"` |
||||
|
QueueList []DocProcessQueue `xml:"QueueList,omitempty"` |
||||
|
NonExistPIDs []string `xml:"NonExistPIDs,omitempty"` |
||||
|
} |
||||
|
|
||||
|
type DocProcessQueue struct { |
||||
|
QueueId string `xml:"QueueId,omitempty"` |
||||
|
Name string `xml:"Name,omitempty"` |
||||
|
State string `xml:"State,omitempty"` |
||||
|
MaxSize int `xml:"MaxSize,omitempty"` |
||||
|
MaxConcurrent int `xml:"MaxConcurrent,omitempty"` |
||||
|
UpdateTime string `xml:"UpdateTime,omitempty"` |
||||
|
CreateTime string `xml:"CreateTime,omitempty"` |
||||
|
NotifyConfig *DocProcessQueueNotifyConfig `xml:"NotifyConfig,omitempty"` |
||||
|
} |
||||
|
|
||||
|
type DocProcessQueueNotifyConfig struct { |
||||
|
Url string `xml:"Url,omitempty"` |
||||
|
State string `xml:"State,omitempty"` |
||||
|
Type string `xml:"Type,omitempty"` |
||||
|
Event string `xml:"Event,omitempty"` |
||||
|
} |
||||
|
|
||||
|
func (s *CIService) DescribeDocProcessQueues(ctx context.Context, opt *DescribeDocProcessQueuesOptions) (*DescribeDocProcessQueuesResult, *Response, error) { |
||||
|
var res DescribeDocProcessQueuesResult |
||||
|
sendOpt := sendOptions{ |
||||
|
baseURL: s.client.BaseURL.CIURL, |
||||
|
uri: "/docqueue", |
||||
|
optQuery: opt, |
||||
|
method: http.MethodGet, |
||||
|
result: &res, |
||||
|
} |
||||
|
resp, err := s.client.send(ctx, &sendOpt) |
||||
|
return &res, resp, err |
||||
|
} |
||||
|
|
||||
|
type UpdateDocProcessQueueOptions struct { |
||||
|
XMLName xml.Name `xml:"Request"` |
||||
|
Name string `xml:"Name,omitempty"` |
||||
|
QueueID string `xml:"QueueID,omitempty"` |
||||
|
State string `xml:"State,omitempty"` |
||||
|
NotifyConfig *DocProcessQueueNotifyConfig `xml:"NotifyConfig,omitempty"` |
||||
|
} |
||||
|
|
||||
|
type UpdateDocProcessQueueResult struct { |
||||
|
XMLName xml.Name `xml:"Response"` |
||||
|
RequestId string `xml:"RequestId"` |
||||
|
Queue *DocProcessQueue `xml:"Queue"` |
||||
|
} |
||||
|
|
||||
|
func (s *CIService) UpdateDocProcessQueue(ctx context.Context, opt *UpdateDocProcessQueueOptions) (*UpdateDocProcessQueueResult, *Response, error) { |
||||
|
var res UpdateDocProcessQueueResult |
||||
|
sendOpt := sendOptions{ |
||||
|
baseURL: s.client.BaseURL.CIURL, |
||||
|
uri: "/docqueue/" + opt.QueueID, |
||||
|
body: opt, |
||||
|
method: http.MethodPut, |
||||
|
result: &res, |
||||
|
} |
||||
|
resp, err := s.client.send(ctx, &sendOpt) |
||||
|
return &res, resp, err |
||||
|
} |
||||
|
|
||||
|
type DescribeDocProcessBucketsOptions struct { |
||||
|
Regions string `url:"regions,omitempty"` |
||||
|
BucketNames string `url:"bucketNames,omitempty"` |
||||
|
BucketName string `url:"bucketName,omitempty"` |
||||
|
PageNumber int `url:"pageNumber,omitempty"` |
||||
|
PageSize int `url:"pageSize,omitempty"` |
||||
|
} |
||||
|
|
||||
|
type DescribeDocProcessBucketsResult struct { |
||||
|
XMLName xml.Name `xml:"Response"` |
||||
|
RequestId string `xml:"RequestId,omitempty"` |
||||
|
TotalCount int `xml:"TotalCount,omitempty"` |
||||
|
PageNumber int `xml:"PageNumber,omitempty"` |
||||
|
PageSize int `xml:"PageSize,omitempty"` |
||||
|
DocBucketList []DocProcessBucket `xml:"DocBucketList,omitempty"` |
||||
|
} |
||||
|
type DocProcessBucket struct { |
||||
|
BucketId string `xml:"BucketId,omitempty"` |
||||
|
Name string `xml:"Name,omitempty"` |
||||
|
Region string `xml:"Region,omitempty"` |
||||
|
CreateTime string `xml:"CreateTime,omitempty"` |
||||
|
AliasBucketId string `xml:"AliasBucketId,omitempty"` |
||||
|
} |
||||
|
|
||||
|
func (s *CIService) DescribeDocProcessBuckets(ctx context.Context, opt *DescribeDocProcessBucketsOptions) (*DescribeDocProcessBucketsResult, *Response, error) { |
||||
|
var res DescribeDocProcessBucketsResult |
||||
|
sendOpt := sendOptions{ |
||||
|
baseURL: s.client.BaseURL.CIURL, |
||||
|
uri: "/docbucket", |
||||
|
optQuery: opt, |
||||
|
method: http.MethodGet, |
||||
|
result: &res, |
||||
|
} |
||||
|
resp, err := s.client.send(ctx, &sendOpt) |
||||
|
return &res, resp, err |
||||
|
} |
||||
|
|
||||
|
type DocPreviewOptions struct { |
||||
|
SrcType string `url:"srcType,omitempty"` |
||||
|
Page int `url:"page,omitempty"` |
||||
|
ImageParams string `url:"ImageParams,omitempty"` |
||||
|
Sheet int `url:"sheet,omitempty"` |
||||
|
DstType string `url:"dstType,omitempty"` |
||||
|
Password string `url:"password,omitempty"` |
||||
|
Comment int `url:"comment,omitempty"` |
||||
|
ExcelPaperDirection int `url:"excelPaperDirection,omitempty"` |
||||
|
Quality int `url:"quality,omitempty"` |
||||
|
Zoom int `url:"zoom,omitempty"` |
||||
|
} |
||||
|
|
||||
|
func (s *CIService) DocPreview(ctx context.Context, name string, opt *DocPreviewOptions) (*Response, error) { |
||||
|
sendOpt := sendOptions{ |
||||
|
baseURL: s.client.BaseURL.BucketURL, |
||||
|
uri: "/" + encodeURIComponent(name) + "?ci-process=doc-preview", |
||||
|
optQuery: opt, |
||||
|
method: http.MethodGet, |
||||
|
} |
||||
|
resp, err := s.client.send(ctx, &sendOpt) |
||||
|
return resp, err |
||||
|
} |
||||
@ -0,0 +1,132 @@ |
|||||
|
package main |
||||
|
|
||||
|
import ( |
||||
|
"context" |
||||
|
"fmt" |
||||
|
"io" |
||||
|
"net/http" |
||||
|
"net/url" |
||||
|
"os" |
||||
|
|
||||
|
"github.com/tencentyun/cos-go-sdk-v5" |
||||
|
"github.com/tencentyun/cos-go-sdk-v5/debug" |
||||
|
) |
||||
|
|
||||
|
func log_status(err error) { |
||||
|
if err == nil { |
||||
|
return |
||||
|
} |
||||
|
if cos.IsNotFoundError(err) { |
||||
|
// WARN
|
||||
|
fmt.Println("WARN: Resource is not existed") |
||||
|
} else if e, ok := cos.IsCOSError(err); ok { |
||||
|
fmt.Printf("ERROR: Code: %v\n", e.Code) |
||||
|
fmt.Printf("ERROR: Message: %v\n", e.Message) |
||||
|
fmt.Printf("ERROR: Resource: %v\n", e.Resource) |
||||
|
fmt.Printf("ERROR: RequestId: %v\n", e.RequestID) |
||||
|
// ERROR
|
||||
|
} else { |
||||
|
fmt.Printf("ERROR: %v\n", err) |
||||
|
// ERROR
|
||||
|
} |
||||
|
} |
||||
|
|
||||
|
func main() { |
||||
|
u, _ := url.Parse("https://test-1259654469.cos.ap-guangzhou.myqcloud.com") |
||||
|
cu, _ := url.Parse("https://test-1259654469.ci.ap-guangzhou.myqcloud.com") |
||||
|
b := &cos.BaseURL{BucketURL: u, CIURL: cu} |
||||
|
c := cos.NewClient(b, &http.Client{ |
||||
|
Transport: &cos.AuthorizationTransport{ |
||||
|
SecretID: os.Getenv("COS_SECRETID"), |
||||
|
SecretKey: os.Getenv("COS_SECRETKEY"), |
||||
|
Transport: &debug.DebugRequestTransport{ |
||||
|
RequestHeader: true, |
||||
|
// Notice when put a large file and set need the request body, might happend out of memory error.
|
||||
|
RequestBody: true, |
||||
|
ResponseHeader: true, |
||||
|
ResponseBody: true, |
||||
|
}, |
||||
|
}, |
||||
|
}) |
||||
|
|
||||
|
// 1、UpdateDocProcessQueue
|
||||
|
updateQueueOpt := &cos.UpdateDocProcessQueueOptions{ |
||||
|
Name: "queue-doc-process-1", |
||||
|
QueueID: "p111a8dd208104ce3b11c78398f658ca8", |
||||
|
State: "Active", |
||||
|
NotifyConfig: &cos.DocProcessQueueNotifyConfig{ |
||||
|
State: "Off", |
||||
|
}, |
||||
|
} |
||||
|
updateQueueRes, _, err := c.CI.UpdateDocProcessQueue(context.Background(), updateQueueOpt) |
||||
|
log_status(err) |
||||
|
fmt.Printf("%+v\n", updateQueueRes) |
||||
|
|
||||
|
// 2、DescribeDocProcessQueues
|
||||
|
DescribeQueueOpt := &cos.DescribeDocProcessQueuesOptions{ |
||||
|
QueueIds: "p111a8dd208104ce3b11c78398f658ca8,p4318f85d2aa14c43b1dba6f9b78be9b3,aacb2bb066e9c4478834d4196e76c49d3", |
||||
|
PageNumber: 1, |
||||
|
PageSize: 2, |
||||
|
} |
||||
|
DescribeQueueRes, _, err := c.CI.DescribeDocProcessQueues(context.Background(), DescribeQueueOpt) |
||||
|
log_status(err) |
||||
|
fmt.Printf("%+v\n", DescribeQueueRes) |
||||
|
|
||||
|
// 3、DescribeDocProcessBuckets
|
||||
|
BucketsOpt := &cos.DescribeDocProcessBucketsOptions{ |
||||
|
Regions: "All", |
||||
|
} |
||||
|
BucketsRes, _, err := c.CI.DescribeDocProcessBuckets(context.Background(), BucketsOpt) |
||||
|
log_status(err) |
||||
|
fmt.Printf("%+v\n", BucketsRes) |
||||
|
|
||||
|
// 4、CreateDocProcessJobs
|
||||
|
createJobOpt := &cos.CreateDocProcessJobsOptions{ |
||||
|
Tag: "DocProcess", |
||||
|
Input: &cos.DocProcessJobInput{ |
||||
|
Object: "form.pdf", |
||||
|
}, |
||||
|
Operation: &cos.DocProcessJobOperation{ |
||||
|
Output: &cos.DocProcessJobOutput{ |
||||
|
Region: "ap-guangzhou", |
||||
|
Object: "test-doc${Number}", |
||||
|
Bucket: "test-1259654469", |
||||
|
}, |
||||
|
DocProcess: &cos.DocProcessJobDocProcess{ |
||||
|
TgtType: "png", |
||||
|
StartPage: 1, |
||||
|
EndPage: -1, |
||||
|
ImageParams: "watermark/1/image/aHR0cDovL3Rlc3QwMDUtMTI1MTcwNDcwOC5jb3MuYXAtY2hvbmdxaW5nLm15cWNsb3VkLmNvbS8xLmpwZw==/gravity/southeast", |
||||
|
}, |
||||
|
}, |
||||
|
QueueId: "p111a8dd208104ce3b11c78398f658ca8", |
||||
|
} |
||||
|
createJobRes, _, err := c.CI.CreateDocProcessJobs(context.Background(), createJobOpt) |
||||
|
log_status(err) |
||||
|
fmt.Printf("%+v\n", createJobRes.JobsDetail) |
||||
|
|
||||
|
// 5、DescribeDocProcessJob
|
||||
|
DescribeJobRes, _, err := c.CI.DescribeDocProcessJob(context.Background(), createJobRes.JobsDetail.JobId) |
||||
|
log_status(err) |
||||
|
fmt.Printf("%+v\n", DescribeJobRes.JobsDetail) |
||||
|
|
||||
|
// 6、DescribeDocProcessJobs
|
||||
|
DescribeJobsOpt := &cos.DescribeDocProcessJobsOptions{ |
||||
|
QueueId: "p111a8dd208104ce3b11c78398f658ca8", |
||||
|
Tag: "DocProcess", |
||||
|
} |
||||
|
DescribeJobsRes, _, err := c.CI.DescribeDocProcessJobs(context.Background(), DescribeJobsOpt) |
||||
|
log_status(err) |
||||
|
fmt.Printf("%+v\n", DescribeJobsRes) |
||||
|
|
||||
|
// 7、doc-preview
|
||||
|
opt := &cos.DocPreviewOptions{ |
||||
|
Page: 1, |
||||
|
} |
||||
|
resp, err := c.CI.DocPreview(context.Background(), "form.pdf", opt) |
||||
|
log_status(err) |
||||
|
fd, _ := os.OpenFile("form.pdf", os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0660) |
||||
|
io.Copy(fd, resp.Body) |
||||
|
fd.Close() |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,59 @@ |
|||||
|
package main |
||||
|
|
||||
|
import ( |
||||
|
"context" |
||||
|
"fmt" |
||||
|
"net/url" |
||||
|
"os" |
||||
|
"time" |
||||
|
|
||||
|
"net/http" |
||||
|
|
||||
|
"github.com/tencentyun/cos-go-sdk-v5" |
||||
|
"github.com/tencentyun/cos-go-sdk-v5/debug" |
||||
|
) |
||||
|
|
||||
|
func log_status(err error) { |
||||
|
if err == nil { |
||||
|
return |
||||
|
} |
||||
|
if cos.IsNotFoundError(err) { |
||||
|
// WARN
|
||||
|
fmt.Println("WARN: Resource is not existed") |
||||
|
} else if e, ok := cos.IsCOSError(err); ok { |
||||
|
fmt.Printf("ERROR: Code: %v\n", e.Code) |
||||
|
fmt.Printf("ERROR: Message: %v\n", e.Message) |
||||
|
fmt.Printf("ERROR: Resource: %v\n", e.Resource) |
||||
|
fmt.Printf("ERROR: RequestId: %v\n", e.RequestID) |
||||
|
// ERROR
|
||||
|
} else { |
||||
|
fmt.Printf("ERROR: %v\n", err) |
||||
|
// ERROR
|
||||
|
} |
||||
|
} |
||||
|
|
||||
|
func main() { |
||||
|
u, _ := url.Parse("https://test-1259654469.cos.ap-guangzhou.myqcloud.com") |
||||
|
b := &cos.BaseURL{BucketURL: u} |
||||
|
c := cos.NewClient(b, &http.Client{ |
||||
|
Transport: &cos.AuthorizationTransport{ |
||||
|
SecretID: os.Getenv("COS_SECRETID"), |
||||
|
SecretKey: os.Getenv("COS_SECRETKEY"), |
||||
|
Transport: &debug.DebugRequestTransport{ |
||||
|
RequestHeader: true, |
||||
|
// Notice when put a large file and set need the request body, might happend out of memory error.
|
||||
|
RequestBody: false, |
||||
|
ResponseHeader: true, |
||||
|
ResponseBody: false, |
||||
|
}, |
||||
|
}, |
||||
|
Timeout: 5 * time.Second, // HTTP超时时间
|
||||
|
}) |
||||
|
|
||||
|
// Case1 上传对象
|
||||
|
name := "test/example" |
||||
|
// Case3 通过本地文件上传对象
|
||||
|
ctx, _ := context.WithTimeout(context.Background(), 2*time.Second) // context超时时间
|
||||
|
_, err := c.Object.PutFromFile(ctx, name, "./test", nil) // 请求的超时时间为 min{context超时时间， HTTP超时时间}
|
||||
|
log_status(err) |
||||
|
} |
||||
@ -0,0 +1,135 @@ |
|||||
|
package cos |
||||
|
|
||||
|
import ( |
||||
|
"fmt" |
||||
|
"io" |
||||
|
) |
||||
|
|
||||
|
type ProgressEventType int |
||||
|
|
||||
|
const ( |
||||
|
// 数据开始传输
|
||||
|
ProgressStartedEvent ProgressEventType = iota |
||||
|
// 数据传输中
|
||||
|
ProgressDataEvent |
||||
|
// 数据传输完成, 但不能表示对应API调用完成
|
||||
|
ProgressCompletedEvent |
||||
|
// 只有在数据传输时发生错误才会返回
|
||||
|
ProgressFailedEvent |
||||
|
) |
||||
|
|
||||
|
type ProgressEvent struct { |
||||
|
EventType ProgressEventType |
||||
|
RWBytes int64 |
||||
|
ConsumedBytes int64 |
||||
|
TotalBytes int64 |
||||
|
Err error |
||||
|
} |
||||
|
|
||||
|
func newProgressEvent(eventType ProgressEventType, rwBytes, consumed, total int64, err ...error) *ProgressEvent { |
||||
|
event := &ProgressEvent{ |
||||
|
EventType: eventType, |
||||
|
RWBytes: rwBytes, |
||||
|
ConsumedBytes: consumed, |
||||
|
TotalBytes: total, |
||||
|
} |
||||
|
if len(err) > 0 { |
||||
|
event.Err = err[0] |
||||
|
} |
||||
|
return event |
||||
|
} |
||||
|
|
||||
|
// 用户自定义Listener需要实现该方法
|
||||
|
type ProgressListener interface { |
||||
|
ProgressChangedCallback(event *ProgressEvent) |
||||
|
} |
||||
|
|
||||
|
func progressCallback(listener ProgressListener, event *ProgressEvent) { |
||||
|
if listener != nil && event != nil { |
||||
|
listener.ProgressChangedCallback(event) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
type teeReader struct { |
||||
|
reader io.Reader |
||||
|
writer io.Writer |
||||
|
consumedBytes int64 |
||||
|
totalBytes int64 |
||||
|
listener ProgressListener |
||||
|
} |
||||
|
|
||||
|
func (r *teeReader) Read(p []byte) (int, error) { |
||||
|
if r.consumedBytes == 0 { |
||||
|
event := newProgressEvent(ProgressStartedEvent, 0, r.consumedBytes, r.totalBytes) |
||||
|
progressCallback(r.listener, event) |
||||
|
} |
||||
|
|
||||
|
n, err := r.reader.Read(p) |
||||
|
if err != nil && err != io.EOF { |
||||
|
event := newProgressEvent(ProgressFailedEvent, 0, r.consumedBytes, r.totalBytes, err) |
||||
|
progressCallback(r.listener, event) |
||||
|
} |
||||
|
if n > 0 { |
||||
|
r.consumedBytes += int64(n) |
||||
|
if r.writer != nil { |
||||
|
if n, err := r.writer.Write(p[:n]); err != nil { |
||||
|
return n, err |
||||
|
} |
||||
|
} |
||||
|
if r.listener != nil { |
||||
|
event := newProgressEvent(ProgressDataEvent, int64(n), r.consumedBytes, r.totalBytes) |
||||
|
progressCallback(r.listener, event) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
if err == io.EOF { |
||||
|
event := newProgressEvent(ProgressCompletedEvent, int64(n), r.consumedBytes, r.totalBytes) |
||||
|
progressCallback(r.listener, event) |
||||
|
} |
||||
|
|
||||
|
return n, err |
||||
|
} |
||||
|
|
||||
|
func (r *teeReader) Close() error { |
||||
|
if rc, ok := r.reader.(io.ReadCloser); ok { |
||||
|
return rc.Close() |
||||
|
} |
||||
|
return nil |
||||
|
} |
||||
|
|
||||
|
func TeeReader(reader io.Reader, writer io.Writer, total int64, listener ProgressListener) *teeReader { |
||||
|
return &teeReader{ |
||||
|
reader: reader, |
||||
|
writer: writer, |
||||
|
consumedBytes: 0, |
||||
|
totalBytes: total, |
||||
|
listener: listener, |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
type FixedLengthReader interface { |
||||
|
io.Reader |
||||
|
Size() int64 |
||||
|
} |
||||
|
|
||||
|
type DefaultProgressListener struct { |
||||
|
} |
||||
|
|
||||
|
func (l *DefaultProgressListener) ProgressChangedCallback(event *ProgressEvent) { |
||||
|
switch event.EventType { |
||||
|
case ProgressStartedEvent: |
||||
|
fmt.Printf("Transfer Start [ConsumedBytes/TotalBytes: %d/%d]\n", |
||||
|
event.ConsumedBytes, event.TotalBytes) |
||||
|
case ProgressDataEvent: |
||||
|
fmt.Printf("\rTransfer Data [ConsumedBytes/TotalBytes: %d/%d, %d%%]", |
||||
|
event.ConsumedBytes, event.TotalBytes, event.ConsumedBytes*100/event.TotalBytes) |
||||
|
case ProgressCompletedEvent: |
||||
|
fmt.Printf("\nTransfer Complete [ConsumedBytes/TotalBytes: %d/%d]\n", |
||||
|
event.ConsumedBytes, event.TotalBytes) |
||||
|
case ProgressFailedEvent: |
||||
|
fmt.Printf("\nTransfer Failed [ConsumedBytes/TotalBytes: %d/%d] [Err: %v]\n", |
||||
|
event.ConsumedBytes, event.TotalBytes, event.Err) |
||||
|
default: |
||||
|
fmt.Printf("Progress Changed Error: unknown progress event type\n") |
||||
|
} |
||||
|
} |
||||
Write
Preview
Loading…
Cancel
Save
Reference in new issue