109 lines
3.1 KiB
Go
109 lines
3.1 KiB
Go
package arm
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strconv"
|
|
|
|
"github.com/Azure/azure-sdk-for-go/services/compute/mgmt/2018-04-01/compute"
|
|
"github.com/Azure/go-autorest/autorest/to"
|
|
"github.com/hashicorp/packer/builder/azure/common/constants"
|
|
"github.com/hashicorp/packer/helper/multistep"
|
|
"github.com/hashicorp/packer/packer"
|
|
)
|
|
|
|
type StepSnapshotDataDisks struct {
|
|
client *AzureClient
|
|
create func(ctx context.Context, resourceGroupName string, srcUriVhd string, location string, tags map[string]*string, dstSnapshotName string) error
|
|
say func(message string)
|
|
error func(e error)
|
|
enable func() bool
|
|
}
|
|
|
|
func NewStepSnapshotDataDisks(client *AzureClient, ui packer.Ui, config *Config) *StepSnapshotDataDisks {
|
|
var step = &StepSnapshotDataDisks{
|
|
client: client,
|
|
say: func(message string) { ui.Say(message) },
|
|
error: func(e error) { ui.Error(e.Error()) },
|
|
enable: func() bool { return config.isManagedImage() && config.ManagedImageDataDiskSnapshotPrefix != "" },
|
|
}
|
|
|
|
step.create = step.createDataDiskSnapshot
|
|
return step
|
|
}
|
|
|
|
func (s *StepSnapshotDataDisks) createDataDiskSnapshot(ctx context.Context, resourceGroupName string, srcUriVhd string, location string, tags map[string]*string, dstSnapshotName string) error {
|
|
|
|
srcVhdToSnapshot := compute.Snapshot{
|
|
DiskProperties: &compute.DiskProperties{
|
|
CreationData: &compute.CreationData{
|
|
CreateOption: compute.Copy,
|
|
SourceResourceID: to.StringPtr(srcUriVhd),
|
|
},
|
|
},
|
|
Location: to.StringPtr(location),
|
|
Tags: tags,
|
|
}
|
|
|
|
f, err := s.client.SnapshotsClient.CreateOrUpdate(ctx, resourceGroupName, dstSnapshotName, srcVhdToSnapshot)
|
|
|
|
if err != nil {
|
|
s.say(s.client.LastError.Error())
|
|
return err
|
|
}
|
|
|
|
err = f.WaitForCompletion(ctx, s.client.SnapshotsClient.Client)
|
|
|
|
if err != nil {
|
|
s.say(s.client.LastError.Error())
|
|
return err
|
|
}
|
|
|
|
createdSnapshot, err := f.Result(s.client.SnapshotsClient)
|
|
|
|
if err != nil {
|
|
s.say(s.client.LastError.Error())
|
|
return err
|
|
}
|
|
|
|
s.say(fmt.Sprintf(" -> Snapshot ID : '%s'", *(createdSnapshot.ID)))
|
|
return nil
|
|
}
|
|
|
|
func (s *StepSnapshotDataDisks) Run(ctx context.Context, stateBag multistep.StateBag) multistep.StepAction {
|
|
if !s.enable() {
|
|
return multistep.ActionContinue
|
|
}
|
|
|
|
var resourceGroupName = stateBag.Get(constants.ArmManagedImageResourceGroupName).(string)
|
|
var location = stateBag.Get(constants.ArmLocation).(string)
|
|
var tags = stateBag.Get(constants.ArmTags).(map[string]*string)
|
|
var additionalDisks = stateBag.Get(constants.ArmAdditionalDiskVhds).([]string)
|
|
var dstSnapshotPrefix = stateBag.Get(constants.ArmManagedImageDataDiskSnapshotPrefix).(string)
|
|
|
|
if len(additionalDisks) == 1 {
|
|
s.say(fmt.Sprintf("Snapshotting data disk ..."))
|
|
} else {
|
|
s.say(fmt.Sprintf("Snapshotting data disks ..."))
|
|
}
|
|
|
|
for i, disk := range additionalDisks {
|
|
s.say(fmt.Sprintf(" -> Data Disk : '%s'", disk))
|
|
|
|
dstSnapshotName := dstSnapshotPrefix + strconv.Itoa(i)
|
|
err := s.create(ctx, resourceGroupName, disk, location, tags, dstSnapshotName)
|
|
|
|
if err != nil {
|
|
stateBag.Put(constants.Error, err)
|
|
s.error(err)
|
|
|
|
return multistep.ActionHalt
|
|
}
|
|
}
|
|
|
|
return multistep.ActionContinue
|
|
}
|
|
|
|
func (*StepSnapshotDataDisks) Cleanup(multistep.StateBag) {
|
|
}
|