pkg/api/deploymentapi/depresourceapi/apm.go

// Licensed to Elasticsearch B.V. under one or more contributor
// license agreements. See the NOTICE file distributed with
// this work for additional information regarding copyright
// ownership. Elasticsearch B.V. licenses this file to you under
// the Apache License, Version 2.0 (the "License"); you may
// not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied.  See the License for the
// specific language governing permissions and limitations
// under the License.

package depresourceapi

import (
	"fmt"

	"github.com/elastic/cloud-sdk-go/pkg/models"
	"github.com/elastic/cloud-sdk-go/pkg/util/ec"
)

const (
	// DefaultApmRefID is used when the RefID is not specified.
	DefaultApmRefID = "main-apm"
)

// NewApm creates a *models.ApmPayload from the parameters.
// It relies on a simplified single dimension memory size and zone count to
// construct the Apm's topology.
func NewApm(params NewStateless) (*models.ApmPayload, error) {
	params.fillDefaults(DefaultApmRefID)
	if err := params.Validate(); err != nil {
		return nil, err
	}

	// When either not set, we obtain the values from the running deployment.
	// Overriding either or both is done at the end of the if.
	if err := getTemplateAndRefID(&params); err != nil {
		return nil, err
	}

	if len(params.DeploymentTemplateInfoV2.DeploymentTemplate.Resources.Apm) == 0 {
		return nil, fmt.Errorf("deployment: the %s template is not configured for APM. Please use another template if you wish to start APM instances",
			params.TemplateID)
	}

	var clusterTopology = params.DeploymentTemplateInfoV2.DeploymentTemplate.Resources.Apm[0].Plan.ClusterTopology
	var topology = models.ApmTopologyElement{Size: new(models.TopologySize)}
	if len(clusterTopology) > 0 {
		topology = *clusterTopology[0]
	}
	var payload = newApmPayload(params, topology)

	return &payload, nil
}

func newApmPayload(params NewStateless, topology models.ApmTopologyElement) models.ApmPayload {
	if params.Size > 0 {
		topology.Size.Value = ec.Int32(params.Size)
	}
	if params.ZoneCount > 0 {
		topology.ZoneCount = params.ZoneCount
	}

	return models.ApmPayload{
		ElasticsearchClusterRefID: ec.String(params.ElasticsearchRefID),
		DisplayName:               params.Name,
		Region:                    ec.String(params.Region),
		RefID:                     ec.String(params.RefID),
		Plan: &models.ApmPlan{
			Apm:             &models.ApmConfiguration{Version: params.Version},
			ClusterTopology: []*models.ApmTopologyElement{&topology},
		},
	}
}