| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730 |
- // Copyright 2019 Yunion
- //
- // Licensed under the Apache License, Version 2.0 (the "License");
- // you may not use this file except in compliance with the License.
- // You may obtain a copy of the License at
- //
- // http://www.apache.org/licenses/LICENSE-2.0
- //
- // Unless required by applicable law or agreed to in writing, software
- // distributed under the License is distributed on an "AS IS" BASIS,
- // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- // See the License for the specific language governing permissions and
- // limitations under the License.
- package guestdrivers
- import (
- "context"
- "fmt"
- "net/http"
- "strings"
- "time"
- "yunion.io/x/cloudmux/pkg/cloudprovider"
- "yunion.io/x/jsonutils"
- "yunion.io/x/log"
- "yunion.io/x/pkg/errors"
- "yunion.io/x/pkg/util/billing"
- "yunion.io/x/pkg/util/fileutils"
- "yunion.io/x/pkg/util/osprofile"
- "yunion.io/x/pkg/util/regutils"
- "yunion.io/x/pkg/utils"
- api "yunion.io/x/onecloud/pkg/apis/compute"
- "yunion.io/x/onecloud/pkg/cloudcommon/consts"
- "yunion.io/x/onecloud/pkg/cloudcommon/db/quotas"
- "yunion.io/x/onecloud/pkg/cloudcommon/db/taskman"
- "yunion.io/x/onecloud/pkg/cloudcommon/tsdb"
- guestdriver_types "yunion.io/x/onecloud/pkg/compute/guestdrivers/types"
- "yunion.io/x/onecloud/pkg/compute/models"
- "yunion.io/x/onecloud/pkg/compute/options"
- "yunion.io/x/onecloud/pkg/httperrors"
- "yunion.io/x/onecloud/pkg/mcclient"
- "yunion.io/x/onecloud/pkg/mcclient/auth"
- "yunion.io/x/onecloud/pkg/mcclient/modules/scheduler"
- )
- type SBaseGuestScheduleDriver struct{}
- func (d SBaseGuestScheduleDriver) DoScheduleSKUFilter() bool { return true }
- func (d SBaseGuestScheduleDriver) DoScheduleCPUFilter() bool { return true }
- func (d SBaseGuestScheduleDriver) DoScheduleMemoryFilter() bool { return true }
- func (d SBaseGuestScheduleDriver) DoScheduleStorageFilter() bool { return true }
- func (d SBaseGuestScheduleDriver) DoScheduleCloudproviderTagFilter() bool { return false }
- type SBaseGuestDriver struct {
- SBaseGuestScheduleDriver
- }
- func (drv *SBaseGuestDriver) IsAllowSaveImageOnRunning() bool {
- return false
- }
- func (drv *SBaseGuestDriver) StartGuestCreateTask(guest *models.SGuest, ctx context.Context, userCred mcclient.TokenCredential, data *jsonutils.JSONDict, pendingUsage quotas.IQuota, parentTaskId string) error {
- task, err := taskman.TaskManager.NewTask(ctx, "GuestCreateTask", guest, userCred, data, parentTaskId, "", pendingUsage)
- if err != nil {
- return err
- }
- task.ScheduleRun(nil)
- return nil
- }
- func (drv *SBaseGuestDriver) OnGuestCreateTaskComplete(ctx context.Context, guest *models.SGuest, task taskman.ITask) error {
- duration, _ := task.GetParams().GetString("duration")
- if len(duration) > 0 {
- bc, err := billing.ParseBillingCycle(duration)
- if err == nil && guest.ExpiredAt.IsZero() {
- models.SaveRenewInfo(ctx, task.GetUserCred(), guest, &bc, nil, "")
- }
- if jsonutils.QueryBoolean(task.GetParams(), "auto_prepaid_recycle", false) {
- err := guest.CanPerformPrepaidRecycle()
- if err == nil {
- task.SetStageComplete(ctx, nil)
- guest.DoPerformPrepaidRecycle(ctx, task.GetUserCred(), true)
- return nil
- }
- }
- }
- if jsonutils.QueryBoolean(task.GetParams(), "auto_start", false) {
- task.SetStage("OnAutoStartGuest", nil)
- return guest.StartGueststartTask(ctx, task.GetUserCred(), nil, task.GetTaskId())
- } else {
- task.SetStage("OnSyncStatusComplete", nil)
- return guest.StartSyncstatus(ctx, task.GetUserCred(), task.GetTaskId())
- }
- }
- func (drv *SBaseGuestDriver) StartDeleteGuestTask(ctx context.Context, userCred mcclient.TokenCredential, guest *models.SGuest, params *jsonutils.JSONDict, parentTaskId string) error {
- task, err := taskman.TaskManager.NewTask(ctx, "GenericGuestDeleteTask", guest, userCred, params, parentTaskId, "", nil)
- if err != nil {
- return err
- }
- task.ScheduleRun(nil)
- return nil
- }
- func (drv *SBaseGuestDriver) ValidateImage(ctx context.Context, image *cloudprovider.SImage) error {
- return nil
- }
- func (drv *SBaseGuestDriver) RequestDetachDisksFromGuestForDelete(ctx context.Context, guest *models.SGuest, task taskman.ITask) error {
- task.ScheduleRun(nil)
- return nil
- }
- func (drv *SBaseGuestDriver) OnDeleteGuestFinalCleanup(ctx context.Context, guest *models.SGuest, userCred mcclient.TokenCredential) error {
- return guest.DeleteAllDisksInDB(ctx, userCred)
- }
- func (drv *SBaseGuestDriver) RequestDetachDisk(ctx context.Context, guest *models.SGuest, disk *models.SDisk, task taskman.ITask) error {
- task.ScheduleRun(nil)
- return nil
- }
- func (drv *SBaseGuestDriver) RequestChangeBillingType(ctx context.Context, guest *models.SGuest, task taskman.ITask) error {
- return errors.Wrapf(cloudprovider.ErrNotImplemented, "RequestChangeBillingType")
- }
- func (drv *SBaseGuestDriver) RequestAttachDisk(ctx context.Context, guest *models.SGuest, disk *models.SDisk, task taskman.ITask) error {
- task.ScheduleRun(nil)
- return nil
- }
- func (drv *SBaseGuestDriver) RequestOpenForward(ctx context.Context, userCred mcclient.TokenCredential, guest *models.SGuest, req *guestdriver_types.OpenForwardRequest) (*guestdriver_types.OpenForwardResponse, error) {
- return nil, cloudprovider.ErrNotImplemented
- }
- func (drv *SBaseGuestDriver) RequestListForward(ctx context.Context, userCred mcclient.TokenCredential, guest *models.SGuest, req *guestdriver_types.ListForwardRequest) (*guestdriver_types.ListForwardResponse, error) {
- return nil, cloudprovider.ErrNotImplemented
- }
- func (drv *SBaseGuestDriver) RequestCloseForward(ctx context.Context, userCred mcclient.TokenCredential, guest *models.SGuest, req *guestdriver_types.CloseForwardRequest) (*guestdriver_types.CloseForwardResponse, error) {
- return nil, cloudprovider.ErrNotImplemented
- }
- func (drv *SBaseGuestDriver) RequestSaveImage(ctx context.Context, userCred mcclient.TokenCredential, guest *models.SGuest, task taskman.ITask) error {
- return errors.Wrapf(cloudprovider.ErrNotImplemented, "RequestSaveImage")
- }
- func (drv *SBaseGuestDriver) RequestGuestCreateAllDisks(ctx context.Context, guest *models.SGuest, task taskman.ITask) error {
- return fmt.Errorf("Not Implement")
- }
- func (drv *SBaseGuestDriver) GetDetachDiskStatus() ([]string, error) {
- return []string{}, fmt.Errorf("This Guest driver dose not implement GetDetachDiskStatus")
- }
- func (drv *SBaseGuestDriver) GetAttachDiskStatus() ([]string, error) {
- return []string{}, fmt.Errorf("This Guest driver dose not implement GetAttachDiskStatus")
- }
- func (drv *SBaseGuestDriver) GetRebuildRootStatus() ([]string, error) {
- return []string{}, fmt.Errorf("This Guest driver dose not implement GetRebuildRootStatus")
- }
- func (drv *SBaseGuestDriver) IsRebuildRootSupportChangeImage() bool {
- return true
- }
- func (drv *SBaseGuestDriver) IsRebuildRootSupportChangeUEFI() bool {
- return true
- }
- func (drv *SBaseGuestDriver) GetChangeInstanceTypeStatus() ([]string, error) {
- return []string{}, fmt.Errorf("This Guest driver dose not implement GetChangeInstanceTypeStatus")
- }
- func (drv *SBaseGuestDriver) ValidateDetachDisk(ctx context.Context, userCred mcclient.TokenCredential, guest *models.SGuest, disk *models.SDisk) error {
- return nil
- }
- func (drv *SBaseGuestDriver) ValidateCreateEip(ctx context.Context, userCred mcclient.TokenCredential, input api.ServerCreateEipInput) error {
- return httperrors.NewInputParameterError("Not Implement ValidateCreateEip")
- }
- func (drv *SBaseGuestDriver) ValidateResizeDisk(guest *models.SGuest, disk *models.SDisk, storage *models.SStorage) error {
- return fmt.Errorf("This Guest driver dose not implement ValidateResizeDisk")
- }
- func (drv *SBaseGuestDriver) GetDeployStatus() ([]string, error) {
- return []string{}, fmt.Errorf("This Guest driver dose not implement GetDeployStatus")
- }
- func (drv *SBaseGuestDriver) IsNeedRestartForResetLoginInfo() bool {
- return true
- }
- func (drv *SBaseGuestDriver) RequestDeleteDetachedDisk(ctx context.Context, disk *models.SDisk, task taskman.ITask, isPurge bool) error {
- return fmt.Errorf("Not Implement")
- }
- func (drv *SBaseGuestDriver) RequestSuspendOnHost(ctx context.Context, guest *models.SGuest, task taskman.ITask) error {
- return fmt.Errorf("Not Implement")
- }
- func (drv *SBaseGuestDriver) RequestResumeOnHost(ctx context.Context, guest *models.SGuest, task taskman.ITask) error {
- return fmt.Errorf("Not Implement")
- }
- func (drv *SBaseGuestDriver) StartGuestResetTask(guest *models.SGuest, ctx context.Context, userCred mcclient.TokenCredential, isHard bool, parentTaskId string) error {
- return fmt.Errorf("Not Implement")
- }
- func (drv *SBaseGuestDriver) StartGuestRestartTask(guest *models.SGuest, ctx context.Context, userCred mcclient.TokenCredential, isForce bool, parentTaskId string) error {
- data := jsonutils.NewDict()
- data.Set("is_force", jsonutils.NewBool(isForce))
- if err := guest.SetStatus(ctx, userCred, api.VM_STOPPING, ""); err != nil {
- return err
- }
- task, err := taskman.TaskManager.NewTask(ctx, "GuestRestartTask", guest, userCred, nil, parentTaskId, "", nil)
- if err != nil {
- return err
- }
- task.ScheduleRun(nil)
- return nil
- }
- func (drv *SBaseGuestDriver) RequestSoftReset(ctx context.Context, guest *models.SGuest, task taskman.ITask) error {
- return fmt.Errorf("Not Implement")
- }
- func (drv *SBaseGuestDriver) AllowReconfigGuest() bool {
- return true
- }
- func (drv *SBaseGuestDriver) DoGuestCreateDisksTask(ctx context.Context, guest *models.SGuest, task taskman.ITask) error {
- return fmt.Errorf("Not Implement")
- }
- func (drv *SBaseGuestDriver) RequestChangeVmConfig(ctx context.Context, guest *models.SGuest, task taskman.ITask, instanceType string, vcpuCount, cpuSockets, vmemSize int64) error {
- return errors.Wrapf(cloudprovider.ErrNotImplemented, "RequestChangeVmConfig")
- }
- func (drv *SBaseGuestDriver) NeedRequestGuestHotAddIso(ctx context.Context, guest *models.SGuest) bool {
- return false
- }
- func (drv *SBaseGuestDriver) RequestGuestHotAddIso(ctx context.Context, guest *models.SGuest, path string, boot bool, task taskman.ITask) error {
- return fmt.Errorf("Not Implement")
- }
- func (drv *SBaseGuestDriver) RequestGuestHotRemoveIso(ctx context.Context, guest *models.SGuest, task taskman.ITask) error {
- return fmt.Errorf("Not Implement")
- }
- func (drv *SBaseGuestDriver) RequestRebuildRootDisk(ctx context.Context, guest *models.SGuest, task taskman.ITask) error {
- return fmt.Errorf("Not Implement")
- }
- func (drv *SBaseGuestDriver) NeedRequestGuestHotAddVfd(ctx context.Context, guest *models.SGuest) bool {
- return false
- }
- func (drv *SBaseGuestDriver) RequestGuestHotAddVfd(ctx context.Context, guest *models.SGuest, path string, boot bool, task taskman.ITask) error {
- return fmt.Errorf("Not Implement")
- }
- func (drv *SBaseGuestDriver) RequestGuestHotRemoveVfd(ctx context.Context, guest *models.SGuest, task taskman.ITask) error {
- return fmt.Errorf("Not Implement")
- }
- func (drv *SBaseGuestDriver) RequestDiskSnapshot(ctx context.Context, guest *models.SGuest, task taskman.ITask, snapshotId, diskId string) error {
- return fmt.Errorf("Not Implement")
- }
- func (drv *SBaseGuestDriver) RequestDeleteSnapshot(ctx context.Context, guest *models.SGuest, task taskman.ITask, params *jsonutils.JSONDict) error {
- return fmt.Errorf("Not Implement")
- }
- func (drv *SBaseGuestDriver) RequestReloadDiskSnapshot(ctx context.Context, guest *models.SGuest, task taskman.ITask, params *jsonutils.JSONDict) error {
- return fmt.Errorf("Not Implement")
- }
- func (drv *SBaseGuestDriver) RequestSyncToBackup(ctx context.Context, guest *models.SGuest, task taskman.ITask) error {
- return fmt.Errorf("Not Implement")
- }
- func (drv *SBaseGuestDriver) RequestSlaveBlockStreamDisks(ctx context.Context, guest *models.SGuest, task taskman.ITask) error {
- return fmt.Errorf("Not Implement")
- }
- func (drv *SBaseGuestDriver) GetMaxSecurityGroupCount() int {
- return 5
- }
- func (drv *SBaseGuestDriver) getTaskRequestHeader(task taskman.ITask) http.Header {
- return task.GetTaskRequestHeader()
- }
- func (drv *SBaseGuestDriver) IsSupportedBillingCycle(bc billing.SBillingCycle) bool {
- return true
- }
- func (drv *SBaseGuestDriver) IsSupportPostpaidExpire() bool {
- return true
- }
- func (drv *SBaseGuestDriver) IsSupportShutdownMode() bool {
- return false
- }
- func (drv *SBaseGuestDriver) RequestRenewInstance(ctx context.Context, guest *models.SGuest, bc billing.SBillingCycle) (time.Time, error) {
- return time.Time{}, nil
- }
- func (drv *SBaseGuestDriver) IsSupportEip() bool {
- return false
- }
- func (drv *SBaseGuestDriver) IsSupportPublicIp() bool {
- return false
- }
- func (drv *SBaseGuestDriver) RemoteDeployGuestForCreate(ctx context.Context, userCred mcclient.TokenCredential, guest *models.SGuest, host *models.SHost, desc cloudprovider.SManagedVMCreateConfig) (jsonutils.JSONObject, error) {
- return nil, cloudprovider.ErrNotSupported
- }
- func (drv *SBaseGuestDriver) RemoteDeployGuestSyncHost(ctx context.Context, userCred mcclient.TokenCredential, guest *models.SGuest, host *models.SHost, iVM cloudprovider.ICloudVM) (cloudprovider.ICloudHost, error) {
- return nil, cloudprovider.ErrNotSupported
- }
- func (drv *SBaseGuestDriver) RemoteActionAfterGuestCreated(ctx context.Context, userCred mcclient.TokenCredential, guest *models.SGuest, host *models.SHost, ivm cloudprovider.ICloudVM, desc *cloudprovider.SManagedVMCreateConfig) {
- return
- }
- func (drv *SBaseGuestDriver) RemoteDeployGuestForDeploy(ctx context.Context, guest *models.SGuest, ihost cloudprovider.ICloudHost, task taskman.ITask, desc cloudprovider.SManagedVMCreateConfig) (jsonutils.JSONObject, error) {
- return nil, cloudprovider.ErrNotSupported
- }
- func (drv *SBaseGuestDriver) RemoteDeployGuestForRebuildRoot(ctx context.Context, guest *models.SGuest, ihost cloudprovider.ICloudHost, task taskman.ITask, desc cloudprovider.SManagedVMCreateConfig) (jsonutils.JSONObject, error) {
- return nil, cloudprovider.ErrNotSupported
- }
- func (drv *SBaseGuestDriver) GetGuestInitialStateAfterCreate() string {
- return api.VM_READY
- }
- func (drv *SBaseGuestDriver) GetGuestInitialStateAfterRebuild() string {
- return api.VM_READY
- }
- func (drv *SBaseGuestDriver) IsNeedInjectPasswordByCloudInit() bool {
- return false
- }
- func (drv *SBaseGuestDriver) GetWindowsUserDataType() string {
- return cloudprovider.CLOUD_POWER_SHELL
- }
- func (drv *SBaseGuestDriver) IsWindowsUserDataTypeNeedEncode() bool {
- return false
- }
- func (drv *SBaseGuestDriver) IsSupportdDcryptPasswordFromSecretKey() bool {
- return true
- }
- func (drv *SBaseGuestDriver) GetUserDataType() string {
- return cloudprovider.CLOUD_CONFIG
- }
- func (drv *SBaseGuestDriver) GetDefaultAccount(osType, osDist, imageType string) string {
- if strings.ToLower(osType) == strings.ToLower(osprofile.OS_TYPE_WINDOWS) {
- return api.VM_DEFAULT_WINDOWS_LOGIN_USER
- }
- return api.VM_DEFAULT_LINUX_LOGIN_USER
- }
- func (drv *SBaseGuestDriver) OnGuestChangeCpuMemFailed(ctx context.Context, guest *models.SGuest, data *jsonutils.JSONDict, task taskman.ITask) error {
- return nil
- }
- func (drv *SBaseGuestDriver) RequestSyncConfigOnHost(ctx context.Context, guest *models.SGuest, host *models.SHost, task taskman.ITask) error {
- return fmt.Errorf("SBaseGuestDriver: Not Implement")
- }
- func (drv *SBaseGuestDriver) IsSupportGuestClone() bool {
- return true
- }
- func (drv *SBaseGuestDriver) RequestSyncSecgroupsOnHost(ctx context.Context, guest *models.SGuest, host *models.SHost, task taskman.ITask) error {
- return nil // do nothing
- }
- func (drv *SBaseGuestDriver) IsSupportPublicipToEip() bool {
- return false
- }
- func (drv *SBaseGuestDriver) RequestConvertPublicipToEip(ctx context.Context, userCred mcclient.TokenCredential, guest *models.SGuest, task taskman.ITask) error {
- return fmt.Errorf("Not Implement RequestConvertPublicipToEip")
- }
- func (drv *SBaseGuestDriver) IsSupportSetAutoRenew() bool {
- return false
- }
- func (drv *SBaseGuestDriver) RequestSetAutoRenewInstance(ctx context.Context, userCred mcclient.TokenCredential, guest *models.SGuest, input api.GuestAutoRenewInput, task taskman.ITask) error {
- return fmt.Errorf("Not Implement RequestSetAutoRenewInstance")
- }
- func (drv *SBaseGuestDriver) IsSupportMigrate() bool {
- return false
- }
- func (drv *SBaseGuestDriver) IsSupportLiveMigrate() bool {
- return false
- }
- func (drv *SBaseGuestDriver) CheckMigrate(ctx context.Context, guest *models.SGuest, userCred mcclient.TokenCredential, input api.GuestMigrateInput) error {
- return nil
- }
- func (drv *SBaseGuestDriver) CheckLiveMigrate(ctx context.Context, guest *models.SGuest, userCred mcclient.TokenCredential, input api.GuestLiveMigrateInput) error {
- return nil
- }
- func (drv *SBaseGuestDriver) RequestMigrate(ctx context.Context, guest *models.SGuest, userCred mcclient.TokenCredential, input api.GuestMigrateInput, task taskman.ITask) error {
- return errors.Wrapf(cloudprovider.ErrNotImplemented, "RequestMigrate")
- }
- func (drv *SBaseGuestDriver) RequestLiveMigrate(ctx context.Context, guest *models.SGuest, userCred mcclient.TokenCredential, input api.GuestLiveMigrateInput, task taskman.ITask) error {
- return errors.Wrapf(cloudprovider.ErrNotImplemented, "RequestLiveMigrate")
- }
- func (drv *SBaseGuestDriver) RequestCancelLiveMigrate(ctx context.Context, guest *models.SGuest, userCred mcclient.TokenCredential) error {
- return errors.Wrapf(cloudprovider.ErrNotImplemented, "RequestCancelLiveMigrate")
- }
- func (drv *SVirtualizedGuestDriver) ValidateCreateData(ctx context.Context, userCred mcclient.TokenCredential, input *api.ServerCreateInput) (*api.ServerCreateInput, error) {
- return input, nil
- }
- func (drv *SBaseGuestDriver) ValidateUpdateData(ctx context.Context, guest *models.SGuest, userCred mcclient.TokenCredential, input api.ServerUpdateInput) (api.ServerUpdateInput, error) {
- return input, nil
- }
- func (drv *SBaseGuestDriver) RequestRemoteUpdate(ctx context.Context, guest *models.SGuest, userCred mcclient.TokenCredential, replaceTags bool) error {
- // nil ops
- return nil
- }
- func (drv *SBaseGuestDriver) ValidateRebuildRoot(ctx context.Context, userCred mcclient.TokenCredential, guest *models.SGuest, input *api.ServerRebuildRootInput) (*api.ServerRebuildRootInput, error) {
- return input, nil
- }
- func (drv *SBaseGuestDriver) ValidateDetachNetwork(ctx context.Context, userCred mcclient.TokenCredential, guest *models.SGuest) error {
- return nil
- }
- func (drv *SBaseGuestDriver) ValidateChangeDiskStorage(ctx context.Context, userCred mcclient.TokenCredential, guest *models.SGuest, targetStorageId string) error {
- return cloudprovider.ErrNotImplemented
- }
- func (drv *SBaseGuestDriver) StartChangeDiskStorageTask(guest *models.SGuest, ctx context.Context, userCred mcclient.TokenCredential, params *api.ServerChangeDiskStorageInternalInput, parentTaskId string) error {
- task, err := taskman.TaskManager.NewTask(ctx, "GuestChangeDiskStorageTask", guest, userCred, jsonutils.Marshal(params).(*jsonutils.JSONDict), parentTaskId, "", nil)
- if err != nil {
- return err
- }
- task.ScheduleRun(nil)
- return nil
- }
- func (drv *SBaseGuestDriver) RequestChangeDiskStorage(ctx context.Context, userCred mcclient.TokenCredential, guest *models.SGuest, input *api.ServerChangeDiskStorageInternalInput, task taskman.ITask) error {
- return cloudprovider.ErrNotImplemented
- }
- func (drv *SBaseGuestDriver) RequestSwitchToTargetStorageDisk(ctx context.Context, userCred mcclient.TokenCredential, guest *models.SGuest, input *api.ServerChangeDiskStorageInternalInput, task taskman.ITask) error {
- return cloudprovider.ErrNotImplemented
- }
- func (drv *SBaseGuestDriver) RequestSyncIsolatedDevice(ctx context.Context, guest *models.SGuest, task taskman.ITask) error {
- task.ScheduleRun(nil)
- return nil
- }
- func (drv *SBaseGuestDriver) RequestCPUSet(ctx context.Context, userCred mcclient.TokenCredential, host *models.SHost, guest *models.SGuest, input *api.ServerCPUSetInput) (*api.ServerCPUSetResp, error) {
- return nil, httperrors.ErrNotImplemented
- }
- func (drv *SBaseGuestDriver) RequestCPUSetRemove(ctx context.Context, userCred mcclient.TokenCredential, host *models.SHost, guest *models.SGuest, input *api.ServerCPUSetRemoveInput) error {
- return httperrors.ErrNotImplemented
- }
- func (drv *SBaseGuestDriver) QgaRequestGuestPing(ctx context.Context, header http.Header, host *models.SHost, guest *models.SGuest, async bool, input *api.ServerQgaTimeoutInput) error {
- return httperrors.ErrNotImplemented
- }
- func (drv *SBaseGuestDriver) QgaRequestSetUserPassword(ctx context.Context, task taskman.ITask, host *models.SHost, guest *models.SGuest, input *api.ServerQgaSetPasswordInput) error {
- return httperrors.ErrNotImplemented
- }
- func (self *SBaseGuestDriver) QgaRequestGuestInfoTask(ctx context.Context, userCred mcclient.TokenCredential, body jsonutils.JSONObject, host *models.SHost, guest *models.SGuest) (jsonutils.JSONObject, error) {
- return nil, httperrors.ErrNotImplemented
- }
- func (self *SBaseGuestDriver) QgaRequestSetNetwork(ctx context.Context, task taskman.ITask, body jsonutils.JSONObject, host *models.SHost, guest *models.SGuest) (jsonutils.JSONObject, error) {
- return nil, httperrors.ErrNotImplemented
- }
- func (self *SBaseGuestDriver) QgaRequestGetNetwork(ctx context.Context, userCred mcclient.TokenCredential, body jsonutils.JSONObject, host *models.SHost, guest *models.SGuest) (jsonutils.JSONObject, error) {
- return nil, httperrors.ErrNotImplemented
- }
- func (drv *SBaseGuestDriver) QgaRequestGetOsInfo(ctx context.Context, userCred mcclient.TokenCredential, body jsonutils.JSONObject, host *models.SHost, guest *models.SGuest) (jsonutils.JSONObject, error) {
- return nil, httperrors.ErrNotImplemented
- }
- func (drv *SBaseGuestDriver) RequestQgaCommand(ctx context.Context, userCred mcclient.TokenCredential, body jsonutils.JSONObject, host *models.SHost, guest *models.SGuest) (jsonutils.JSONObject, error) {
- return nil, httperrors.ErrNotImplemented
- }
- func (self *SBaseGuestDriver) RequestGuestScreenDump(ctx context.Context, userCred mcclient.TokenCredential, body jsonutils.JSONObject, host *models.SHost, guest *models.SGuest) (jsonutils.JSONObject, error) {
- return nil, httperrors.ErrNotImplemented
- }
- func (drv *SBaseGuestDriver) FetchMonitorUrl(ctx context.Context, guest *models.SGuest) string {
- s := auth.GetAdminSessionWithPublic(ctx, consts.GetRegion())
- tsdbURL, err := tsdb.GetDefaultServiceSourceURL(s, options.Options.MonitorEndpointType)
- if err != nil {
- log.Errorf("FetchMonitorUrl fail %s", err)
- return ""
- }
- return tsdbURL
- }
- func (drv *SBaseGuestDriver) RequestResetNicTrafficLimit(ctx context.Context, task taskman.ITask, host *models.SHost, guest *models.SGuest, input []api.ServerNicTrafficLimit) error {
- return httperrors.ErrNotImplemented
- }
- func (drv *SBaseGuestDriver) RequestSetNicTrafficLimit(ctx context.Context, task taskman.ITask, host *models.SHost, guest *models.SGuest, input []api.ServerNicTrafficLimit) error {
- return httperrors.ErrNotImplemented
- }
- func (drv *SBaseGuestDriver) SyncOsInfo(ctx context.Context, userCred mcclient.TokenCredential, g *models.SGuest, extVM cloudprovider.IOSInfo) error {
- return nil
- }
- func (self *SBaseGuestDriver) ValidateSetOSInfo(ctx context.Context, userCred mcclient.TokenCredential, _ *models.SGuest, _ *api.ServerSetOSInfoInput) error {
- return nil
- }
- func (self *SBaseGuestDriver) ValidateSyncOSInfo(ctx context.Context, userCred mcclient.TokenCredential, _ *models.SGuest) error {
- return httperrors.ErrNotImplemented
- }
- func (self *SBaseGuestDriver) RequestStartRescue(ctx context.Context, task taskman.ITask, body jsonutils.JSONObject, host *models.SHost, guest *models.SGuest) error {
- return httperrors.ErrNotImplemented
- }
- func (base *SBaseGuestDriver) ValidateGuestChangeConfigInput(ctx context.Context, guest *models.SGuest, input api.ServerChangeConfigInput) (*api.ServerChangeConfigSettings, error) {
- confs := api.ServerChangeConfigSettings{}
- confs.Old.InstanceType = guest.InstanceType
- confs.Old.VcpuCount = guest.VcpuCount
- confs.Old.CpuSockets = guest.CpuSockets
- confs.Old.VmemSize = guest.VmemSize
- confs.Old.ExtraCpuCount = guest.ExtraCpuCount
- region, err := guest.GetRegion()
- if err != nil {
- return nil, err
- }
- if len(input.InstanceType) > 0 {
- sku, err := models.ServerSkuManager.FetchSkuByNameAndProvider(input.InstanceType, region.Provider, true)
- if err != nil {
- return nil, errors.Wrap(err, "FetchSkuByNameAndProvider")
- }
- confs.InstanceTypeFamily = sku.InstanceTypeFamily
- confs.InstanceType = sku.GetName()
- confs.VcpuCount = sku.CpuCoreCount
- confs.VmemSize = sku.MemorySizeMB
- } else {
- if input.VcpuCount != nil {
- confs.VcpuCount = *input.VcpuCount
- } else {
- confs.VcpuCount = guest.VcpuCount
- }
- if input.ExtraCpuCount != nil {
- confs.ExtraCpuCount = *input.ExtraCpuCount
- }
- if len(input.VmemSize) > 0 {
- if !regutils.MatchSize(input.VmemSize) {
- return nil, httperrors.NewBadRequestError("Memory size %q must be number[+unit], like 256M, 1G or 256", input.VmemSize)
- }
- nVmem, err := fileutils.GetSizeMb(input.VmemSize, 'M', 1024)
- if err != nil {
- return nil, httperrors.NewBadRequestError("Params vmem_size parse error")
- }
- confs.VmemSize = nVmem
- } else {
- confs.VmemSize = guest.VmemSize
- }
- }
- disks, err := guest.GetGuestDisks()
- if err != nil {
- return nil, errors.Wrap(err, "GetGuestDisks")
- }
- var newDisks = make([]*api.DiskConfig, 0)
- var resizeDisks = make([]*api.DiskResizeSpec, 0)
- var schedInputDisks = make([]*api.DiskConfig, 0)
- // input.Disks start from index 1
- for i := range input.Disks {
- disk := input.Disks[i]
- if len(disk.SnapshotId) > 0 {
- snapObj, err := models.SnapshotManager.FetchById(disk.SnapshotId)
- if err != nil {
- return nil, httperrors.NewResourceNotFoundError("snapshot %s not found", disk.SnapshotId)
- }
- snap := snapObj.(*models.SSnapshot)
- disk.Storage = snap.StorageId
- }
- var guestDisk models.SGuestdisk
- if disk.Index >= len(disks) {
- // last disk
- guestDisk = disks[len(disks)-1]
- } else {
- guestDisk = disks[disk.Index]
- }
- diskObj := guestDisk.GetDisk()
- if diskObj == nil {
- return nil, errors.Wrapf(errors.ErrInvalidStatus, "fail to fetch disk at %d", disk.Index)
- }
- storage, err := diskObj.GetStorage()
- if err != nil {
- return nil, errors.Wrap(err, "GetStorage")
- }
- if len(disk.Backend) == 0 && len(disk.Storage) == 0 {
- disk.Backend = storage.StorageType
- disk.Storage = storage.Id
- }
- if disk.SizeMb > 0 {
- if disk.Index >= len(disks) {
- // new disk
- newDisks = append(newDisks, &disk)
- schedInputDisks = append(schedInputDisks, &disk)
- } else {
- // resize disk
- if disk.SizeMb < diskObj.DiskSize {
- return nil, httperrors.NewInputParameterError("Cannot reduce disk size for %dth disk", disk.Index)
- } else if disk.SizeMb > diskObj.DiskSize {
- resizeDisks = append(resizeDisks, &api.DiskResizeSpec{
- DiskId: diskObj.Id,
- SizeMb: disk.SizeMb,
- OldSizeMb: diskObj.DiskSize,
- })
- schedInputDisks = append(schedInputDisks, &api.DiskConfig{
- SizeMb: disk.SizeMb - diskObj.DiskSize,
- Index: disk.Index,
- Storage: storage.Id,
- })
- }
- }
- }
- }
- if len(resizeDisks) > 0 {
- confs.Resize = resizeDisks
- }
- if len(newDisks) > 0 {
- confs.Create = newDisks
- }
- if guest.Status != api.VM_RUNNING && input.AutoStart {
- confs.AutoStart = true
- }
- if guest.Status == api.VM_RUNNING {
- confs.GuestOnline = true
- }
- // schedulr forecast
- schedDesc := guest.ChangeConfToSchedDesc(confs.AddedCpu(), confs.AddedExtraCpu(), confs.AddedMem(), schedInputDisks)
- s := auth.GetAdminSession(ctx, options.Options.Region)
- canChangeConf, res, err := scheduler.SchedManager.DoScheduleForecast(s, schedDesc, 1)
- if err != nil {
- return nil, errors.Wrap(err, "SchedManager.DoScheduleForecast")
- }
- if !canChangeConf {
- return nil, httperrors.NewInsufficientResourceError("%s", res.String())
- }
- confs.SchedDesc = jsonutils.Marshal(schedDesc)
- return &confs, nil
- }
- func (base *SBaseGuestDriver) ValidateGuestHotChangeConfigInput(ctx context.Context, guest *models.SGuest, confs *api.ServerChangeConfigSettings) (*api.ServerChangeConfigSettings, error) {
- return confs, nil
- }
- func (base *SBaseGuestDriver) BeforeDetachIsolatedDevice(ctx context.Context, cred mcclient.TokenCredential, guest *models.SGuest, dev *models.SIsolatedDevice) error {
- return nil
- }
- func (base *SBaseGuestDriver) BeforeAttachIsolatedDevice(ctx context.Context, cred mcclient.TokenCredential, guest *models.SGuest, dev *models.SIsolatedDevice) error {
- return nil
- }
- func (base *SBaseGuestDriver) RequestUploadGuestStatus(ctx context.Context, guest *models.SGuest, task taskman.ITask) error {
- return errors.Wrapf(cloudprovider.ErrNotImplemented, "RequestUploadGuestStatus")
- }
- func (base *SBaseGuestDriver) CanStop(guest *models.SGuest) error {
- if utils.IsInStringArray(guest.Status, []string{api.VM_RUNNING, api.VM_STOP_FAILED, api.POD_STATUS_CRASH_LOOP_BACK_OFF, api.POD_STATUS_CONTAINER_EXITED, api.VM_KICKSTART_INSTALLING, api.VM_KICKSTART_FAILED, api.VM_KICKSTART_COMPLETED}) {
- return nil
- }
- return errors.Wrapf(errors.ErrInvalidStatus, "Cannot stop server in status %s", guest.Status)
- }
- func (base *SBaseGuestDriver) IsNeedCleanDisksAfterUndeploy() bool {
- return true
- }
|