Merge pull request Azure#1139 from anuchandy/gen-compute

VirtualMachine annotation specific changes
jianghaolu · Sep 29, 2016 · 08f4761 · 08f4761
2 parents 90b936d + d5114a7
commit 08f4761
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 16 deletions.
diff --git a/azure-mgmt-compute/src/main/java/com/microsoft/azure/management/compute/VirtualMachine.java b/azure-mgmt-compute/src/main/java/com/microsoft/azure/management/compute/VirtualMachine.java
@@ -72,6 +72,7 @@ public interface VirtualMachine extends
      * 
      * @return the virtual machine sizes
      */
+    @Method
     PagedList<VirtualMachineSize> availableSizes();
 
     /**
@@ -530,13 +531,15 @@ interface WithWindowsCreate extends WithCreate {
              *
              * @return the stage representing creatable Windows VM definition
              */
+            @Method
             WithWindowsCreate disableVmAgent();
 
             /**
              * Specifies that automatic updates should be disabled.
              *
              * @return the stage representing creatable Windows VM definition
              */
+            @Method
             WithWindowsCreate disableAutoUpdate();
 
             /**

diff --git a/...-compute/src/main/java/com/microsoft/azure/management/compute/VirtualMachineDataDisk.java b/...-compute/src/main/java/com/microsoft/azure/management/compute/VirtualMachineDataDisk.java
@@ -270,35 +270,52 @@ interface UpdateDefinition<ParentT>  extends
      */
     interface UpdateStages {
         /**
-         * Specifies the new size in GB for data disk.
-         *
-         * @param sizeInGB the disk size in GB
-         * @return the next stage of data disk update
+         * The stage of the virtual machine data disk update allowing to set the disk size.
          */
-        Update withSizeInGB(Integer sizeInGB);
+        interface WithDiskSize {
+            /**
+             * Specifies the new size in GB for data disk.
+             *
+             * @param sizeInGB the disk size in GB
+             * @return the next stage of data disk update
+             */
+            Update withSizeInGB(Integer sizeInGB);
+        }
 
         /**
-         * Specifies the new logical unit number for the data disk.
-         *
-         * @param lun the logical unit number
-         * @return the next stage of data disk update
+         * The stage of the virtual machine data disk update allowing to set the disk lun.
          */
-        Update withLun(Integer lun);
+        interface WithDiskLun {
+            /**
+             * Specifies the new logical unit number for the data disk.
+             *
+             * @param lun the logical unit number
+             * @return the next stage of data disk update
+             */
+            Update withLun(Integer lun);
+        }
 
         /**
-         * Specifies the new caching type for the data disk.
-         *
-         * @param cachingType the disk caching type. Possible values include: 'None', 'ReadOnly', 'ReadWrite'
-         * @return the next stage of data disk update
+         * The stage of the virtual machine data disk update allowing to set the disk caching type.
          */
-        Update withCaching(CachingTypes cachingType);
+        interface WithDiskCaching {
+            /**
+             * Specifies the new caching type for the data disk.
+             *
+             * @param cachingType the disk caching type. Possible values include: 'None', 'ReadOnly', 'ReadWrite'
+             * @return the next stage of data disk update
+             */
+            Update withCaching(CachingTypes cachingType);
+        }
     }
 
     /**
      * The entirety of a data disk update as part of a virtual machine update.
      */
     interface Update extends
-            UpdateStages,
+            UpdateStages.WithDiskSize,
+            UpdateStages.WithDiskLun,
+            UpdateStages.WithDiskCaching,
             Settable<VirtualMachine.Update> {
     }
 }