Added 'disableHighAvailability' flag for resources

services/compute/compute.go

@@ -850,6 +850,8 @@ type GalleryImageProperties struct {
 	SourceType common.ImageSource `json:"sourceType,omitempty"`
 	// CloudInitDataSource - READ-ONLY; The cloud init data source to be used with the image. [NoCloud, Azure]. Default Value – NoCloud. For marketplace images it will be Azure.
 	CloudInitDataSource common.CloudInitDataSource `json:"cloudInitDataSource,omitempty"`
+	// Disable High Availability
+	DisableHighAvailability *bool `json:"disableHighAvailability,omitempty"`
 }
 
 // GalleryImage specifies information about the gallery Image Definition that you want to create or update.

services/compute/galleryimage/galleryimage.go

@@ -30,6 +30,17 @@ func getWssdGalleryImage(c *compute.GalleryImage, locationName, imagePath string
 		wssdgalleryimage.ContainerName = *c.GalleryImageProperties.ContainerName
 	}
 
+	if c.DisableHighAvailability != nil {
+		wssdgalleryimage.DisableHighAvailability = *c.DisableHighAvailability
+		// If HA is disabled, then throw error
+		if wssdgalleryimage.DisableHighAvailability {
+			return nil, errors.Wrapf(errors.NotSupported, "GalleryImage currently does not support Non-HighAvailability mode")
+		}
+	} else {
+		// If no value specified, set value as false
+		wssdgalleryimage.DisableHighAvailability = false
+	}
+
 	if c.GalleryImageProperties != nil {
 		wssdgalleryimage.SourceType = c.SourceType
 		wssdgalleryimage.CloudInitDataSource = c.GalleryImageProperties.CloudInitDataSource
@@ -56,6 +67,7 @@ func getGalleryImage(c *wssdcloudcompute.GalleryImage, location string) *compute
 			Statuses:         status.GetStatuses(c.GetStatus()),
 			ContainerName:    &c.ContainerName,
 			HyperVGeneration: c.HyperVGeneration,
+			DisableHighAvailability: &c.DisableHighAvailability,
 		},
 		Tags: tags.ProtoToMap(c.Tags),
 	}

services/storage/container/container.go

@@ -40,6 +40,14 @@ func getWssdContainer(c *storage.Container, locationName string) (*wssdcloudstor
 		}
 		wssdcontainer.Isolated = c.Isolated
 	}
+
+	if c.DisableHighAvailability != nil {
+		wssdcontainer.DisableHighAvailability = *c.DisableHighAvailability
+	} else {
+		// If HighAvailability mode isn't specified, proceed with default behavior i.e., HA
+		wssdcontainer.DisableHighAvailability = false
+	}
+
 	return wssdcontainer, nil
 }
 
@@ -71,6 +79,7 @@ func getContainer(c *wssdcloudstorage.Container, location string) *storage.Conta
 				AvailableSize: availSize,
 				TotalSize:     totalSize,
 			},
+			DisableHighAvailability: &c.DisableHighAvailability,
 		},
 		Version: &c.Status.Version.Number,
 		Tags:    tags.ProtoToMap(c.Tags),

services/storage/storage.go

@@ -40,6 +40,8 @@ type VirtualHardDiskProperties struct {
 	CloudInitDataSource common.CloudInitDataSource `json:"cloudInitDataSource,omitempty"`
 	// Container name
 	ContainerName *string `json:"containername,omitempty"`
+	// Disable High Availability
+	DisableHighAvailability *bool `json:"disableHighAvailability,omitempty"`
 }
 
 // VirtualHardDisk defines the structure of a VHD
@@ -72,6 +74,8 @@ type ContainerProperties struct {
 	// State - State
 	Statuses       map[string]*string `json:"statuses"`
 	*ContainerInfo `json:"info"`
+	// Disable High Availability
+	DisableHighAvailability *bool `json:"disableHighAvailability,omitempty"`
 }
 
 // Container defines the structure of a VHD

services/storage/virtualharddisk/virtualharddisk.go

@@ -58,6 +58,10 @@ func getWssdVirtualHardDisk(c *storage.VirtualHardDisk, groupName, containerName
 
 		wssdvhd.CloudInitDataSource = c.CloudInitDataSource
 	}
+
+	if c.DisableHighAvailability != nil {
+		wssdvhd.DisableHighAvailability = *c.DisableHighAvailability
+	}
 	return wssdvhd, nil
 }
 
@@ -82,6 +86,7 @@ func getVirtualHardDisk(c *wssdcloudstorage.VirtualHardDisk, group string) *stor
 			HyperVGeneration:    c.HyperVGeneration,
 			DiskFileFormat:      c.DiskFileFormat,
 			ContainerName:       &c.ContainerName,
+			DisableHighAvailability: &c.DisableHighAvailability,
 		},
 		Tags: tags.ProtoToMap(c.Tags),
 	}