mirror of
https://github.com/yunionio/cloudpods.git
synced 2026-09-24 16:03:43 +08:00
65 lines
2.0 KiB
Go
65 lines
2.0 KiB
Go
package tasks
|
|
|
|
import (
|
|
"context"
|
|
|
|
"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/cloudprovider"
|
|
"yunion.io/x/onecloud/pkg/compute/models"
|
|
"yunion.io/x/onecloud/pkg/util/logclient"
|
|
)
|
|
|
|
type GuestSyncstatusTask struct {
|
|
SGuestBaseTask
|
|
}
|
|
|
|
func init() {
|
|
taskman.RegisterTask(GuestSyncstatusTask{})
|
|
}
|
|
|
|
func (self *GuestSyncstatusTask) OnInit(ctx context.Context, obj db.IStandaloneModel, data jsonutils.JSONObject) {
|
|
guest := obj.(*models.SGuest)
|
|
host := guest.GetHost()
|
|
if host == nil || host.HostStatus == models.HOST_OFFLINE {
|
|
log.Errorf("host is not reachable")
|
|
guest.SetStatus(self.UserCred, models.VM_UNKNOWN, "Host not responding")
|
|
self.SetStageComplete(ctx, nil)
|
|
return
|
|
}
|
|
body, err := guest.GetDriver().RequestSyncstatusOnHost(ctx, guest, host, self.UserCred)
|
|
if err != nil {
|
|
log.Errorf("request_syncstatus_on_host: %s", err)
|
|
self.OnGetStatusFail(ctx, guest, err)
|
|
return
|
|
}
|
|
self.OnGetStatusSucc(ctx, guest, body)
|
|
}
|
|
|
|
func (self *GuestSyncstatusTask) OnGetStatusSucc(ctx context.Context, guest *models.SGuest, body jsonutils.JSONObject) {
|
|
statusStr, _ := body.GetString("status")
|
|
switch statusStr {
|
|
case cloudprovider.CloudVMStatusRunning:
|
|
statusStr = models.VM_RUNNING
|
|
case cloudprovider.CloudVMStatusSuspend:
|
|
statusStr = models.VM_SUSPEND
|
|
case cloudprovider.CloudVMStatusStopped:
|
|
statusStr = models.VM_READY
|
|
default:
|
|
statusStr = models.VM_UNKNOWN
|
|
}
|
|
statusData := jsonutils.NewDict()
|
|
statusData.Add(jsonutils.NewString(statusStr), "status")
|
|
guest.PerformStatus(ctx, self.UserCred, nil, statusData)
|
|
self.SetStageComplete(ctx, nil)
|
|
logclient.AddActionLog(guest, logclient.ACT_VM_SYNC_STATUS, "", self.UserCred, true)
|
|
}
|
|
|
|
func (self *GuestSyncstatusTask) OnGetStatusFail(ctx context.Context, guest *models.SGuest, err error) {
|
|
guest.SetStatus(self.UserCred, models.VM_UNKNOWN, err.Error())
|
|
self.SetStageComplete(ctx, nil)
|
|
logclient.AddActionLog(guest, logclient.ACT_VM_SYNC_STATUS, err, self.UserCred, false)
|
|
}
|