mirror of
https://github.com/yunionio/cloudpods.git
synced 2026-07-05 04:24:23 +08:00
110 lines
3.6 KiB
Go
110 lines
3.6 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 DiskResizeTask struct {
|
|
SDiskBaseTask
|
|
}
|
|
|
|
func init() {
|
|
taskman.RegisterTask(DiskResizeTask{})
|
|
}
|
|
|
|
func (self *DiskResizeTask) OnInit(ctx context.Context, obj db.IStandaloneModel, data jsonutils.JSONObject) {
|
|
disk := obj.(*models.SDisk)
|
|
storage := disk.GetStorage()
|
|
host := storage.GetMasterHost()
|
|
online := disk.GetRuningGuestCount() > 0
|
|
if online {
|
|
for _, guest := range disk.GetGuests() {
|
|
host = guest.GetHost()
|
|
}
|
|
}
|
|
resion := "Cannot find host for disk"
|
|
if host == nil || host.HostStatus != models.HOST_ONLINE {
|
|
disk.SetDiskReady(ctx, self.GetUserCred(), resion)
|
|
self.SetStageFailed(ctx, resion)
|
|
db.OpsLog.LogEvent(disk, db.ACT_RESIZE_FAIL, resion, self.GetUserCred())
|
|
} else {
|
|
disk.SetStatus(self.GetUserCred(), models.DISK_START_RESIZE, "")
|
|
for _, guest := range disk.GetGuests() {
|
|
guest.SetStatus(self.GetUserCred(), models.VM_RESIZE_DISK, "")
|
|
}
|
|
self.StartResizeDisk(ctx, host, storage, disk, online)
|
|
}
|
|
}
|
|
|
|
func (self *DiskResizeTask) StartResizeDisk(ctx context.Context, host *models.SHost, storage *models.SStorage, disk *models.SDisk, online bool) {
|
|
log.Infof("Resizing disk on host %s ...", host.GetName())
|
|
self.SetStage("on_disk_resize_complete", nil)
|
|
size, _ := self.GetParams().Int("size")
|
|
proc := host.GetHostDriver().RequestResizeDiskOnHost
|
|
if online {
|
|
proc = host.GetHostDriver().RequestResizeDiskOnHostOnline
|
|
}
|
|
if err := proc(host, storage, disk, size, self); err != nil {
|
|
log.Errorf("request_resize_disk_on_host: %v", err)
|
|
self.OnStartResizeDiskFailed(ctx, disk, err)
|
|
return
|
|
}
|
|
self.OnStartResizeDiskSucc(ctx, disk)
|
|
}
|
|
|
|
func (self *DiskResizeTask) OnStartResizeDiskSucc(ctx context.Context, disk *models.SDisk) {
|
|
disk.SetStatus(self.GetUserCred(), models.DISK_RESIZING, "")
|
|
}
|
|
|
|
func (self *DiskResizeTask) OnStartResizeDiskFailed(ctx context.Context, disk *models.SDisk, resion error) {
|
|
disk.SetDiskReady(ctx, self.GetUserCred(), resion.Error())
|
|
self.SetStageFailed(ctx, resion.Error())
|
|
db.OpsLog.LogEvent(disk, db.ACT_RESIZE_FAIL, resion.Error(), self.GetUserCred())
|
|
}
|
|
|
|
func (self *DiskResizeTask) OnDiskResizeComplete(ctx context.Context, disk *models.SDisk, data jsonutils.JSONObject) {
|
|
jSize, err := data.Get("disk_size")
|
|
if err != nil {
|
|
log.Errorf("OnDiskResizeComplete error: %s", err.Error())
|
|
self.OnStartResizeDiskFailed(ctx, disk, err)
|
|
return
|
|
}
|
|
size, err := jSize.Int()
|
|
if err != nil {
|
|
log.Errorf("OnDiskResizeComplete error: %s", err.Error())
|
|
self.OnStartResizeDiskFailed(ctx, disk, err)
|
|
return
|
|
}
|
|
oldStatus := disk.Status
|
|
_, err = disk.GetModelManager().TableSpec().Update(disk, func() error {
|
|
disk.Status = models.DISK_READY
|
|
disk.DiskSize = int(size)
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
log.Errorf("OnDiskResizeComplete error: %s", err.Error())
|
|
self.OnStartResizeDiskFailed(ctx, disk, err)
|
|
return
|
|
}
|
|
disk.SetDiskReady(ctx, self.GetUserCred(), "")
|
|
notes := fmt.Sprintf("%s=>%s", oldStatus, disk.Status)
|
|
db.OpsLog.LogEvent(disk, db.ACT_UPDATE_STATUS, notes, self.UserCred)
|
|
self.CleanHostSchedCache(disk)
|
|
db.OpsLog.LogEvent(disk, db.ACT_RESIZE, disk.GetShortDesc(), self.UserCred)
|
|
self.SetStageComplete(ctx, disk.GetShortDesc())
|
|
self.finalReleasePendingUsage(ctx)
|
|
}
|
|
|
|
func (self *DiskResizeTask) OnDiskResizeCompleteFailed(ctx context.Context, disk *models.SDisk, resion error) {
|
|
disk.SetDiskReady(ctx, self.GetUserCred(), resion.Error())
|
|
db.OpsLog.LogEvent(disk, db.ACT_RESIZE_FAIL, disk.GetShortDesc(), self.UserCred)
|
|
}
|