Rainbond/api/util/volume.go

package util

import (
	"github.com/Sirupsen/logrus"
	"strings"

	"encoding/json"
	api_model "github.com/goodrain/rainbond/api/model"
	dbmodel "github.com/goodrain/rainbond/db/model"
	v1 "github.com/goodrain/rainbond/worker/appm/types/v1"
	"github.com/goodrain/rainbond/worker/server/pb"
	"github.com/twinj/uuid"
)

// SetVolumeDefaultValue set volume default value
func SetVolumeDefaultValue(info *dbmodel.TenantServiceVolume) {
	if info.VolumeName == "" {
		info.VolumeName = uuid.NewV4().String()
	}

	if info.AccessMode != "" {
		info.AccessMode = strings.ToUpper(info.AccessMode)
	} else {
		info.AccessMode = "RWO"
	}

	if info.SharePolicy == "" {
		info.SharePolicy = "exclusive"
	}

	if info.BackupPolicy == "" {
		info.BackupPolicy = "exclusive"
	}

	if info.ReclaimPolicy == "" {
		info.ReclaimPolicy = "retain"
	}
}

// ParseVolumeTypeOption parse volume Option name show and volume type
func ParseVolumeTypeOption(detail *pb.StorageClassDetail) string {
	volumeType := transferVolumeTypeOption(detail.Name, detail.Provisioner, detail.Parameters)
	if volumeType != nil {
		return volumeType.String()
	}

	return "unknown"
}

func transferVolumeTypeOption(name string, opts ...interface{}) *dbmodel.VolumeType {
	if name == v1.RainbondStatefuleShareStorageClass {
		return &dbmodel.ShareFileVolumeType
	}
	if name == v1.RainbondStatefuleLocalStorageClass {
		return &dbmodel.LocalVolumeType
	}
	vt := dbmodel.MakeNewVolume(name)
	return &vt
}

// opts[0]: kind is storageClass's provisioner
// opts[1]: parameters is storageClass's parameter
func transferCustomVolumeOptionName2Kind(opts ...interface{}) *dbmodel.VolumeType {
	if len(opts) != 2 {
		return nil
	}
	kind := opts[0].(string)
	if strings.HasSuffix(kind, "rbd") {
		if parameters, ok := opts[1].(map[string]string); ok {
			if parameters["adminId"] != "" && parameters["monitors"] != "" && parameters["pool"] != "" && parameters["userId"] != "" {
				return &dbmodel.CephRBDVolumeType
			}
		}
	}
	if kind == "alicloud/disk" {
		return &dbmodel.AliCloudVolumeType
	}
	return nil
}

// HackVolumeOptionDetailFromDB hack volumeOptionDetail from db
func HackVolumeOptionDetailFromDB(detail *api_model.VolumeTypeStruct, data *dbmodel.TenantServiceVolumeType) {
	if data != nil {
		detail.Description = data.Description
		detail.NameShow = data.NameShow
		if err := json.Unmarshal([]byte(data.CapacityValidation), &detail.CapacityValidation); err != nil {
			logrus.Warnf("unmarshal volumetype's capacityValidation error: %s, set capacityValidation to default", err.Error())
			detail.CapacityValidation = defaultcapacityValidation
		}
		detail.AccessMode = strings.Split(data.AccessMode, ",")
		detail.SharePolicy = strings.Split(data.SharePolicy, ",")
		detail.BackupPolicy = strings.Split(data.BackupPolicy, ",")
		detail.ReclaimPolicy = data.ReclaimPolicy
		detail.Sort = data.Sort
	}
}

var defaultcapacityValidation map[string]interface{}

func init() {
	capacityValidation := make(map[string]interface{})
	capacityValidation["min"] = 1
	capacityValidation["required"] = false
	capacityValidation["max"] = 999999999
}

// HackVolumeOptionDetail hack volume Option detail, like accessMode, sharePolicy, backupPolicy
func HackVolumeOptionDetail(volumeType string, detail *api_model.VolumeTypeStruct, more ...interface{}) {
	/*
		RWO - ReadWriteOnce
		ROX - ReadOnlyMany
		RWX - ReadWriteMany
	*/
	detail.AccessMode = append(detail.AccessMode, hackVolumeOptionAccessMode(volumeType)...)
	detail.SharePolicy = append(detail.SharePolicy, hackVolumeOptionSharePolicy(volumeType)...)
	detail.BackupPolicy = append(detail.BackupPolicy, hackVolumeOptionBackupPolicy(volumeType)...)
	detail.CapacityValidation = hackVolumeOptionCapacityValidation(volumeType)
	detail.Description = hackVolumeOptionDesc(volumeType)
	detail.NameShow = hackVolumeOptionNameShow(volumeType)
	if len(more) == 4 {
		detail.ReclaimPolicy = more[1].(string)
	}
}

func hackVolumeOptionNameShow(volumeType string) string {
	nameShow := volumeType
	if volumeType == "alicloud-disk-available" {
		nameShow = "阿里云盘（智能选择）"
	} else if volumeType == "alicloud-disk-common" {
		nameShow = "阿里云盘（基础）"
	} else if volumeType == "alicloud-disk-efficiency" {
		nameShow = "阿里云盘（高效）"
	} else if volumeType == "alicloud-disk-ssd" {
		nameShow = "阿里云盘（SSD）"
	}
	return nameShow
}

func hackVolumeOptionDesc(vt string) string {
	volumeType := dbmodel.VolumeType(vt)
	switch volumeType {
	case dbmodel.ShareFileVolumeType:
		return "default分布式文件存储，可租户内共享挂载，适用于所有类型应用"
	case dbmodel.LocalVolumeType:
		return "default本地存储设备，适用于有状态数据库服务"
	case dbmodel.MemoryFSVolumeType:
		return "default基于内存的存储设备，容量由内存量限制。应用重启数据即丢失，适用于高速暂存数据"
	default:
		return ""
	}
}

func hackVolumeOptionCapacityValidation(volumeType string) map[string]interface{} {
	data := make(map[string]interface{})
	data["required"] = true
	data["default"] = 1
	if strings.HasPrefix(volumeType, "alicloud-disk") {
		data["min"] = 20
		data["default"] = 20
		data["max"] = 32768 // [ali-cloud-disk usage limit](https://help.aliyun.com/document_detail/25412.html?spm=5176.2020520101.0.0.41d84df5faliP4)
	} else {
		data["min"] = 1
		data["max"] = 999999999
	}
	return data
}

/*

## volume accessMode
---

Volume Plugin 		| ReadWriteOnce        |    ReadOnlyMany          | ReadWriteMany
--------------------|----------------------|--------------------------|-----------------------
AWSElasticBlockStore| 	✓		           |   	-	      	          |  -
AzureFile			|    ✓			       |   	✓		  	          |  ✓
AzureDisk			|    ✓			       |   	-		  	          |  -
CephFS			    |    ✓		           |      ✓		          	  |  ✓
Cinder			    |    ✓		           |      -		  	          |  -
CSI					| depends on the driver|	depends on the driver |	depends on the driver
FC					|  ✓				   |   ✓					  | -
FlexVolume			| ✓					   |	✓					  | depends on the driver
Flocker				|	✓				   |  -						  | -
GCEPersistentDisk	|	✓				   | ✓						  | -
Glusterfs			|  ✓				   | ✓	                      | ✓
HostPath	        |  ✓				   | -						  | -
iSCSI				| ✓					   | ✓						  | -
Quobyte				| ✓					   | ✓						  | ✓
NFS					| ✓					   | ✓						  | ✓
RBD					| ✓					   | ✓						  | -
VsphereVolume		| ✓					   | -						  | - (works when Pods are collocated)
PortworxVolume		| ✓					   | -						  | ✓
ScaleIO				| ✓					   | ✓						  | -
StorageOS			| ✓					   | -						  | -

*/
func hackVolumeOptionAccessMode(vt string) []string {
	volumeType := dbmodel.VolumeType(vt)
	switch volumeType {
	case dbmodel.ShareFileVolumeType:
		return []string{"RWO", "ROX", "RWX"}
	case dbmodel.LocalVolumeType:
		return []string{"RWO", "ROX", "RWX"}
	case dbmodel.ConfigFileVolumeType:
		return []string{"ROX"}
	case dbmodel.MemoryFSVolumeType:
		return []string{"ROX"}
	default:
		return []string{"RWO"}
	}
}

// TODO finish volume share policy
func hackVolumeOptionSharePolicy(volumeType string) []string {
	return []string{"exclusive"}
}

// TODO finish vollume backup policy
func hackVolumeOptionBackupPolicy(volumeType string) []string {
	return []string{"exclusive"}
}