v16.1.0

v16.1.0

pkg/cloudbroker/disks/limit_io.go

@@ -2,6 +2,7 @@ package disks
 
 import (
 	"context"
+	"encoding/json"
 	"net/http"
 	"strconv"
 
@@ -71,6 +72,12 @@ type LimitIORequest struct {
 	SizeIOPSSec uint64 `url:"size_iops_sec,omitempty" json:"size_iops_sec,omitempty"`
 }
 
+type wrapperLimitIORequest struct {
+	LimitIORequest
+
+	AsyncMode bool `url:"async_mode" json:"async_mode"`
+}
+
 // LimitIO limits IO for a certain disk
 // total and read/write options are not allowed to be combined
 // see http://libvirt.org/formatdomain.html#elementsDisks iotune section for more details
@@ -82,7 +89,12 @@ func (d Disks) LimitIO(ctx context.Context, req LimitIORequest) (bool, error) {
 
 	url := "/cloudbroker/disks/limitIO"
 
-	res, err := d.client.DecortApiCall(ctx, http.MethodPost, url, req)
+	wrappedReq := wrapperLimitIORequest{
+		LimitIORequest: req,
+		AsyncMode:      false,
+	}
+
+	res, err := d.client.DecortApiCall(ctx, http.MethodPost, url, wrappedReq)
 	if err != nil {
 		return false, err
 	}
@@ -94,3 +106,32 @@ func (d Disks) LimitIO(ctx context.Context, req LimitIORequest) (bool, error) {
 
 	return result, nil
 }
+
+// LimitIOAsync limits IO for a certain disk in async mode
+// total and read/write options are not allowed to be combined
+// see http://libvirt.org/formatdomain.html#elementsDisks iotune section for more details
+func (d Disks) LimitIOAsync(ctx context.Context, req LimitIORequest) (string, error) {
+	err := validators.ValidateRequest(req)
+	if err != nil {
+		return "", validators.ValidationErrors(validators.GetErrors(err))
+	}
+
+	url := "/cloudbroker/disks/limitIO"
+
+	wrappedReq := wrapperLimitIORequest{
+		LimitIORequest: 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
+}