Skip to content

Commit

Permalink
revert NewAzureLoadBalancerCommand
Browse files Browse the repository at this point in the history
  • Loading branch information
sergey-shandar committed Dec 5, 2017
1 parent 0bddd80 commit d016c0d
Showing 1 changed file with 3 additions and 50 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -13,12 +13,9 @@
// ----------------------------------------------------------------------------------

using AutoMapper;
using Microsoft.Azure.Commands.Common.Strategies;
using Microsoft.Azure.Commands.Common.Strategies.Network;
using Microsoft.Azure.Commands.Network.Models;
using Microsoft.Azure.Commands.ResourceManager.Common.ArgumentCompleters;
using Microsoft.Azure.Commands.ResourceManager.Common.Tags;
using Microsoft.Azure.Management.Internal.Resources.Models;
using Microsoft.Azure.Management.Network;
using System.Collections;
using System.Collections.Generic;
Expand All @@ -31,17 +28,12 @@ namespace Microsoft.Azure.Commands.Network
OutputType(typeof(PSLoadBalancer))]
public class NewAzureLoadBalancerCommand : LoadBalancerBaseCmdlet
{
public const string SimpleParameterSet = "SimpleParameterSet";

[Alias("ResourceName")]
[Parameter(
Mandatory = true,
ValueFromPipelineByPropertyName = true,
HelpMessage = "The resource name.")]
[ValidateNotNullOrEmpty]
[Parameter(
ParameterSetName = SimpleParameterSet,
Mandatory = true)]
public virtual string Name { get; set; }

[Parameter(
Expand All @@ -50,9 +42,6 @@ public class NewAzureLoadBalancerCommand : LoadBalancerBaseCmdlet
HelpMessage = "The resource group name.")]
[ResourceGroupCompleter]
[ValidateNotNullOrEmpty]
[Parameter(
ParameterSetName = SimpleParameterSet,
Mandatory = false)]
public virtual string ResourceGroupName { get; set; }

[Parameter(
Expand Down Expand Up @@ -134,26 +123,12 @@ public override void Execute()
Name,
() =>
{
PSLoadBalancer loadBalancer = null;
switch (ParameterSetName)
{
case SimpleParameterSet:
loadBalancer = SimpleParameterSetExecuteCmdlet();
break;
default:
loadBalancer = this.CreateLoadBalancer();
break;
}
var loadBalancer = this.CreateLoadBalancer();
WriteObject(loadBalancer);
},
() => present);
}
public PSLoadBalancer SimpleParameterSetExecuteCmdlet()
{
ResourceGroupName = ResourceGroupName ?? Name;
return null;
}

private PSLoadBalancer CreateLoadBalancer()
{
var loadBalancer = new PSLoadBalancer();
Expand Down Expand Up @@ -215,26 +190,4 @@ private PSLoadBalancer CreateLoadBalancer()
return getLoadBalancer;
}
}

public static class LoadBalancerExtensions
{
public static MNM.LoadBalancer ToLoadBalancer(this PSLoadBalancer psLb, Hashtable tags)
{
// Map to the sdk object
var lbModel = NetworkResourceManagerProfile.Mapper.Map<MNM.LoadBalancer>(psLb);
lbModel.Tags = TagsConversionHelper.CreateTagDictionary(tags, validate: true);

return lbModel;
}

public static PSLoadBalancer ToPsLoadBalancer(this MNM.LoadBalancer lb, string resourceGroupName)
{
var psLoadBalancer = NetworkResourceManagerProfile.Mapper.Map<PSLoadBalancer>(lb);
psLoadBalancer.ResourceGroupName = resourceGroupName;
psLoadBalancer.Tag =
TagsConversionHelper.CreateTagHashtable(lb.Tags);

return psLoadBalancer;
}
}
}
}

0 comments on commit d016c0d

Please sign in to comment.