size

huangpf · Nov 30, 2017 · 56a866a · 56a866a
1 parent 5d6cb34
commit 56a866a
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 6 deletions.
diff --git a/src/ResourceManager/Common/Commands.Common.Strategies/Compute/VirtualMachineStrategy.cs b/src/ResourceManager/Common/Commands.Common.Strategies/Compute/VirtualMachineStrategy.cs
@@ -35,9 +35,11 @@ public static ResourceConfig<VirtualMachine> CreateVirtualMachineConfig(
             this ResourceConfig<ResourceGroup> resourceGroup,
             string name,
             ResourceConfig<NetworkInterface> networkInterface,
+            bool isWindows,
             string adminUsername,
             string adminPassword,
-            Image image)
+            Image image,
+            string size)
             => Strategy.CreateConfig(
                 resourceGroup,
                 name, 
@@ -46,9 +48,8 @@ public static ResourceConfig<VirtualMachine> CreateVirtualMachineConfig(
                     OsProfile = new OSProfile
                     {
                         ComputerName = name,
-                        WindowsConfiguration = new WindowsConfiguration
-                        {
-                        },
+                        WindowsConfiguration = isWindows ? new WindowsConfiguration { } : null,
+                        LinuxConfiguration = isWindows ? null : new LinuxConfiguration(),
                         AdminUsername = adminUsername,
                         AdminPassword = adminPassword,
                     },
@@ -64,7 +65,7 @@ public static ResourceConfig<VirtualMachine> CreateVirtualMachineConfig(
                     },
                     HardwareProfile = new HardwareProfile
                     {
-                        VmSize = "Standard_DS1_v2"
+                        VmSize = size
                     },
                     StorageProfile = new StorageProfile
                     {

diff --git a/src/ResourceManager/Compute/Commands.Compute/VirtualMachine/Operation/NewAzureVMCommand.cs b/src/ResourceManager/Compute/Commands.Compute/VirtualMachine/Operation/NewAzureVMCommand.cs
@@ -156,6 +156,9 @@ public class NewAzureVMCommand : VirtualMachineBaseCmdlet
         [Parameter(ParameterSetName = StrategyParameterSet, Mandatory = false)]
         public string ImageName { get; set; } = "Win2016Datacenter";
 
+        [Parameter(ParameterSetName = StrategyParameterSet, Mandatory = false)]
+        public string Size { get; set; } = "Standard_DS1_v2";
+
         public override void ExecuteCmdlet()
         {
             switch (ParameterSetName)
@@ -206,9 +209,11 @@ async Task StrategyExecuteCmdletAsync(IAsyncCmdlet asyncCmdlet)
             var virtualMachine = resourceGroup.CreateVirtualMachineConfig(
                 name: Name,
                 networkInterface: networkInterface,
+                isWindows: isWindows,
                 adminUsername: Credential.UserName,
                 adminPassword: new NetworkCredential(string.Empty, Credential.Password).Password,
-                image: image.Image);
+                image: image.Image,
+                size: Size);
 
             var client = new Client(DefaultProfile.DefaultContext);