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

package compute

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

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

// AffinityRuleAddRequest struct to add affinity rule
type AffinityRuleAddRequest struct {
	// IDs of the compute instances
	// Required: true
	ComputeIDs []uint64 `url:"computeIds" json:"computeIds" validate:"min=1"`

	// Should be one of:
	//	- node
	//	- compute
	// 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" json:"async_mode"`
}

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

	url := "/cloudbroker/compute/affinityRuleAdd"

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

	res, err := c.client.DecortApiCallCtype(ctx, http.MethodPost, url, constants.MIMEJSON, aReq)
	if err != nil {
		return AffinityRuleAddResult{}, err
	}

	if !req.Migrate {
		result, err := strconv.ParseBool(string(res))
		return AffinityRuleAddResult{Ok: result}, err
	}

	var report MigrationReport

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

	return AffinityRuleAddResult{Ok: report.Applied, MigrationReport: &report}, 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 := "/cloudbroker/compute/affinityRuleAdd"

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

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

	var resp string

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

	return resp, nil
}