Added a waitforattach verification at the end of ControllerPublishVolume

Author: davidz627

pkg/gce-cloud-provider/fake-gce.go

@@ -34,6 +34,8 @@ type FakeCloudProvider struct {
 	instances map[string]*compute.Instance
 }
 
+var _ GCECompute = &FakeCloudProvider{}
+
 func FakeCreateCloudProvider(project, zone string) (*FakeCloudProvider, error) {
 	return &FakeCloudProvider{
 		project:   project,
@@ -156,6 +158,11 @@ func (cloud *FakeCloudProvider) GetDiskTypeURI(zone, diskType string) string {
 	return ""
 }
 
+func (cloud *FakeCloudProvider) WaitForAttach(ctx context.Context, zone, diskName, instanceName string) error {
+	// TODO: Implement
+	return nil
+}
+
 // Instance Methods
 func (cloud *FakeCloudProvider) InsertInstance(instance *compute.Instance, instanceName string) {
 	cloud.instances[instanceName] = instance

pkg/gce-cloud-provider/gce-compute.go

@@ -41,6 +41,7 @@ type GCECompute interface {
 	DetachDisk(ctx context.Context, volumeZone, instanceName, volumeName string) (*compute.Operation, error)
 	GetDiskSourceURI(disk *compute.Disk, zone string) string
 	GetDiskTypeURI(zone, diskType string) string
+	WaitForAttach(ctx context.Context, zone, diskName, instanceName string) error
 	// Instance Methods
 	GetInstanceOrError(ctx context.Context, instanceZone, instanceName string) (*compute.Instance, error)
 	// Operation Methods
@@ -158,6 +159,23 @@ func (cloud *CloudProvider) WaitForOp(ctx context.Context, op *compute.Operation
 	})
 }
 
+func (cloud *CloudProvider) WaitForAttach(ctx context.Context, zone, diskName, instanceName string) error {
+	return wait.Poll(5*time.Second, 2*time.Minute, func() (bool, error) {
+		disk, err := cloud.GetDiskOrError(ctx, zone, diskName)
+		if err != nil {
+			glog.Errorf("GetDiskOrError failed to get disk: %v", err)
+			return false, err
+		}
+
+		for _, user := range disk.Users {
+			if strings.Contains(user, instanceName) && strings.Contains(user, zone) {
+				return true, nil
+			}
+		}
+		return false, nil
+	})
+}
+
 func opIsDone(op *compute.Operation) bool {
 	return op != nil && op.Status == "DONE"
 }

pkg/gce-cloud-provider/gce.go

@@ -44,6 +44,8 @@ type CloudProvider struct {
 	zone    string
 }
 
+var _ GCECompute = &CloudProvider{}
+
 func CreateCloudProvider() (*CloudProvider, error) {
 	svc, err := createCloudService()
 	if err != nil {

pkg/gce-pd-csi-driver/controller.go

@@ -296,6 +296,8 @@ func (gceCS *GCEControllerServer) ControllerPublishVolume(ctx context.Context, r
 		return nil, status.Error(codes.Internal, fmt.Sprintf("unknown Attach operation error: %v", err))
 	}
 
+	err = gceCS.CloudProvider.WaitForAttach(ctx, volumeZone, disk.Name, nodeID)
+
 	glog.Infof("Disk %v attached to instance %v successfully", disk.Name, nodeID)
 	return pubVolResp, nil
 }