- instance snapshot

- instance clone - create form snapshot don't reset passwd
2026-09-21 08:33:54 +08:00 · 2019-08-29 17:49:24 +08:00
parent 905fb307a8
commit 8fc4b2920a
26 changed files with 1203 additions and 26 deletions
--- a/pkg/compute/tasks/instance_snapshot_reset_task.go
+++ b/pkg/compute/tasks/instance_snapshot_reset_task.go
@@ -0,0 +1,103 @@
+package tasks
+
+import (
+	"context"
+
+	"yunion.io/x/jsonutils"
+
+	"yunion.io/x/onecloud/pkg/apis/compute"
+	"yunion.io/x/onecloud/pkg/cloudcommon/db"
+	"yunion.io/x/onecloud/pkg/cloudcommon/db/taskman"
+	"yunion.io/x/onecloud/pkg/cloudcommon/notifyclient"
+	"yunion.io/x/onecloud/pkg/compute/models"
+	"yunion.io/x/onecloud/pkg/util/logclient"
+)
+
+type InstanceSnapshotResetTask struct {
+	taskman.STask
+}
+
+func init() {
+	taskman.RegisterTask(InstanceSnapshotResetTask{})
+}
+
+func (self *InstanceSnapshotResetTask) taskFail(
+	ctx context.Context, isp *models.SInstanceSnapshot, guest *models.SGuest, reason string) {
+
+	if guest == nil {
+		guest = models.GuestManager.FetchGuestById(isp.GuestId)
+	}
+	guest.SetStatus(self.UserCred, compute.VM_SNAPSHOT_RESET_FAILED, reason)
+
+	db.OpsLog.LogEvent(guest, db.ACT_VM_RESET_SNAPSHOT_FAILED, reason, self.UserCred)
+	logclient.AddActionLogWithStartable(self, guest, logclient.ACT_VM_RESET, false, self.UserCred, false)
+	notifyclient.NotifySystemError(guest.GetId(), isp.Name, compute.VM_SNAPSHOT_RESET_FAILED, reason)
+	self.SetStageFailed(ctx, reason)
+}
+
+func (self *InstanceSnapshotResetTask) taskComplete(
+	ctx context.Context, isp *models.SInstanceSnapshot, guest *models.SGuest, data jsonutils.JSONObject) {
+
+	if guest == nil {
+		guest = models.GuestManager.FetchGuestById(isp.GuestId)
+	}
+	guest.StartSyncstatus(ctx, self.UserCred, "")
+
+	db.OpsLog.LogEvent(isp, db.ACT_VM_RESET_SNAPSHOT, "instance snapshot reset success", self.UserCred)
+	logclient.AddActionLogWithStartable(self, guest, logclient.ACT_VM_RESET, false, self.UserCred, true)
+	self.SetStageComplete(ctx, nil)
+}
+
+func (self *InstanceSnapshotResetTask) OnInit(
+	ctx context.Context, obj db.IStandaloneModel, data jsonutils.JSONObject) {
+
+	isp := obj.(*models.SInstanceSnapshot)
+	guest := models.GuestManager.FetchGuestById(isp.GuestId)
+
+	self.GuestDiskResetTask(ctx, isp, guest, 0)
+}
+
+func (self *InstanceSnapshotResetTask) GuestDiskResetTask(
+	ctx context.Context, isp *models.SInstanceSnapshot, guest *models.SGuest, diskIndex int) {
+
+	disks := guest.GetDisks()
+	if diskIndex >= len(disks) {
+		self.taskComplete(ctx, isp, guest, nil)
+		return
+	}
+
+	isj, err := isp.GetInstanceSnapshotJointAt(diskIndex)
+	if err != nil {
+		self.taskFail(ctx, isp, guest, err.Error())
+		return
+	}
+
+	params := jsonutils.NewDict()
+	params.Set("disk_index", jsonutils.NewInt(int64(diskIndex)))
+	self.SetStage("OnDiskReset", params)
+
+	disk := disks[diskIndex].GetDisk()
+	err = disk.StartResetDisk(ctx, self.UserCred, isj.SnapshotId, false, guest, self.Id)
+	if err != nil {
+		self.taskFail(ctx, isp, guest, err.Error())
+		return
+	}
+}
+
+func (self *InstanceSnapshotResetTask) OnDiskReset(
+	ctx context.Context, isp *models.SInstanceSnapshot, data jsonutils.JSONObject) {
+
+	guest := models.GuestManager.FetchGuestById(isp.GuestId)
+
+	diskIndex, err := self.Params.Int("disk_index")
+	if err != nil {
+		self.taskFail(ctx, isp, guest, err.Error())
+		return
+	}
+	self.GuestDiskResetTask(ctx, isp, guest, int(diskIndex+1))
+}
+
+func (self *InstanceSnapshotResetTask) OnDiskResetFailed(
+	ctx context.Context, isp *models.SInstanceSnapshot, data jsonutils.JSONObject) {
+	self.taskFail(ctx, isp, nil, data.String())
+}