step_suspend.go

//  Copyright 2017 Google Inc. All Rights Reserved.
//
//  Licensed 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 daisy

import (
	"context"
	"fmt"
)

// Suspend is a Daisy suspend workflow step.
type Suspend struct {
	Project  string
	Zone     string
	Instance string
}

// populate preprocesses fields: Instance, Project, Zone
// - sets defaults
// - extends short partial URLs to include "projects/<project>"
func (sp *Suspend) populate(ctx context.Context, s *Step) DError {
	if sp.Project == "" {
		sp.Project = s.w.Project
	}
	if sp.Zone == "" {
		sp.Zone = s.w.Zone
	}
	return nil
}

func (sp *Suspend) validate(ctx context.Context, s *Step) DError {
	var errs DError
	if sp.Project == "" {
		errs = addErrs(errs, fmt.Errorf("must specify project"))
	}
	if sp.Zone == "" {
		errs = addErrs(errs, fmt.Errorf("must specify zone"))
	}
	if sp.Instance == "" {
		errs = addErrs(errs, fmt.Errorf("must specify instance"))
	}
	return errs
}

func (sp *Suspend) run(ctx context.Context, s *Step) DError {
	prj := sp.Project
	zone := sp.Zone
	inst := sp.Instance
	i, ok := s.w.instances.get(inst)
	if ok {
		m := NamedSubexp(instanceURLRgx, i.link)
		prj = m["project"]
		zone = m["zone"]
		inst = m["instance"]
	}
	return addErrs(nil, s.w.ComputeClient.Suspend(prj, zone, inst))
}