35
35
using Microsoft . Azure . Commands . Common . Strategies . Network ;
36
36
using Microsoft . Azure . Commands . Compute . Common ;
37
37
using System . Net ;
38
+ using Microsoft . Azure . Commands . Compute . Strategies ;
39
+ using System . Threading . Tasks ;
38
40
39
41
namespace Microsoft . Azure . Commands . Compute . Automation
40
42
{
@@ -130,7 +132,7 @@ protected override void ProcessRecord()
130
132
switch ( ParameterSetName )
131
133
{
132
134
case SimpleParameterSet :
133
- SimpleParameterSetExecuteCmdlet ( ) ;
135
+ this . StartAndWait ( SimpleParameterSetExecuteCmdlet ) ;
134
136
break ;
135
137
default :
136
138
ExecuteClientAction ( ( ) =>
@@ -152,13 +154,9 @@ protected override void ProcessRecord()
152
154
}
153
155
}
154
156
155
- public void SimpleParameterSetExecuteCmdlet ( )
157
+ async Task SimpleParameterSetExecuteCmdlet ( IAsyncCmdlet asyncCmdlet )
156
158
{
157
159
ResourceGroupName = ResourceGroupName ?? VMScaleSetName ;
158
- InstanceCount = InstanceCount ?? 2 ;
159
- VmSku = VmSku ?? "Standard_DS2" ;
160
- UpgradePolicyMode = UpgradePolicyMode ?? UpgradeMode . Automatic ;
161
-
162
160
VirtualNetworkName = VirtualNetworkName ?? VMScaleSetName ;
163
161
SubnetName = SubnetName ?? VMScaleSetName ;
164
162
PublicIpAddressName = PublicIpAddressName ?? VMScaleSetName ;
@@ -225,16 +223,15 @@ public void SimpleParameterSetExecuteCmdlet()
225
223
226
224
if ( ShouldProcess ( VMScaleSetName , VerbsCommon . New ) )
227
225
{
228
- var result = virtualMachineScaleSet
226
+ var result = await virtualMachineScaleSet
229
227
. UpdateStateAsync (
230
228
client ,
231
229
target ,
232
230
new CancellationToken ( ) ,
233
- new ShouldProcessType ( this ) ,
234
- new ProgressReportType ( this ) )
235
- . GetAwaiter ( )
236
- . GetResult ( ) ;
237
- WriteObject ( result ) ;
231
+ new ShouldProcess ( asyncCmdlet ) ,
232
+ new ProgressReport ( asyncCmdlet ) ) ;
233
+
234
+ asyncCmdlet . WriteObject ( result ) ;
238
235
}
239
236
}
240
237
@@ -282,7 +279,7 @@ public void SimpleParameterSetExecuteCmdlet()
282
279
public PSCredential Credential { get ; set ; }
283
280
284
281
[ Parameter ( ParameterSetName = SimpleParameterSet , Mandatory = false ) ]
285
- public int ? InstanceCount { get ; set ; }
282
+ public int InstanceCount { get ; set ; } = 2 ;
286
283
287
284
[ Parameter ( ParameterSetName = SimpleParameterSet , Mandatory = false ) ]
288
285
public string VirtualNetworkName { get ; set ; }
@@ -308,12 +305,13 @@ public void SimpleParameterSetExecuteCmdlet()
308
305
[ Parameter ( ParameterSetName = SimpleParameterSet , Mandatory = false ) ]
309
306
[ LocationCompleter ]
310
307
public string Location { get ; set ; }
311
-
308
+
309
+ // this corresponds to VmSku in the Azure CLI
312
310
[ Parameter ( ParameterSetName = SimpleParameterSet , Mandatory = false ) ]
313
- public string VmSku { get ; set ; }
311
+ public string VmSize { get ; set ; } = "Standard_DS1_v2" ;
314
312
315
313
[ Parameter ( ParameterSetName = SimpleParameterSet , Mandatory = false ) ]
316
- public UpgradeMode ? UpgradePolicyMode { get ; set ; }
314
+ public UpgradeMode UpgradePolicyMode { get ; set ; }
317
315
318
316
[ Parameter ( ParameterSetName = SimpleParameterSet , Mandatory = false ) ]
319
317
[ ValidateSet ( "Static" , "Dynamic" ) ]
0 commit comments