Files
cloudpods/pkg/multicloud/openstack/disk.go
2021-05-28 20:18:10 +08:00

449 lines
12 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 openstack
import (
"context"
"fmt"
"net/url"
"strings"
"time"
"yunion.io/x/jsonutils"
"yunion.io/x/log"
"yunion.io/x/pkg/errors"
billing_api "yunion.io/x/onecloud/pkg/apis/billing"
api "yunion.io/x/onecloud/pkg/apis/compute"
"yunion.io/x/onecloud/pkg/cloudprovider"
"yunion.io/x/onecloud/pkg/multicloud"
"yunion.io/x/onecloud/pkg/util/version"
)
const (
DISK_STATUS_CREATING = "creating" // The volume is being created.
DISK_STATUS_ATTACHING = "attaching" // The volume is attaching to an instance.
DISK_STATUS_DETACHING = "detaching" // The volume is detaching from an instance.
DISK_STATUS_EXTENDING = "extending" // The volume is being extended.
DISK_STATUS_DELETING = "deleting" // The volume is being deleted.
DISK_STATUS_RETYPING = "retyping" // The volume is changing type to another volume type.
DISK_STATUS_AVAILABLE = "available" // The volume is ready to attach to an instance.
DISK_STATUS_RESERVED = "reserved" // The volume is reserved for attaching or shelved.
DISK_STATUS_IN_USE = "in-use" // The volume is attached to an instance.
DISK_STATUS_MAINTENANCE = "maintenance" // The volume is locked and being migrated.
DISK_STATUS_AWAITING_TRANSFER = "awaiting-transfer" // The volume is awaiting for transfer.
DISK_STATUS_BACKING_UP = "backing-up" // The volume is being backed up.
DISK_STATUS_RESTORING_BACKUP = "restoring-backup" // A backup is being restored to the volume.
DISK_STATUS_DOWNLOADING = "downloading" // The volume is downloading an image.
DISK_STATUS_UPLOADING = "uploading" // The volume is being uploaded to an image.
DISK_STATUS_ERROR = "error" // A volume creation error occurred.
DISK_STATUS_ERROR_DELETING = "error_deleting" // A volume deletion error occurred.
DISK_STATUS_ERROR_BACKING_UP = "error_backing-up" // A backup error occurred.
DISK_STATUS_ERROR_RESTORING = "error_restoring" // A backup restoration error occurred.
DISK_STATUS_ERROR_EXTENDING = "error_extending" // An error occurred while attempting to extend a volume.
)
type Attachment struct {
ServerId string
AttachmentId string
HostName string
VolumeId string
Device string
Id string
}
type Link struct {
Href string
Rel string
}
type VolumeImageMetadata struct {
Checksum string
MinRAM int
DiskFormat string
ImageName string
ImageId string
ContainerFormat string
MinDisk int
Size int
}
type SDisk struct {
storage *SStorage
multicloud.SDisk
multicloud.OpenStackTags
Id string
Name string
MigrationStatus string
Attachments []Attachment
Links []Link
AvailabilityZone string
Host string `json:"os-vol-host-attr:host"`
Encrypted bool
ReplicationStatus string
SnapshotId string
Size int
UserId string
TenantId string `json:"os-vol-tenant-attr:tenant_id"`
Migstat string `json:"os-vol-mig-status-attr:migstat"`
Status string
Description string
Multiattach string
SourceVolid string
ConsistencygroupId string
VolumeImageMetadata VolumeImageMetadata
NameId string `json:"os-vol-mig-status-attr:name_id"`
Bootable bool
CreatedAt time.Time
VolumeType string
}
func (region *SRegion) GetDisks() ([]SDisk, error) {
disks := []SDisk{}
resource := "/volumes/detail"
query := url.Values{}
query.Set("all_tenants", "true")
for {
resp, err := region.bsList(resource, query)
if err != nil {
return nil, errors.Wrap(err, "bsList")
}
part := struct {
Volumes []SDisk
VolumesLinks SNextLinks
}{}
err = resp.Unmarshal(&part)
if err != nil {
return nil, errors.Wrap(err, "resp.Unmarshal")
}
disks = append(disks, part.Volumes...)
marker := part.VolumesLinks.GetNextMark()
if len(marker) == 0 {
break
}
query.Set("marker", marker)
}
return disks, nil
}
func (disk *SDisk) GetId() string {
return disk.Id
}
func (disk *SDisk) Delete(ctx context.Context) error {
err := disk.storage.zone.region.DeleteDisk(disk.Id)
if err != nil {
return err
}
return cloudprovider.WaitDeleted(disk, 10*time.Second, 8*time.Minute)
}
func (disk *SDisk) attachInstances(attachments []Attachment) error {
for _, attachment := range attachments {
startTime := time.Now()
for time.Now().Sub(startTime) < 5*time.Minute {
if err := disk.storage.zone.region.AttachDisk(attachment.ServerId, disk.Id); err != nil {
if strings.Contains(err.Error(), "status must be available or downloading") {
time.Sleep(time.Second * 10)
continue
}
log.Errorf("recover attach disk %s => instance %s error: %v", disk.Id, attachment.ServerId, err)
return err
} else {
return nil
}
}
}
return nil
}
func (disk *SDisk) Resize(ctx context.Context, sizeMb int64) error {
maxVersion := ""
for _, service := range []string{OPENSTACK_SERVICE_VOLUMEV3, OPENSTACK_SERVICE_VOLUMEV2, OPENSTACK_SERVICE_VOLUME} {
maxVersion, _ = disk.storage.zone.region.GetMaxVersion(service)
if len(maxVersion) > 0 {
break
}
}
if version.GE(maxVersion, "3.42") {
return disk.storage.zone.region.ResizeDisk(disk.Id, sizeMb)
}
instanceIds := []string{}
for _, attachement := range disk.Attachments {
if err := disk.storage.zone.region.DetachDisk(attachement.ServerId, disk.Id); err != nil {
return err
}
instanceIds = append(instanceIds, attachement.ServerId)
}
err := disk.storage.zone.region.ResizeDisk(disk.Id, sizeMb)
if err != nil {
disk.attachInstances(disk.Attachments)
return err
}
return disk.attachInstances(disk.Attachments)
}
func (disk *SDisk) GetName() string {
if len(disk.Name) > 0 {
return disk.Name
}
return disk.Id
}
func (disk *SDisk) GetGlobalId() string {
return disk.Id
}
func (disk *SDisk) IsEmulated() bool {
return false
}
func (disk *SDisk) GetIStorage() (cloudprovider.ICloudStorage, error) {
return disk.storage, nil
}
func (disk *SDisk) GetStatus() string {
switch disk.Status {
case DISK_STATUS_CREATING, DISK_STATUS_DOWNLOADING:
return api.DISK_ALLOCATING
case DISK_STATUS_ATTACHING:
return api.DISK_ATTACHING
case DISK_STATUS_DETACHING:
return api.DISK_DETACHING
case DISK_STATUS_EXTENDING:
return api.DISK_RESIZING
case DISK_STATUS_RETYPING, DISK_STATUS_AVAILABLE, DISK_STATUS_RESERVED, DISK_STATUS_IN_USE, DISK_STATUS_MAINTENANCE, DISK_STATUS_AWAITING_TRANSFER, DISK_STATUS_BACKING_UP, DISK_STATUS_RESTORING_BACKUP, DISK_STATUS_UPLOADING:
return api.DISK_READY
case DISK_STATUS_DELETING:
return api.DISK_DEALLOC
case DISK_STATUS_ERROR:
return api.DISK_ALLOC_FAILED
default:
return api.DISK_UNKNOWN
}
}
func (disk *SDisk) Refresh() error {
_disk, err := disk.storage.zone.region.GetDisk(disk.Id)
if err != nil {
return err
}
return jsonutils.Update(disk, _disk)
}
func (disk *SDisk) ResizeDisk(sizeMb int64) error {
return disk.storage.zone.region.ResizeDisk(disk.Id, sizeMb)
}
func (disk *SDisk) GetDiskFormat() string {
return "lvm"
}
func (disk *SDisk) GetDiskSizeMB() int {
return disk.Size * 1024
}
func (disk *SDisk) GetIsAutoDelete() bool {
return false
}
func (disk *SDisk) GetTemplateId() string {
return disk.VolumeImageMetadata.ImageId
}
func (disk *SDisk) GetDiskType() string {
if disk.Bootable {
return api.DISK_TYPE_SYS
}
return api.DISK_TYPE_DATA
}
func (disk *SDisk) GetFsFormat() string {
return ""
}
func (disk *SDisk) GetIsNonPersistent() bool {
return false
}
func (disk *SDisk) GetDriver() string {
return "scsi"
}
func (disk *SDisk) GetCacheMode() string {
return "none"
}
func (disk *SDisk) GetMountpoint() string {
return ""
}
func (region *SRegion) CreateDisk(imageRef string, volumeType string, name string, sizeGb int, desc string, projectId string) (*SDisk, error) {
params := map[string]map[string]interface{}{
"volume": {
"size": sizeGb,
"volume_type": volumeType,
"name": name,
"description": desc,
},
}
if len(imageRef) > 0 {
params["volume"]["imageRef"] = imageRef
}
resp, err := region.bsCreate(projectId, "/volumes", params)
if err != nil {
return nil, errors.Wrap(err, "bsCreate")
}
disk := &SDisk{}
err = resp.Unmarshal(disk, "volume")
if err != nil {
return nil, errors.Wrap(err, "resp.Unmarshal")
}
//这里由于不好初始化disk的storage就手动循环了,如果是通过镜像创建,有个下载过程,比较慢,等待时间较长
startTime := time.Now()
timeout := time.Minute * 10
//若是通过镜像创建,需要先下载镜像,需要的时间更长
if len(imageRef) > 0 {
timeout = time.Minute * 30
}
for time.Now().Sub(startTime) < timeout {
disk, err = region.GetDisk(disk.GetGlobalId())
if err != nil {
return nil, errors.Wrapf(err, "GetDisk(%s)", disk.GetGlobalId())
}
log.Debugf("disk status %s expect %s", disk.GetStatus(), api.DISK_READY)
status := disk.GetStatus()
if status == api.DISK_READY {
break
}
if status == api.DISK_ALLOC_FAILED {
messages, _ := region.GetMessages(disk.Id)
if len(messages) > 0 {
return nil, fmt.Errorf("allocate disk %s failed, status is %s message: %s", disk.Name, disk.Status, messages[0].UserMessage)
}
return nil, fmt.Errorf("allocate disk %s failed, status is %s", disk.Name, disk.Status)
}
time.Sleep(time.Second * 10)
}
if disk.GetStatus() != api.DISK_READY {
return nil, fmt.Errorf("timeout for waitting disk ready, current status: %s", disk.Status)
}
return disk, nil
}
func (region *SRegion) GetDisk(diskId string) (*SDisk, error) {
resource := fmt.Sprintf("/volumes/%s", diskId)
resp, err := region.bsGet(resource)
if err != nil {
return nil, errors.Wrap(err, "bsGet")
}
disk := &SDisk{}
err = resp.Unmarshal(disk, "volume")
if err != nil {
return nil, errors.Wrap(err, "resp.Unmarshal")
}
return disk, nil
}
func (region *SRegion) DeleteDisk(diskId string) error {
resource := fmt.Sprintf("/volumes/%s", diskId)
_, err := region.bsDelete(resource)
return err
}
func (region *SRegion) ResizeDisk(diskId string, sizeMb int64) error {
params := map[string]map[string]interface{}{
"os-extend": {
"new_size": sizeMb / 1024,
},
}
resource := fmt.Sprintf("/volumes/%s/action", diskId)
_, err := region.bsPost(resource, params)
return err
}
func (region *SRegion) ResetDisk(diskId, snapshotId string) error {
params := map[string]map[string]interface{}{
"revert": {
"snapshot_id": snapshotId,
},
}
resource := fmt.Sprintf("/volumes/%s/action", diskId)
_, err := region.bsPost(resource, params)
return err
}
func (disk *SDisk) CreateISnapshot(ctx context.Context, name, desc string) (cloudprovider.ICloudSnapshot, error) {
snapshot, err := disk.storage.zone.region.CreateSnapshot(disk.Id, name, desc)
if err != nil {
return nil, err
}
return snapshot, cloudprovider.WaitStatus(snapshot, api.SNAPSHOT_READY, time.Second*5, time.Minute*5)
}
func (disk *SDisk) GetISnapshot(snapshotId string) (cloudprovider.ICloudSnapshot, error) {
return disk.storage.zone.region.GetISnapshotById(snapshotId)
}
func (disk *SDisk) GetISnapshots() ([]cloudprovider.ICloudSnapshot, error) {
snapshots, err := disk.storage.zone.region.GetSnapshots(disk.Id)
if err != nil {
return nil, errors.Wrapf(err, "GetSnapshots(%s)", disk.Id)
}
isnapshots := []cloudprovider.ICloudSnapshot{}
for i := range snapshots {
snapshots[i].region = disk.storage.zone.region
isnapshots = append(isnapshots, &snapshots[i])
}
return isnapshots, nil
}
func (disk *SDisk) Reset(ctx context.Context, snapshotId string) (string, error) {
return disk.Id, disk.storage.zone.region.ResetDisk(disk.Id, snapshotId)
}
func (disk *SDisk) GetBillingType() string {
return billing_api.BILLING_TYPE_POSTPAID
}
func (disk *SDisk) GetCreatedAt() time.Time {
return disk.CreatedAt
}
func (disk *SDisk) GetExpiredAt() time.Time {
return time.Time{}
}
func (disk *SDisk) GetAccessPath() string {
return ""
}
func (disk *SDisk) Rebuild(ctx context.Context) error {
return cloudprovider.ErrNotSupported
}
func (disk *SDisk) GetProjectId() string {
return disk.TenantId
}