mirror of
https://github.com/yunionio/cloudpods.git
synced 2026-09-19 10:46:58 +08:00
265 lines
9.4 KiB
Go
265 lines
9.4 KiB
Go
// 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"
|
|
|
|
"yunion.io/x/pkg/utils"
|
|
|
|
api "yunion.io/x/onecloud/pkg/apis/compute"
|
|
"yunion.io/x/onecloud/pkg/cloudcommon/db/quotas"
|
|
"yunion.io/x/onecloud/pkg/cloudprovider"
|
|
"yunion.io/x/onecloud/pkg/compute/models"
|
|
"yunion.io/x/onecloud/pkg/httperrors"
|
|
"yunion.io/x/onecloud/pkg/mcclient"
|
|
"yunion.io/x/onecloud/pkg/util/billing"
|
|
"yunion.io/x/onecloud/pkg/util/rbacutils"
|
|
)
|
|
|
|
type SQcloudGuestDriver struct {
|
|
SManagedVirtualizedGuestDriver
|
|
}
|
|
|
|
func init() {
|
|
driver := SQcloudGuestDriver{}
|
|
models.RegisterGuestDriver(&driver)
|
|
}
|
|
|
|
func (self *SQcloudGuestDriver) GetHypervisor() string {
|
|
return api.HYPERVISOR_QCLOUD
|
|
}
|
|
|
|
func (self *SQcloudGuestDriver) GetProvider() string {
|
|
return api.CLOUD_PROVIDER_QCLOUD
|
|
}
|
|
|
|
func (self *SQcloudGuestDriver) GetComputeQuotaKeys(scope rbacutils.TRbacScope, ownerId mcclient.IIdentityProvider, brand string) models.SComputeResourceKeys {
|
|
keys := models.SComputeResourceKeys{}
|
|
keys.SBaseProjectQuotaKeys = quotas.OwnerIdProjectQuotaKeys(scope, ownerId)
|
|
keys.CloudEnv = api.CLOUD_ENV_PUBLIC_CLOUD
|
|
keys.Provider = api.CLOUD_PROVIDER_QCLOUD
|
|
keys.Brand = api.CLOUD_PROVIDER_QCLOUD
|
|
keys.Hypervisor = api.HYPERVISOR_QCLOUD
|
|
return keys
|
|
}
|
|
|
|
func (self *SQcloudGuestDriver) GetDefaultSysDiskBackend() string {
|
|
return api.STORAGE_CLOUD_PREMIUM
|
|
}
|
|
|
|
func (self *SQcloudGuestDriver) GetMinimalSysDiskSizeGb() int {
|
|
return 50
|
|
}
|
|
|
|
func (self *SQcloudGuestDriver) GetStorageTypes() []string {
|
|
return []string{
|
|
api.STORAGE_CLOUD_BASIC,
|
|
api.STORAGE_CLOUD_PREMIUM,
|
|
api.STORAGE_CLOUD_SSD,
|
|
api.STORAGE_LOCAL_BASIC,
|
|
api.STORAGE_LOCAL_SSD,
|
|
}
|
|
}
|
|
|
|
func (self *SQcloudGuestDriver) ChooseHostStorage(host *models.SHost, backend string, storageIds []string) *models.SStorage {
|
|
return self.chooseHostStorage(self, host, backend, storageIds)
|
|
}
|
|
|
|
func (self *SQcloudGuestDriver) GetDetachDiskStatus() ([]string, error) {
|
|
return []string{api.VM_READY, api.VM_RUNNING}, nil
|
|
}
|
|
|
|
func (self *SQcloudGuestDriver) GetAttachDiskStatus() ([]string, error) {
|
|
return []string{api.VM_READY, api.VM_RUNNING}, nil
|
|
}
|
|
|
|
func (self *SQcloudGuestDriver) GetRebuildRootStatus() ([]string, error) {
|
|
return []string{api.VM_READY, api.VM_RUNNING}, nil
|
|
}
|
|
|
|
func (self *SQcloudGuestDriver) GetChangeConfigStatus(guest *models.SGuest) ([]string, error) {
|
|
return []string{api.VM_READY, api.VM_RUNNING}, nil
|
|
}
|
|
|
|
func (self *SQcloudGuestDriver) GetDeployStatus() ([]string, error) {
|
|
return []string{api.VM_READY, api.VM_RUNNING}, nil
|
|
}
|
|
|
|
func (self *SQcloudGuestDriver) ValidateResizeDisk(guest *models.SGuest, disk *models.SDisk, storage *models.SStorage) error {
|
|
//https://cloud.tencent.com/document/product/362/5747
|
|
if !utils.IsInStringArray(guest.Status, []string{api.VM_READY, api.VM_RUNNING}) {
|
|
return fmt.Errorf("Cannot resize disk when guest in status %s", guest.Status)
|
|
}
|
|
if disk.DiskType == api.DISK_TYPE_SYS {
|
|
return fmt.Errorf("Cannot resize system disk")
|
|
}
|
|
if utils.IsInStringArray(storage.StorageType, []string{api.STORAGE_LOCAL_BASIC, api.STORAGE_LOCAL_SSD, api.STORAGE_LOCAL_PRO}) {
|
|
return fmt.Errorf("Cannot resize %s disk", storage.StorageType)
|
|
}
|
|
if disk.DiskSize/1024%10 > 0 {
|
|
return fmt.Errorf("Resize disk size must be an integer multiple of 10G")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (self *SQcloudGuestDriver) ValidateCreateData(ctx context.Context, userCred mcclient.TokenCredential, input *api.ServerCreateInput) (*api.ServerCreateInput, error) {
|
|
input, err := self.SManagedVirtualizedGuestDriver.ValidateCreateData(ctx, userCred, input)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if len(input.Networks) > 2 {
|
|
return nil, httperrors.NewInputParameterError("cannot support more than 1 nic")
|
|
}
|
|
|
|
sysDisk := input.Disks[0]
|
|
switch sysDisk.Backend {
|
|
case api.STORAGE_CLOUD_BASIC, api.STORAGE_CLOUD_SSD:
|
|
if sysDisk.SizeMb > 500*1024 {
|
|
return nil, fmt.Errorf("The %s system disk size must be less than 500GB", sysDisk.Backend)
|
|
}
|
|
case api.STORAGE_CLOUD_PREMIUM:
|
|
if sysDisk.SizeMb > 1024*1024 {
|
|
return nil, fmt.Errorf("The %s system disk size must be less than 1024GB", sysDisk.Backend)
|
|
}
|
|
case api.STORAGE_LOCAL_PRO:
|
|
return nil, fmt.Errorf("storage %s can not be system disk", sysDisk.Backend)
|
|
}
|
|
|
|
for i := 1; i < len(input.Disks); i++ {
|
|
disk := input.Disks[i]
|
|
switch disk.Backend {
|
|
case api.STORAGE_CLOUD_BASIC:
|
|
if disk.SizeMb < 10*1024 || disk.SizeMb > 16000*1024 {
|
|
return nil, httperrors.NewInputParameterError("The %s disk size must be in the range of 10GB ~ 16000GB", disk.Backend)
|
|
}
|
|
case api.STORAGE_CLOUD_PREMIUM:
|
|
if disk.SizeMb < 50*1024 || disk.SizeMb > 16000*1024 {
|
|
return nil, httperrors.NewInputParameterError("The %s disk size must be in the range of 50GB ~ 16000GB", disk.Backend)
|
|
}
|
|
case api.STORAGE_CLOUD_SSD:
|
|
if disk.SizeMb < 100*1024 || disk.SizeMb > 16000*1024 {
|
|
return nil, httperrors.NewInputParameterError("The %s disk size must be in the range of 100GB ~ 16000GB", disk.Backend)
|
|
}
|
|
case api.STORAGE_LOCAL_PRO:
|
|
return nil, httperrors.NewInputParameterError("storage %s can not be data disk")
|
|
}
|
|
if disk.SizeMb/1024%10 > 0 {
|
|
return nil, httperrors.NewInputParameterError("Data disk size must be an integer multiple of 10G")
|
|
}
|
|
}
|
|
return input, nil
|
|
}
|
|
|
|
func (self *SQcloudGuestDriver) ValidateChangeConfig(ctx context.Context, userCred mcclient.TokenCredential, guest *models.SGuest, cpuChanged bool, memChanged bool, newDisks []*api.DiskConfig) error {
|
|
if cpuChanged || memChanged {
|
|
disk, err := guest.GetSystemDisk()
|
|
if err != nil {
|
|
return httperrors.NewResourceNotFoundError("failed to found system disk error: %v", err)
|
|
}
|
|
storage := disk.GetStorage()
|
|
if storage == nil {
|
|
return httperrors.NewResourceNotFoundError("failed to found storage for disk %s(%s)", disk.Name, disk.Id)
|
|
}
|
|
// 腾讯云系统盘为本地存储，不支持调整配置
|
|
if utils.IsInStringArray(storage.StorageType, []string{api.STORAGE_LOCAL_BASIC, api.STORAGE_LOCAL_SSD, api.STORAGE_LOCAL_PRO}) {
|
|
return httperrors.NewUnsupportOperationError("The system disk is locally stored and does not support changing configuration")
|
|
}
|
|
}
|
|
|
|
for _, newDisk := range newDisks {
|
|
switch newDisk.Backend {
|
|
case api.STORAGE_CLOUD_BASIC:
|
|
if newDisk.SizeMb < 10*1024 || newDisk.SizeMb > 16000*1024 {
|
|
return httperrors.NewInputParameterError("The %s disk size must be in the range of 10GB ~ 16000GB", newDisk.Backend)
|
|
}
|
|
case api.STORAGE_CLOUD_PREMIUM:
|
|
if newDisk.SizeMb < 50*1024 || newDisk.SizeMb > 16000*1024 {
|
|
return httperrors.NewInputParameterError("The %s disk size must be in the range of 50GB ~ 16000GB", newDisk.Backend)
|
|
}
|
|
case api.STORAGE_CLOUD_SSD:
|
|
if newDisk.SizeMb < 100*1024 || newDisk.SizeMb > 16000*1024 {
|
|
return httperrors.NewInputParameterError("The %s disk size must be in the range of 100GB ~ 16000GB", newDisk.Backend)
|
|
}
|
|
case api.STORAGE_LOCAL_BASIC, api.STORAGE_LOCAL_SSD, api.STORAGE_LOCAL_PRO:
|
|
return httperrors.NewUnsupportOperationError("Not support create local storage disks")
|
|
case "": //这里Backend为空有可能会导致创建出来还是local storage,依然会出错,需要用户显式指定
|
|
return httperrors.NewInputParameterError("Please input new disk backend type")
|
|
}
|
|
if newDisk.SizeMb/1024%10 > 0 {
|
|
return httperrors.NewInputParameterError("Data disk size must be an integer multiple of 10G")
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (self *SQcloudGuestDriver) ValidateDetachDisk(ctx context.Context, userCred mcclient.TokenCredential, guest *models.SGuest, disk *models.SDisk) error {
|
|
storage := disk.GetStorage()
|
|
if storage == nil {
|
|
return httperrors.NewResourceNotFoundError("failed to found storage for disk %s(%s)", disk.Name, disk.Id)
|
|
}
|
|
// 腾讯云本地盘不支持卸载
|
|
if utils.IsInStringArray(storage.StorageType, []string{api.STORAGE_LOCAL_BASIC, api.STORAGE_LOCAL_SSD}) {
|
|
return httperrors.NewUnsupportOperationError("The disk is locally stored and does not support detach")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (self *SQcloudGuestDriver) GetGuestInitialStateAfterCreate() string {
|
|
return api.VM_RUNNING
|
|
}
|
|
|
|
func (self *SQcloudGuestDriver) GetGuestInitialStateAfterRebuild() string {
|
|
return api.VM_RUNNING
|
|
}
|
|
|
|
func (self *SQcloudGuestDriver) GetUserDataType() string {
|
|
return cloudprovider.CLOUD_SHELL
|
|
}
|
|
|
|
func (self *SQcloudGuestDriver) GetLinuxDefaultAccount(desc cloudprovider.SManagedVMCreateConfig) string {
|
|
userName := "root"
|
|
if desc.ImageType == "system" {
|
|
if desc.OsDistribution == "Ubuntu" {
|
|
userName = "ubuntu"
|
|
}
|
|
}
|
|
if desc.OsType == "Windows" {
|
|
userName = "Administrator"
|
|
}
|
|
return userName
|
|
}
|
|
|
|
func (self *SQcloudGuestDriver) AllowReconfigGuest() bool {
|
|
return true
|
|
}
|
|
|
|
func (self *SQcloudGuestDriver) IsSupportedBillingCycle(bc billing.SBillingCycle) bool {
|
|
months := bc.GetMonths()
|
|
if (months >= 1 && months <= 12) || (months == 24) || (months == 36) {
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (self *SQcloudGuestDriver) IsSupportPublicipToEip() bool {
|
|
return true
|
|
}
|
|
|
|
func (self *SQcloudGuestDriver) IsSupportSetAutoRenew() bool {
|
|
return true
|
|
}
|