add placeGroupID support for storageclass

Makefile

@@ -17,7 +17,7 @@
 .PHONY: all disk
 
 DISK_IMAGE_NAME=csiplugin/csi-qingcloud
-DISK_VERSION=v1.4.1-alpha
+DISK_VERSION=v1.4.1
 ROOT_PATH=$(pwd)
 PACKAGE_LIST=./cmd/... ./pkg/...
 

pkg/cloud/cloud_manager.go

@@ -39,7 +39,7 @@ type VolumeManager interface {
 	// Return:
 	//   volume id, nil: succeed to create volume and return volume id
 	//   nil, error: failed to create volume
-	CreateVolume(volName string, requestSize int, replicas int, volType int, zone string, containerConfID string, rg string) (volId string, err error)
+	CreateVolume(volName string, requestSize int, replicas int, volType int, zone string, containerConfID string, rg string, placeGroupID string) (volId string, err error)
 	// DeleteVolume deletes volume by id.
 	// Return:
 	//   nil: succeed to delete volume

pkg/cloud/mock/mock_cloud_manager.go

@@ -160,7 +160,7 @@ func (m *MockCloudManager) FindVolumeByName(volName string) (volInfo *qcservice.
 	return nil, nil
 }
 
-func (m *MockCloudManager) CreateVolume(volName string, requestSize int, replicas int, volType int, zone string, containerConfID string, rg string) (
+func (m *MockCloudManager) CreateVolume(volName string, requestSize int, replicas int, volType int, zone string, containerConfID string, rg string, placeGroupPD string) (
 	volId string, err error) {
 	exVol, err := m.FindVolumeByName(volName)
 	if err != nil {

pkg/cloud/qingcloud_manager.go

@@ -302,7 +302,7 @@ func (cm *qingCloudManager) FindVolumeByName(name string) (volume *qcservice.Vol
 // 1. format volume size
 // 2. create volume
 // 3. wait job
-func (qm *qingCloudManager) CreateVolume(volName string, requestSize int, replicas int, volType int, zone string, containerConfID string, rg string) (
+func (qm *qingCloudManager) CreateVolume(volName string, requestSize int, replicas int, volType int, zone string, containerConfID string, rg string, placeGroupID string) (
 	newVolId string, err error) {
 	// 0. Set CreateVolume args
 	// create volume count
@@ -326,9 +326,12 @@ func (qm *qingCloudManager) CreateVolume(volName string, requestSize int, replic
 	if (volType == driver.NeonSANVolumeType.Int() || volType == driver.NeonSANHDDVolumeType.Int() || volType == driver.NeonSANRDMAVolumeType.Int()) && rg != "" {
 		input.RG = &rg
 	}
+	if placeGroupID != "" {
+		input.PlaceGroupID = &placeGroupID
+	}
 
-	klog.Infof("Call IaaS CreateVolume request name: %s, size: %d GB, type: %d, zone: %s, count: %d, replica: %s, replica_count: %d, container_conf_id: %s, rg: %s",
-		*input.VolumeName, *input.Size, *input.VolumeType, *input.Zone, *input.Count, *input.Repl, *input.ReplicaCount, containerConfID, rg)
+	klog.Infof("Call IaaS CreateVolume request name: %s, size: %d GB, type: %d, zone: %s, count: %d, replica: %s, replica_count: %d, container_conf_id: %s, rg: %s, place_group_id: %s",
+		*input.VolumeName, *input.Size, *input.VolumeType, *input.Zone, *input.Count, *input.Repl, *input.ReplicaCount, containerConfID, rg, placeGroupID)
 
 	// 1. Create volume
 	output, err := qm.volumeService.CreateVolumes(input)

pkg/cloud/qingcloud_manager_test.go

@@ -67,7 +67,7 @@ func TestQingCloudManager_CreateVolume(t *testing.T) {
 	}
 
 	for _, test := range tests {
-		volId, err := cfg.CreateVolume(test.volName, test.volSize, test.volRepl, test.volType, test.volZone, "", "")
+		volId, err := cfg.CreateVolume(test.volName, test.volSize, test.volRepl, test.volType, test.volZone, "", "", "")
 		if err != nil {
 			if !test.isError {
 				t.Errorf("testcase %s: expect error %t, but actually error: %s", test.name, test.isError, err)

pkg/disk/driver/storageclass.go

@@ -36,6 +36,7 @@ const (
 	StorageClassTagsName        = "tags"
 	StorageClassContainerConfID = "containerConfID"
 	StorageClassRG              = "rg"
+	StorageClassPlaceGroupID    = "placeGroupID"
 )
 
 type QingStorageClass struct {
@@ -48,6 +49,7 @@ type QingStorageClass struct {
 	tags            []string
 	containerConfID string
 	rg              string
+	placeGroupID    string
 }
 
 // NewDefaultQingStorageClassFromType create default qingStorageClass by specified volume type
@@ -73,6 +75,7 @@ func NewQingStorageClassFromMap(opt map[string]string, topology *Topology) (*Qin
 	replica := -1
 	containerConfID := ""
 	rg := ""
+	placeGroupID := ""
 	var tags []string
 	for k, v := range opt {
 		switch strings.ToLower(k) {
@@ -123,6 +126,8 @@ func NewQingStorageClassFromMap(opt map[string]string, topology *Topology) (*Qin
 			containerConfID = v
 		case strings.ToLower(StorageClassRG):
 			rg = v
+		case strings.ToLower(StorageClassPlaceGroupID):
+			placeGroupID = v
 		}
 	}
 
@@ -155,6 +160,7 @@ func NewQingStorageClassFromMap(opt map[string]string, topology *Topology) (*Qin
 	sc.setTags(tags)
 	sc.setContainerConfID(containerConfID)
 	sc.setRG(rg)
+	sc.setPlaceGroupID(placeGroupID)
 	return sc, nil
 }
 
@@ -193,6 +199,10 @@ func (sc QingStorageClass) GetRG() string {
 	return sc.rg
 }
 
+func (sc QingStorageClass) GetPlaceGroupID() string {
+	return sc.placeGroupID
+}
+
 func (sc *QingStorageClass) setFsType(fs string) error {
 	if !IsValidFileSystemType(fs) {
 		return fmt.Errorf("unsupported filesystem type %s", fs)
@@ -233,6 +243,10 @@ func (sc *QingStorageClass) setRG(rg string) {
 	sc.rg = rg
 }
 
+func (sc *QingStorageClass) setPlaceGroupID(placeGroupID string) {
+	sc.placeGroupID = placeGroupID
+}
+
 // FormatVolumeSize transfer to proper volume size
 func (sc QingStorageClass) FormatVolumeSizeByte(sizeByte int64) int64 {
 	if sizeByte <= sc.GetMinSizeByte() {

pkg/disk/rpcserver/controllerserver.go

@@ -162,7 +162,7 @@ func (cs *ControllerServer) CreateVolume(ctx context.Context, req *csi.CreateVol
 		requiredSizeGib := common.ByteCeilToGib(requiredSizeByte)
 		klog.Infof("%s: Creating empty volume %s with %d Gib in zone %s...", hash, volName, requiredSizeGib,
 			top.GetZone())
-		newVolId, err := cs.cloud.CreateVolume(volName, requiredSizeGib, sc.GetReplica(), sc.GetDiskType().Int(), top.GetZone(), sc.GetContainerConfID(), sc.GetRG())
+		newVolId, err := cs.cloud.CreateVolume(volName, requiredSizeGib, sc.GetReplica(), sc.GetDiskType().Int(), top.GetZone(), sc.GetContainerConfID(), sc.GetRG(), sc.GetPlaceGroupID())
 		if err != nil {
 			klog.Errorf("%s: Failed to create volume %s, error: %v", hash, volName, err)
 			return nil, status.Error(codes.Internal, err.Error())