mirror of
https://github.com/yunionio/cloudpods.git
synced 2026-10-29 17:58:36 +08:00
66 lines
2.2 KiB
Go
66 lines
2.2 KiB
Go
package tasks
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"yunion.io/x/jsonutils"
|
|
"yunion.io/x/log"
|
|
"yunion.io/x/onecloud/pkg/cloudcommon/db"
|
|
"yunion.io/x/onecloud/pkg/cloudcommon/db/taskman"
|
|
"yunion.io/x/onecloud/pkg/compute/models"
|
|
)
|
|
|
|
type GuestSaveImageTask struct {
|
|
SGuestBaseTask
|
|
}
|
|
|
|
func init() {
|
|
taskman.RegisterTask(GuestSaveImageTask{})
|
|
}
|
|
|
|
func (self *GuestSaveImageTask) OnInit(ctx context.Context, obj db.IStandaloneModel, body jsonutils.JSONObject) {
|
|
guest := obj.(*models.SGuest)
|
|
log.Infof("Saving server image: %s", guest.Name)
|
|
if restart, _ := self.GetParams().Bool("restart"); restart {
|
|
self.SetStage("on_stop_server_complete", nil)
|
|
guest.StartGuestStopTask(ctx, self.GetUserCred(), false, self.GetTaskId())
|
|
} else {
|
|
self.OnStopServerComplete(ctx, guest, nil)
|
|
}
|
|
}
|
|
|
|
func (self *GuestSaveImageTask) OnStopServerComplete(ctx context.Context, guest *models.SGuest, body jsonutils.JSONObject) {
|
|
if guest.Status != models.VM_READY {
|
|
resion := fmt.Sprintf("Server %s not in ready status", guest.Name)
|
|
log.Errorf(resion)
|
|
self.SetStageFailed(ctx, resion)
|
|
} else {
|
|
self.SetStage("on_save_root_image_complete", nil)
|
|
guest.SetStatus(self.GetUserCred(), models.VM_START_SAVE_DISK, "")
|
|
disks := guest.CategorizeDisks()
|
|
if err := disks.Root.StartDiskSaveTask(ctx, self.GetUserCred(), self.GetParams(), self.GetTaskId()); err != nil {
|
|
self.SetStageFailed(ctx, err.Error())
|
|
}
|
|
}
|
|
}
|
|
|
|
func (self *GuestSaveImageTask) OnSaveRootImageComplete(ctx context.Context, guest *models.SGuest, data jsonutils.JSONObject) {
|
|
if restart, _ := self.GetParams().Bool("restart"); restart {
|
|
self.SetStage("on_start_server_complete", nil)
|
|
guest.StartGueststartTask(ctx, self.GetUserCred(), nil, self.GetTaskId())
|
|
} else {
|
|
self.SetStageComplete(ctx, nil)
|
|
}
|
|
}
|
|
|
|
func (self *GuestSaveImageTask) OnSaveRootImageCompleteFailed(ctx context.Context, guest *models.SGuest, data jsonutils.JSONObject) {
|
|
log.Errorf("Guest save root image failed: %s", data.PrettyString())
|
|
guest.SetStatus(self.GetUserCred(), models.VM_SAVE_DISK_FAILED, data.PrettyString())
|
|
self.SetStageFailed(ctx, data.PrettyString())
|
|
}
|
|
|
|
func (self *GuestSaveImageTask) OnStartServerCompleteFailed(ctx context.Context, guest *models.SGuest, data jsonutils.JSONObject) {
|
|
self.SetStageComplete(ctx, nil)
|
|
}
|