add Put ObjectRetention

Author: jojoliang

bucket_object_lock.go

@@ -48,6 +48,7 @@ type ObjectGetRetentionOptions struct {
 type ObjectGetRetentionResult struct {
 	XMLName         xml.Name `xml:"Retention"`
 	RetainUntilDate string   `xml:"RetainUntilDate,omitempty"`
+	Mode            string   `xml:"Mode,omitempty"`
 }
 
 func (s *ObjectService) GetRetention(ctx context.Context, key string, opt *ObjectGetRetentionOptions) (*ObjectGetRetentionResult, *Response, error) {
@@ -62,3 +63,20 @@ func (s *ObjectService) GetRetention(ctx context.Context, key string, opt *Objec
 	resp, err := s.client.doRetry(ctx, &sendOpt)
 	return &res, resp, err
 }
+
+type ObjectPutRetentionOptions struct {
+	XMLName         xml.Name `xml:"Retention"`
+	RetainUntilDate string   `xml:"RetainUntilDate,omitempty"`
+	Mode            string   `xml:"Mode,omitempty"`
+}
+
+func (s *ObjectService) PutRetention(ctx context.Context, key string, opt *ObjectPutRetentionOptions) (*Response, error) {
+	sendOpt := sendOptions{
+		baseURL: s.client.BaseURL.BucketURL,
+		uri:     "/" + encodeURIComponent(key) + "?retention",
+		method:  http.MethodPut,
+		body:    opt,
+	}
+	resp, err := s.client.doRetry(ctx, &sendOpt)
+	return resp, err
+}

bucket_object_lock_test.go

@@ -110,3 +110,34 @@ func TestBucketService_GetRetention(t *testing.T) {
 		t.Errorf("Object.GetRetention returned %+v, want %+v", res, want)
 	}
 }
+
+func TestBucketService_PutRetention(t *testing.T) {
+	setup()
+	defer teardown()
+
+	opt := &ObjectPutRetentionOptions{
+		RetainUntilDate: "2022-12-10T08:34:48.000Z",
+		Mode:            "COMPLIANCE",
+	}
+
+	key := "example"
+	mux.HandleFunc("/example", func(w http.ResponseWriter, r *http.Request) {
+		testMethod(t, r, "PUT")
+		vs := values{
+			"retention": "",
+		}
+		testFormValues(t, r, vs)
+		body := new(ObjectPutRetentionOptions)
+		xml.NewDecoder(r.Body).Decode(body)
+		want := opt
+		want.XMLName = xml.Name{Local: "Retention"}
+		if !reflect.DeepEqual(body, want) {
+			t.Errorf("Object.PutRetention request\n body: %+v\nwant %+v\n", body, want)
+		}
+	})
+
+	_, err := client.Object.PutRetention(context.Background(), key, opt)
+	if err != nil {
+		t.Fatalf("Object.GetRetention returned error: %v", err)
+	}
+}

cos.go

@@ -24,7 +24,7 @@ import (
 
 const (
 	// Version current go sdk version
-	Version               = "0.7.42"
+	Version               = "0.7.43"
 	UserAgent             = "cos-go-sdk-v5/" + Version
 	contentTypeXML        = "application/xml"
 	defaultServiceBaseURL = "http://service.cos.myqcloud.com"

example/bucket/object_lock.go

@@ -60,6 +60,13 @@ func main() {
 	log_status(err)
 	fmt.Printf("%+v\n", res)
 
+	ropt := &cos.ObjectPutRetentionOptions{
+		RetainUntilDate: "2022-12-10T08:34:48.000Z",
+		Mode: "COMPLIANCE",
+	}
+	_, err = c.Object.PutRetention(context.Background(), "test", ropt)
+	log_status(err)
+
 	r, _, err := c.Object.GetRetention(context.Background(), "test", nil)
 	log_status(err)
 	fmt.Printf("%+v\n", r)