91 lines
2.8 KiB
Go
91 lines
2.8 KiB
Go
package ecs
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/aliyun/alibaba-cloud-sdk-go/sdk/errors"
|
|
"github.com/aliyun/alibaba-cloud-sdk-go/services/ecs"
|
|
"github.com/hashicorp/packer/packer"
|
|
"github.com/hashicorp/packer/packer-plugin-sdk/multistep"
|
|
)
|
|
|
|
type stepCreateAlicloudSnapshot struct {
|
|
snapshot *ecs.Snapshot
|
|
WaitSnapshotReadyTimeout int
|
|
}
|
|
|
|
func (s *stepCreateAlicloudSnapshot) Run(ctx context.Context, state multistep.StateBag) multistep.StepAction {
|
|
config := state.Get("config").(*Config)
|
|
client := state.Get("client").(*ClientWrapper)
|
|
ui := state.Get("ui").(packer.Ui)
|
|
instance := state.Get("instance").(*ecs.Instance)
|
|
|
|
describeDisksRequest := ecs.CreateDescribeDisksRequest()
|
|
describeDisksRequest.RegionId = config.AlicloudRegion
|
|
describeDisksRequest.InstanceId = instance.InstanceId
|
|
describeDisksRequest.DiskType = DiskTypeSystem
|
|
disksResponse, err := client.DescribeDisks(describeDisksRequest)
|
|
if err != nil {
|
|
return halt(state, err, "Error describe disks")
|
|
}
|
|
|
|
disks := disksResponse.Disks.Disk
|
|
if len(disks) == 0 {
|
|
return halt(state, err, "Unable to find system disk of instance")
|
|
}
|
|
|
|
createSnapshotRequest := ecs.CreateCreateSnapshotRequest()
|
|
createSnapshotRequest.DiskId = disks[0].DiskId
|
|
snapshot, err := client.CreateSnapshot(createSnapshotRequest)
|
|
if err != nil {
|
|
return halt(state, err, "Error creating snapshot")
|
|
}
|
|
|
|
// Create the alicloud snapshot
|
|
ui.Say(fmt.Sprintf("Creating snapshot from system disk %s: %s", disks[0].DiskId, snapshot.SnapshotId))
|
|
|
|
snapshotsResponse, err := client.WaitForSnapshotStatus(config.AlicloudRegion, snapshot.SnapshotId, SnapshotStatusAccomplished, time.Duration(s.WaitSnapshotReadyTimeout)*time.Second)
|
|
if err != nil {
|
|
_, ok := err.(errors.Error)
|
|
if ok {
|
|
return halt(state, err, "Error querying created snapshot")
|
|
}
|
|
|
|
return halt(state, err, "Timeout waiting for snapshot to be created")
|
|
}
|
|
|
|
snapshots := snapshotsResponse.(*ecs.DescribeSnapshotsResponse).Snapshots.Snapshot
|
|
if len(snapshots) == 0 {
|
|
return halt(state, err, "Unable to find created snapshot")
|
|
}
|
|
|
|
s.snapshot = &snapshots[0]
|
|
state.Put("alicloudsnapshot", snapshot.SnapshotId)
|
|
return multistep.ActionContinue
|
|
}
|
|
|
|
func (s *stepCreateAlicloudSnapshot) Cleanup(state multistep.StateBag) {
|
|
if s.snapshot == nil {
|
|
return
|
|
}
|
|
_, cancelled := state.GetOk(multistep.StateCancelled)
|
|
_, halted := state.GetOk(multistep.StateHalted)
|
|
if !cancelled && !halted {
|
|
return
|
|
}
|
|
|
|
client := state.Get("client").(*ClientWrapper)
|
|
ui := state.Get("ui").(packer.Ui)
|
|
|
|
ui.Say("Deleting the snapshot because of cancellation or error...")
|
|
|
|
deleteSnapshotRequest := ecs.CreateDeleteSnapshotRequest()
|
|
deleteSnapshotRequest.SnapshotId = s.snapshot.SnapshotId
|
|
if _, err := client.DeleteSnapshot(deleteSnapshotRequest); err != nil {
|
|
ui.Error(fmt.Sprintf("Error deleting snapshot, it may still be around: %s", err))
|
|
return
|
|
}
|
|
}
|