From 7269ed131c234ad3cbc6c40b60ee430d2c8b5fa1 Mon Sep 17 00:00:00 2001
From: jojoliang <agin719@126.com>
Date: Mon, 20 Jan 2020 10:07:05 +0800
Subject: [PATCH] add restore object for batch

---
 batch.go                            |   9 +++-
 example/batch/create_job_restore.go | 105 ++++++++++++++++++++++++++++++++++++
 2 files changed, 113 insertions(+), 1 deletion(-)
 create mode 100644 example/batch/create_job_restore.go

diff --git a/batch.go b/batch.go
index 390d20c..05089dd 100644
--- a/batch.go
+++ b/batch.go
@@ -69,9 +69,16 @@ type BatchJobOperationCopy struct {
 	TargetResource            string                    `xml:"TargetResource" header:"-" url:"-"`
 }
 
+// BatchInitiateRestoreObject
+type BatchInitiateRestoreObject struct {
+	ExpirationInDays int    `xml:"ExpirationInDays"`
+	JobTier          string `xml:"JobTier"`
+}
+
 // BatchJobOperation
 type BatchJobOperation struct {
-	PutObjectCopy *BatchJobOperationCopy `xml:"COSPutObjectCopy,omitempty" header:"-" url:"-"`
+	PutObjectCopy *BatchJobOperationCopy      `xml:"COSPutObjectCopy,omitempty" header:"-" url:"-"`
+	RestoreObject *BatchInitiateRestoreObject `xml:"COSInitiateRestoreObject,omitempty" header:"-" url:"-"`
 }
 
 // BatchJobManifest
diff --git a/example/batch/create_job_restore.go b/example/batch/create_job_restore.go
new file mode 100644
index 0000000..e51627f
--- /dev/null
+++ b/example/batch/create_job_restore.go
@@ -0,0 +1,105 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"net/http"
+	"net/url"
+	"os"
+	"strings"
+
+	"github.com/google/uuid"
+	"github.com/tencentyun/cos-go-sdk-v5"
+	"github.com/tencentyun/cos-go-sdk-v5/debug"
+)
+
+func main() {
+	test_batch_bucket := "testcd-1259654469"
+	appid := 1259654469
+	uin := "100010805041"
+	region := "ap-chengdu"
+
+	// bucket url：<Bucketname-Appid>.cos.<region>.mycloud.com
+	bucketurl, _ := url.Parse("https://" + test_batch_bucket + ".cos." + region + ".myqcloud.com")
+	// batch url：<uin>.cos-control.<region>.myqcloud.ccom
+	batchurl, _ := url.Parse("https://" + uin + ".cos-control." + region + ".myqcloud.com")
+
+	b := &cos.BaseURL{BucketURL: bucketurl, BatchURL: batchurl}
+	c := cos.NewClient(b, &http.Client{
+		Transport: &cos.AuthorizationTransport{
+			SecretID:  os.Getenv("COS_SECRETID"),
+			SecretKey: os.Getenv("COS_SECRETKEY"),
+			Transport: &debug.DebugRequestTransport{
+				RequestHeader:  true,
+				RequestBody:    true,
+				ResponseHeader: true,
+				ResponseBody:   true,
+			},
+		},
+	})
+
+	// 创建需要归档恢复的文件
+	source_name := "test/restore.txt"
+	sf := strings.NewReader("batch test content")
+	objopt := &cos.ObjectPutOptions{
+		nil,
+		&cos.ObjectPutHeaderOptions{
+			XCosStorageClass: "Archive",
+		},
+	}
+	_, err := c.Object.Put(context.Background(), source_name, sf, objopt)
+	if err != nil {
+		panic(err)
+	}
+
+	// 创建清单文件
+	manifest_name := "test/manifest.csv"
+	f := strings.NewReader(test_batch_bucket + "," + source_name)
+	resp, err := c.Object.Put(context.Background(), manifest_name, f, nil)
+	if err != nil {
+		panic(err)
+	}
+	etag := resp.Header.Get("ETag")
+
+	uuid_str := uuid.New().String()
+	opt := &cos.BatchCreateJobOptions{
+		ClientRequestToken:   uuid_str,
+		ConfirmationRequired: "true",
+		Description:          "test batch",
+		Manifest: &cos.BatchJobManifest{
+			Location: &cos.BatchJobManifestLocation{
+				ETag:      etag,
+				ObjectArn: "qcs::cos:" + region + "::" + test_batch_bucket + "/" + manifest_name,
+			},
+			Spec: &cos.BatchJobManifestSpec{
+				Fields: []string{"Bucket", "Key"},
+				Format: "COSBatchOperations_CSV_V1",
+			},
+		},
+		Operation: &cos.BatchJobOperation{
+			RestoreObject: &cos.BatchInitiateRestoreObject{
+				ExpirationInDays: 3,
+				JobTier:          "Standard",
+			},
+		},
+		Priority: 1,
+		Report: &cos.BatchJobReport{
+			Bucket:      "qcs::cos:" + region + "::" + test_batch_bucket,
+			Enabled:     "true",
+			Format:      "Report_CSV_V1",
+			Prefix:      "job-result",
+			ReportScope: "AllTasks",
+		},
+		RoleArn: "qcs::cam::uin/" + uin + ":roleName/COSBatch_QcsRole",
+	}
+	headers := &cos.BatchRequestHeaders{
+		XCosAppid: appid,
+	}
+
+	res, _, err := c.Batch.CreateJob(context.Background(), opt, headers)
+	if err != nil {
+		panic(err)
+	}
+	fmt.Println(res)
+
+}