package ncloud import ( "context" "errors" "fmt" "io/ioutil" "log" "time" "github.com/NaverCloudPlatform/ncloud-sdk-go-v2/ncloud" "github.com/NaverCloudPlatform/ncloud-sdk-go-v2/services/server" "github.com/hashicorp/packer/helper/multistep" "github.com/hashicorp/packer/packer" ) type StepCreateServerInstance struct { Conn *NcloudAPIClient CreateServerInstance func(loginKeyName string, zoneNo string, feeSystemTypeCode string) (string, error) CheckServerInstanceStatusIsRunning func(serverInstanceNo string) error Say func(message string) Error func(e error) Config *Config serverInstanceNo string } func NewStepCreateServerInstance(conn *NcloudAPIClient, ui packer.Ui, config *Config) *StepCreateServerInstance { var step = &StepCreateServerInstance{ Conn: conn, Say: func(message string) { ui.Say(message) }, Error: func(e error) { ui.Error(e.Error()) }, Config: config, } step.CreateServerInstance = step.createServerInstance return step } func (s *StepCreateServerInstance) createServerInstance(loginKeyName string, zoneNo string, feeSystemTypeCode string) (string, error) { reqParams := new(server.CreateServerInstancesRequest) reqParams.ServerProductCode = &s.Config.ServerProductCode reqParams.MemberServerImageNo = &s.Config.MemberServerImageNo if s.Config.MemberServerImageNo == "" { reqParams.ServerImageProductCode = &s.Config.ServerImageProductCode } reqParams.LoginKeyName = &loginKeyName reqParams.ZoneNo = &zoneNo reqParams.FeeSystemTypeCode = &feeSystemTypeCode if s.Config.UserData != "" { reqParams.UserData = &s.Config.UserData } if s.Config.UserDataFile != "" { contents, err := ioutil.ReadFile(s.Config.UserDataFile) if err != nil { return "", fmt.Errorf("Problem reading user data file: %s", err) } reqParams.UserData = ncloud.String(string(contents)) } if s.Config.AccessControlGroupConfigurationNo != "" { reqParams.AccessControlGroupConfigurationNoList = []*string{&s.Config.AccessControlGroupConfigurationNo} } serverInstanceList, err := s.Conn.server.V2Api.CreateServerInstances(reqParams) if err != nil { return "", err } s.serverInstanceNo = *serverInstanceList.ServerInstanceList[0].ServerInstanceNo s.Say(fmt.Sprintf("Server Instance is creating. Server InstanceNo is %s", s.serverInstanceNo)) log.Println("Server Instance information : ", serverInstanceList.ServerInstanceList[0]) if err := waiterServerInstanceStatus(s.Conn, s.serverInstanceNo, "RUN", 30*time.Minute); err != nil { return "", errors.New("TIMEOUT : server instance status is not running") } s.Say(fmt.Sprintf("Server Instance is created. Server InstanceNo is %s", s.serverInstanceNo)) return s.serverInstanceNo, nil } func (s *StepCreateServerInstance) Run(ctx context.Context, state multistep.StateBag) multistep.StepAction { s.Say("Create Server Instance") var loginKey = state.Get("LoginKey").(*LoginKey) var zoneNo = state.Get("ZoneNo").(string) feeSystemTypeCode := "MTRAT" if _, ok := state.GetOk("FeeSystemTypeCode"); ok { feeSystemTypeCode = state.Get("FeeSystemTypeCode").(string) } serverInstanceNo, err := s.CreateServerInstance(loginKey.KeyName, zoneNo, feeSystemTypeCode) if err == nil { state.Put("InstanceNo", serverInstanceNo) // instance_id is the generic term used so that users can have access to the // instance id inside of the provisioners, used in step_provision. state.Put("instance_id", serverInstanceNo) } return processStepResult(err, s.Error, state) } func (s *StepCreateServerInstance) Cleanup(state multistep.StateBag) { _, cancelled := state.GetOk(multistep.StateCancelled) _, halted := state.GetOk(multistep.StateHalted) if !cancelled && !halted { return } if s.serverInstanceNo == "" { return } reqParams := new(server.GetServerInstanceListRequest) reqParams.ServerInstanceNoList = []*string{&s.serverInstanceNo} serverInstanceList, err := s.Conn.server.V2Api.GetServerInstanceList(reqParams) if err != nil || *serverInstanceList.TotalRows == 0 { return } s.Say("Clean up Server Instance") serverInstance := serverInstanceList.ServerInstanceList[0] // stop server instance if *serverInstance.ServerInstanceStatus.Code != "NSTOP" && *serverInstance.ServerInstanceStatus.Code != "TERMT" { reqParams := new(server.StopServerInstancesRequest) reqParams.ServerInstanceNoList = []*string{&s.serverInstanceNo} log.Println("Stop Server Instance") s.Conn.server.V2Api.StopServerInstances(reqParams) waiterServerInstanceStatus(s.Conn, s.serverInstanceNo, "NSTOP", time.Minute) } // terminate server instance if *serverInstance.ServerInstanceStatus.Code != "TERMT" { reqParams := new(server.TerminateServerInstancesRequest) reqParams.ServerInstanceNoList = []*string{&s.serverInstanceNo} log.Println("Terminate Server Instance") s.Conn.server.V2Api.TerminateServerInstances(reqParams) } }