mirror of
https://github.com/yunionio/cloudpods.git
synced 2026-06-24 18:28:02 +08:00
118 lines
3.5 KiB
Go
118 lines
3.5 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 storageman
|
|
|
|
import (
|
|
"context"
|
|
|
|
"yunion.io/x/jsonutils"
|
|
"yunion.io/x/pkg/errors"
|
|
|
|
"yunion.io/x/onecloud/pkg/cloudprovider"
|
|
"yunion.io/x/onecloud/pkg/hostman/hostutils"
|
|
"yunion.io/x/onecloud/pkg/httperrors"
|
|
"yunion.io/x/onecloud/pkg/multicloud/esxi"
|
|
"yunion.io/x/onecloud/pkg/multicloud/esxi/vcenter"
|
|
)
|
|
|
|
type SAgentDisk struct {
|
|
SLocalDisk
|
|
}
|
|
|
|
func NewAgentDisk(storage IStorage, id string) *SAgentDisk {
|
|
return &SAgentDisk{*NewLocalDisk(storage, id)}
|
|
}
|
|
|
|
type PrepareSaveToGlanceParams struct {
|
|
TaskId string
|
|
DiskInfo jsonutils.JSONObject
|
|
}
|
|
|
|
func (sd *SAgentDisk) PrepareSaveToGlance(ctx context.Context, params interface{}) (jsonutils.JSONObject, error) {
|
|
p, ok := params.(PrepareSaveToGlanceParams)
|
|
if !ok {
|
|
return nil, errors.Wrap(hostutils.ParamsError, "Resize params format error")
|
|
}
|
|
storage := sd.Storage.(*SAgentStorage)
|
|
return storage.PrepareSaveToGlance(ctx, p.TaskId, p.DiskInfo)
|
|
}
|
|
|
|
func (sd *SAgentDisk) ReSize(ctx context.Context, diskInfo interface{}) (jsonutils.JSONObject, error) {
|
|
body, ok := diskInfo.(*jsonutils.JSONDict)
|
|
if !ok {
|
|
return nil, errors.Wrap(hostutils.ParamsError, "PrepareSaveToGlance params format error")
|
|
}
|
|
// check parameters
|
|
params := []string{"size", "host_info", "vm_private_id", "disk_private_id"}
|
|
for _, param := range params {
|
|
if !body.Contains(param) {
|
|
return nil, httperrors.NewMissingParameterError(param)
|
|
}
|
|
}
|
|
|
|
type sResize struct {
|
|
SizeMb int64 `json:"size"`
|
|
HostInfo vcenter.SVCenterAccessInfo
|
|
VMId string `json:"vm_private_id"`
|
|
DiskId string `json:"disk_private_id"`
|
|
}
|
|
resize := sResize{}
|
|
err := body.Unmarshal(&resize)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "%s: unmarshal to sResize", hostutils.ParamsError)
|
|
}
|
|
|
|
esxiClient, err := esxi.NewESXiClientFromAccessInfo(ctx, &resize.HostInfo)
|
|
if err != nil {
|
|
return nil, httperrors.NewInputParameterError("info of host_info error")
|
|
}
|
|
host, err := esxiClient.FindHostByIp(resize.HostInfo.PrivateId)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "fail to find host by ip %s", resize.HostInfo.PrivateId)
|
|
}
|
|
ivm, err := host.GetIVMById(resize.VMId)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "fail to find vm by ID %s", resize.VMId)
|
|
}
|
|
idisks, err := ivm.GetIDisks()
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "fail to get idisks of vm %s", resize.VMId)
|
|
}
|
|
var (
|
|
idisk cloudprovider.ICloudDisk
|
|
hasDisk bool
|
|
)
|
|
for i := range idisks {
|
|
if idisks[i].GetId() == resize.DiskId {
|
|
idisk = idisks[i]
|
|
hasDisk = true
|
|
}
|
|
}
|
|
if !hasDisk {
|
|
return nil, errors.Wrapf(err, "no such disk %s", resize.DiskId)
|
|
}
|
|
url := idisk.GetAccessPath()
|
|
vm, disk := ivm.(*esxi.SVirtualMachine), idisk.(*esxi.SVirtualDisk)
|
|
online := jsonutils.QueryBoolean(body, "online", false)
|
|
if online {
|
|
esxiClient.DoExtendDiskOnline(vm, disk, resize.SizeMb)
|
|
} else {
|
|
esxiClient.ExtendDisk(url, resize.SizeMb)
|
|
}
|
|
desc := jsonutils.NewDict()
|
|
desc.Add(jsonutils.NewInt(resize.SizeMb), "disk_size")
|
|
return desc, nil
|
|
}
|