120 lines
3.1 KiB
Go
120 lines
3.1 KiB
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 {
|
|
// 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" json:"async_mode"`
|
|
}
|
|
|
|
// AffinityRuleAdd add 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 := "/cloudapi/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 := "/cloudapi/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
|
|
}
|