decort-golang-sdk/pkg/cloudapi/compute/affinity_rule_add.go

package compute

import (
	"context"
	"encoding/json"
	"net/http"
	"strconv"

	"repository.basistech.ru/BASIS/decort-golang-sdk/internal/validators"
)

// AffinityRuleAddRequest struct to add affinity rule
type AffinityRuleAddRequest struct {
	// ID of the compute instance
	// Required: true
	ComputeID uint64 `url:"computeId" json:"computeId" validate:"required"`

	// Compute or node, for whom rule applies
	// Required: true
	Topology string `url:"topology" json:"topology" validate:"computeTopology"`

	// The degree of 'strictness' of this rule
	// Should be one of:
	//	- RECOMMENDED
	//	- REQUIRED
	// Required: true
	Policy string `url:"policy" json:"policy" validate:"computePolicy"`

	// The comparison mode is 'value', recorded by the specified 'key'
	// Should be one of:
	//	- EQ
	//	- EN
	//	- ANY
	// Required: true
	Mode string `url:"mode" json:"mode" validate:"computeMode"`

	// Key that are taken into account when analyzing this rule will be identified
	// Required: true
	Key string `url:"key" json:"key" validate:"required"`

	// Value that must match the key to be taken into account when analyzing this rule
	// Required: true
	Value string `url:"value" json:"value" validate:"required"`

	// Migrate compute according to the rule
	// Default: false
	// Required: false
	Migrate bool `url:"migrate" json:"migrate"`
}

type wrapperAffinityRuleAddRequest struct {
	AffinityRuleAddRequest

	AsyncMode bool `url:"async_mode"`
}

// AffinityRuleAdd add affinity rule
func (c Compute) AffinityRuleAdd(ctx context.Context, req AffinityRuleAddRequest) (bool, error) {
	err := validators.ValidateRequest(req)
	if err != nil {
		return false, validators.ValidationErrors(validators.GetErrors(err))
	}

	url := "/cloudapi/compute/affinityRuleAdd"

	aReq := wrapperAffinityRuleAddRequest{
		AffinityRuleAddRequest: req,
		AsyncMode:              false,
	}

	res, err := c.client.DecortApiCall(ctx, http.MethodPost, url, aReq)
	if err != nil {
		return false, err
	}

	result, err := strconv.ParseBool(string(res))
	if err != nil {
		return false, err
	}

	return result, nil
}

// AsyncAffinityRuleAdd add affinity rule in async mode
func (c Compute) AsyncAffinityRuleAdd(ctx context.Context, req AffinityRuleAddRequest) (string, error) {
	err := validators.ValidateRequest(req)
	if err != nil {
		return "", validators.ValidationErrors(validators.GetErrors(err))
	}

	url := "/cloudapi/compute/affinityRuleAdd"

	aReq := wrapperAffinityRuleAddRequest{
		AffinityRuleAddRequest: req,
		AsyncMode:              true,
	}

	res, err := c.client.DecortApiCall(ctx, http.MethodPost, url, aReq)
	if err != nil {
		return "", err
	}

	var resp string

	err = json.Unmarshal(res, &resp)
	if err != nil {
		return "", err
	}

	return resp, nil
}