github.com/kikitux/packer@v0.10.1-0.20160322154024-6237df566f9f/builder/azure/arm/step_power_off_compute.go (about) 1 // Copyright (c) Microsoft Corporation. All rights reserved. 2 // Licensed under the MIT License. See the LICENSE file in builder/azure for license information. 3 4 package arm 5 6 import ( 7 "fmt" 8 9 "github.com/mitchellh/multistep" 10 "github.com/mitchellh/packer/builder/azure/common/constants" 11 "github.com/mitchellh/packer/packer" 12 ) 13 14 type StepPowerOffCompute struct { 15 client *AzureClient 16 powerOff func(resourceGroupName string, computeName string) error 17 say func(message string) 18 error func(e error) 19 } 20 21 func NewStepPowerOffCompute(client *AzureClient, ui packer.Ui) *StepPowerOffCompute { 22 var step = &StepPowerOffCompute{ 23 client: client, 24 say: func(message string) { ui.Say(message) }, 25 error: func(e error) { ui.Error(e.Error()) }, 26 } 27 28 step.powerOff = step.powerOffCompute 29 return step 30 } 31 32 func (s *StepPowerOffCompute) powerOffCompute(resourceGroupName string, computeName string) error { 33 res, err := s.client.PowerOff(resourceGroupName, computeName) 34 if err != nil { 35 return err 36 } 37 38 s.client.VirtualMachinesClient.PollAsNeeded(res.Response) 39 return nil 40 } 41 42 func (s *StepPowerOffCompute) Run(state multistep.StateBag) multistep.StepAction { 43 s.say("Powering off machine ...") 44 45 var resourceGroupName = state.Get(constants.ArmResourceGroupName).(string) 46 var computeName = state.Get(constants.ArmComputeName).(string) 47 48 s.say(fmt.Sprintf(" -> ResourceGroupName : '%s'", resourceGroupName)) 49 s.say(fmt.Sprintf(" -> ComputeName : '%s'", computeName)) 50 51 err := s.powerOff(resourceGroupName, computeName) 52 if err != nil { 53 state.Put(constants.Error, err) 54 s.error(err) 55 56 return multistep.ActionHalt 57 } 58 59 return multistep.ActionContinue 60 } 61 62 func (*StepPowerOffCompute) Cleanup(multistep.StateBag) { 63 }