52 lines
1.5 KiB
Go
52 lines
1.5 KiB
Go
|
package cvm
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/hashicorp/packer/common/retry"
|
||
|
|
"github.com/hashicorp/packer/helper/multistep"
|
||
|
|
"github.com/hashicorp/packer/packer"
|
||
|
|
"github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/common"
|
||
|
|
cvm "github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/cvm/v20170312"
|
||
|
|
)
|
||
|
|
|
||
|
|
type stepDetachTempKeyPair struct {
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *stepDetachTempKeyPair) Run(ctx context.Context, state multistep.StateBag) multistep.StepAction {
|
||
|
|
client := state.Get("cvm_client").(*cvm.Client)
|
||
|
|
instance := state.Get("instance").(*cvm.Instance)
|
||
|
|
if _, ok := state.GetOk("temporary_key_pair_id"); !ok {
|
||
|
|
return multistep.ActionContinue
|
||
|
|
}
|
||
|
|
keyId := state.Get("temporary_key_pair_id").(*string)
|
||
|
|
ui := state.Get("ui").(packer.Ui)
|
||
|
|
ui.Say(fmt.Sprintf("Detaching temporary key pair %s...", *keyId))
|
||
|
|
req := cvm.NewDisassociateInstancesKeyPairsRequest()
|
||
|
|
req.KeyIds = []*string{keyId}
|
||
|
|
req.InstanceIds = []*string{instance.InstanceId}
|
||
|
|
req.ForceStop = common.BoolPtr(true)
|
||
|
|
err := retry.Config{
|
||
|
|
Tries: 60,
|
||
|
|
RetryDelay: (&retry.Backoff{
|
||
|
|
InitialBackoff: 5 * time.Second,
|
||
|
|
MaxBackoff: 5 * time.Second,
|
||
|
|
Multiplier: 2,
|
||
|
|
}).Linear,
|
||
|
|
}.Run(ctx, func(ctx context.Context) error {
|
||
|
|
_, err := client.DisassociateInstancesKeyPairs(req)
|
||
|
|
return err
|
||
|
|
})
|
||
|
|
if err != nil {
|
||
|
|
ui.Error(fmt.Sprintf("Fail to detach temporary key pair from instance! Error: %s", err))
|
||
|
|
state.Put("error", err)
|
||
|
|
return multistep.ActionHalt
|
||
|
|
}
|
||
|
|
return multistep.ActionContinue
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *stepDetachTempKeyPair) Cleanup(state multistep.StateBag) {
|
||
|
|
}
|