139 lines
3.9 KiB
Go
139 lines
3.9 KiB
Go
package uhost
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"github.com/hashicorp/packer/common/retry"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/hashicorp/packer/helper/multistep"
|
|
"github.com/hashicorp/packer/packer"
|
|
"github.com/ucloud/ucloud-sdk-go/ucloud"
|
|
)
|
|
|
|
type stepCopyUCloudImage struct {
|
|
ImageDestinations []ImageDestination
|
|
RegionId string
|
|
ProjectId string
|
|
}
|
|
|
|
func (s *stepCopyUCloudImage) Run(ctx context.Context, state multistep.StateBag) multistep.StepAction {
|
|
if len(s.ImageDestinations) == 0 {
|
|
return multistep.ActionContinue
|
|
}
|
|
|
|
client := state.Get("client").(*UCloudClient)
|
|
conn := client.uhostconn
|
|
ui := state.Get("ui").(packer.Ui)
|
|
|
|
srcImageId := state.Get("image_id").(string)
|
|
artifactImages := state.Get("ucloud_images").(*imageInfoSet)
|
|
expectedImages := newImageInfoSet(nil)
|
|
ui.Say(fmt.Sprintf("Copying image with %q...", srcImageId))
|
|
for _, v := range s.ImageDestinations {
|
|
if v.ProjectId == s.ProjectId && v.Region == s.RegionId {
|
|
continue
|
|
}
|
|
|
|
req := conn.NewCopyCustomImageRequest()
|
|
req.TargetProjectId = ucloud.String(v.ProjectId)
|
|
req.TargetRegion = ucloud.String(v.Region)
|
|
req.SourceImageId = ucloud.String(srcImageId)
|
|
req.TargetImageName = ucloud.String(v.Name)
|
|
req.TargetImageDescription = ucloud.String(v.Description)
|
|
|
|
resp, err := conn.CopyCustomImage(req)
|
|
if err != nil {
|
|
return halt(state, err, fmt.Sprintf("Error on copying image %q to %s:%s", srcImageId, v.ProjectId, v.Region))
|
|
}
|
|
|
|
image := imageInfo{
|
|
Region: v.Region,
|
|
ProjectId: v.ProjectId,
|
|
ImageId: resp.TargetImageId,
|
|
}
|
|
expectedImages.Set(image)
|
|
artifactImages.Set(image)
|
|
|
|
ui.Message(fmt.Sprintf("Copying image from %s:%s:%s to %s:%s:%s)",
|
|
s.ProjectId, s.RegionId, srcImageId, v.ProjectId, v.Region, resp.TargetImageId))
|
|
}
|
|
|
|
err := retry.Config{
|
|
Tries: 200,
|
|
ShouldRetry: func(err error) bool { return isNotCompleteError(err) },
|
|
RetryDelay: (&retry.Backoff{InitialBackoff: 2 * time.Second, MaxBackoff: 12 * time.Second, Multiplier: 2}).Linear,
|
|
}.Run(ctx, func(ctx context.Context) error {
|
|
for _, v := range expectedImages.GetAll() {
|
|
imageSet, err := client.describeImageByInfo(v.ProjectId, v.Region, v.ImageId)
|
|
if err != nil {
|
|
return fmt.Errorf("reading %s:%s:%s failed, %s", v.ProjectId, v.Region, v.ImageId, err)
|
|
}
|
|
|
|
if imageSet.State == imageStateAvailable {
|
|
expectedImages.Remove(v.Id())
|
|
continue
|
|
}
|
|
}
|
|
|
|
if len(expectedImages.GetAll()) != 0 {
|
|
return newNotCompleteError("copying image")
|
|
}
|
|
|
|
return nil
|
|
})
|
|
|
|
if err != nil {
|
|
var s []string
|
|
for _, v := range expectedImages.GetAll() {
|
|
s = append(s, fmt.Sprintf("%s:%s:%s", v.ProjectId, v.Region, v.ImageId))
|
|
}
|
|
|
|
return halt(state, err, fmt.Sprintf("Error on waiting for copying images %q to become available", strings.Join(s, ",")))
|
|
}
|
|
|
|
ui.Message(fmt.Sprintf("Copying image complete"))
|
|
return multistep.ActionContinue
|
|
}
|
|
|
|
func (s *stepCopyUCloudImage) Cleanup(state multistep.StateBag) {
|
|
_, cancelled := state.GetOk(multistep.StateCancelled)
|
|
_, halted := state.GetOk(multistep.StateHalted)
|
|
|
|
if !cancelled && !halted {
|
|
return
|
|
}
|
|
|
|
srcImageId := state.Get("image_id").(string)
|
|
ucloudImages := state.Get("ucloud_images").(*imageInfoSet)
|
|
imageInfos := ucloudImages.GetAll()
|
|
if len(imageInfos) == 0 {
|
|
return
|
|
} else if len(imageInfos) == 1 && imageInfos[0].ImageId == srcImageId {
|
|
return
|
|
}
|
|
|
|
ui := state.Get("ui").(packer.Ui)
|
|
client := state.Get("client").(*UCloudClient)
|
|
conn := client.uhostconn
|
|
ui.Say(fmt.Sprintf("Deleting copied image because of cancellation or error..."))
|
|
|
|
for _, v := range imageInfos {
|
|
if v.ImageId == srcImageId {
|
|
continue
|
|
}
|
|
|
|
req := conn.NewTerminateCustomImageRequest()
|
|
req.ProjectId = ucloud.String(v.ProjectId)
|
|
req.Region = ucloud.String(v.Region)
|
|
req.ImageId = ucloud.String(v.ImageId)
|
|
_, err := conn.TerminateCustomImage(req)
|
|
if err != nil {
|
|
ui.Error(fmt.Sprintf("Error on deleting copied image %q", v.ImageId))
|
|
}
|
|
}
|
|
|
|
ui.Message("Deleting copied image complete")
|
|
}
|