82 lines
2.2 KiB
Go
82 lines
2.2 KiB
Go
package ncloud
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"time"
|
|
|
|
ncloud "github.com/NaverCloudPlatform/ncloud-sdk-go/sdk"
|
|
"github.com/hashicorp/packer/packer"
|
|
"github.com/mitchellh/multistep"
|
|
)
|
|
|
|
type StepTerminateServerInstance struct {
|
|
Conn *ncloud.Conn
|
|
TerminateServerInstance func(serverInstanceNo string) error
|
|
Say func(message string)
|
|
Error func(e error)
|
|
}
|
|
|
|
func NewStepTerminateServerInstance(conn *ncloud.Conn, ui packer.Ui) *StepTerminateServerInstance {
|
|
var step = &StepTerminateServerInstance{
|
|
Conn: conn,
|
|
Say: func(message string) { ui.Say(message) },
|
|
Error: func(e error) { ui.Error(e.Error()) },
|
|
}
|
|
|
|
step.TerminateServerInstance = step.terminateServerInstance
|
|
|
|
return step
|
|
}
|
|
|
|
func (s *StepTerminateServerInstance) terminateServerInstance(serverInstanceNo string) error {
|
|
reqParams := new(ncloud.RequestTerminateServerInstances)
|
|
reqParams.ServerInstanceNoList = []string{serverInstanceNo}
|
|
|
|
serverInstanceList, err := s.Conn.TerminateServerInstances(reqParams)
|
|
if err != nil {
|
|
return fmt.Errorf("error code: %d , error message: %s", serverInstanceList.ReturnCode, serverInstanceList.ReturnMessage)
|
|
}
|
|
|
|
c1 := make(chan error, 1)
|
|
|
|
go func() {
|
|
reqParams := new(ncloud.RequestGetServerInstanceList)
|
|
reqParams.ServerInstanceNoList = []string{serverInstanceNo}
|
|
|
|
for {
|
|
|
|
serverInstanceList, err := s.Conn.GetServerInstanceList(reqParams)
|
|
if err != nil {
|
|
c1 <- fmt.Errorf("error code: %d , error message: %s", serverInstanceList.ReturnCode, serverInstanceList.ReturnMessage)
|
|
return
|
|
} else if serverInstanceList.TotalRows == 0 {
|
|
c1 <- nil
|
|
return
|
|
}
|
|
|
|
time.Sleep(time.Second * 3)
|
|
}
|
|
}()
|
|
|
|
select {
|
|
case res := <-c1:
|
|
return res
|
|
case <-time.After(time.Second * 60):
|
|
return errors.New("TIMEOUT : Can't terminate server instance")
|
|
}
|
|
}
|
|
|
|
func (s *StepTerminateServerInstance) Run(state multistep.StateBag) multistep.StepAction {
|
|
s.Say("Terminate Server Instance")
|
|
|
|
var serverInstanceNo = state.Get("InstanceNo").(string)
|
|
|
|
err := s.TerminateServerInstance(serverInstanceNo)
|
|
|
|
return processStepResult(err, s.Error, state)
|
|
}
|
|
|
|
func (*StepTerminateServerInstance) Cleanup(multistep.StateBag) {
|
|
}
|