v1.16.1

v1.16.1

pkg/cloudapi/disks/snapshot_rollback.go

@@ -2,6 +2,7 @@ package disks
 
 import (
 	"context"
+	"encoding/json"
 	"net/http"
 	"strconv"
 
@@ -23,6 +24,12 @@ type SnapshotRollbackRequest struct {
 	TimeStamp uint64 `url:"timestamp,omitempty" json:"timestamp,omitempty"`
 }
 
+type wrapperSnapshotRollbackRequest struct {
+	SnapshotRollbackRequest
+
+	AsyncMode bool `url:"async_mode" json:"async_mode"`
+}
+
 // SnapshotRollback rollbacks an individual disk snapshot
 func (d Disks) SnapshotRollback(ctx context.Context, req SnapshotRollbackRequest) (bool, error) {
 	err := validators.ValidateRequest(req)
@@ -32,7 +39,12 @@ func (d Disks) SnapshotRollback(ctx context.Context, req SnapshotRollbackRequest
 
 	url := "/cloudapi/disks/snapshotRollback"
 
-	res, err := d.client.DecortApiCall(ctx, http.MethodPost, url, req)
+	wrappedReq := wrapperSnapshotRollbackRequest{
+		SnapshotRollbackRequest: req,
+		AsyncMode:               false,
+	}
+
+	res, err := d.client.DecortApiCall(ctx, http.MethodPost, url, wrappedReq)
 	if err != nil {
 		return false, err
 	}
@@ -43,5 +55,31 @@ func (d Disks) SnapshotRollback(ctx context.Context, req SnapshotRollbackRequest
 	}
 
 	return result, nil
-
+}
+
+// SnapshotRollbackAsync rollbacks an individual disk snapshot in async mode
+func (d Disks) SnapshotRollbackAsync(ctx context.Context, req SnapshotRollbackRequest) (string, error) {
+	err := validators.ValidateRequest(req)
+	if err != nil {
+		return "", validators.ValidationErrors(validators.GetErrors(err))
+	}
+
+	url := "/cloudapi/disks/snapshotRollback"
+
+	wrappedReq := wrapperSnapshotRollbackRequest{
+		SnapshotRollbackRequest: req,
+		AsyncMode:               true,
+	}
+
+	res, err := d.client.DecortApiCall(ctx, http.MethodPost, url, wrappedReq)
+	if err != nil {
+		return "", err
+	}
+
+	var result string
+	if err := json.Unmarshal(res, &result); err != nil {
+		return "", err
+	}
+
+	return result, nil
 }