Adding Filtering

Author: ms-alguerra

src/ResourceManager/Dns/Commands.Dns/Models/DnsClient.cs

@@ -126,9 +126,14 @@ public DnsZone GetDnsZone(string name, string resourceGroupName)
             };
         }
 
-        public List<DnsZone> ListDnsZones(string resourceGroupName)
+        public List<DnsZone> ListDnsZones(string resourceGroupName, string endsWith)
         {
-            ZoneListResponse getResponse = this.DnsManagementClient.Zones.List(resourceGroupName);
+            ZoneListParameters zoneListParameters = new ZoneListParameters
+            {
+                Filter = string.Format("endswith(Name,'{0}')", endsWith)
+            };
+
+            ZoneListResponse getResponse = this.DnsManagementClient.Zones.List(resourceGroupName, zoneListParameters);
             return getResponse.Zones.Select(zoneInResponse => new DnsZone
             {
                 Name = zoneInResponse.Name,
@@ -229,18 +234,28 @@ public DnsRecordSet GetDnsRecordSet(string name, string zoneName, string resourc
             return GetPowerShellRecordSet(zoneName, resourceGroupName, getResponse.RecordSet);
         }
 
-        public List<DnsRecordSet> ListRecordSets(string zoneName, string resourceGroupName, RecordType recordType)
+        public List<DnsRecordSet> ListRecordSets(string zoneName, string resourceGroupName, RecordType recordType, string endsWith)
         {
-            RecordListResponse listResponse = this.DnsManagementClient.Records.List(resourceGroupName, zoneName, recordType);
+            RecordListParameters recordListParameters = new RecordListParameters
+            {
+                Filter = string.Format("endswith(Name,'{0}')", endsWith)
+            };
+
+            RecordListResponse listResponse = this.DnsManagementClient.Records.List(resourceGroupName, zoneName, recordType, recordListParameters);
             return listResponse
                 .RecordSets
                 .Select(recordSetInResponse => GetPowerShellRecordSet(zoneName, resourceGroupName, recordSetInResponse))
                 .ToList();
         }
 
-        public List<DnsRecordSet> ListRecordSets(string zoneName, string resourceGroupName)
+        public List<DnsRecordSet> ListRecordSets(string zoneName, string resourceGroupName, string endsWith)
         {
-            RecordListResponse listResponse = this.DnsManagementClient.Records.ListAll(resourceGroupName, zoneName);
+            RecordListParameters recordListParameters = new RecordListParameters
+            {
+                Filter = string.Format("endswith(Name,'{0}')", endsWith)
+            };
+
+            RecordListResponse listResponse = this.DnsManagementClient.Records.ListAll(resourceGroupName, zoneName, recordListParameters);
             return listResponse
                 .RecordSets
                 .Select(recordSetInResponse => GetPowerShellRecordSet(zoneName, resourceGroupName, recordSetInResponse))

src/ResourceManager/Dns/Commands.Dns/Properties/Resources.Designer.cs

@@ -135,6 +135,9 @@
   <data name="Error_EtagNotSpecified" xml:space="preserve">
     <value>The ETag property of the {0} object is empty or "*". In order to perform this operation with optimistic concurrency checks, please set the Etag property (you may need to Get the {0} first). In order to perform the operation without optimistic concurrency checks, please specify the -IgnoreEtag switch. </value>
   </data>
+  <data name="Error_NameAndEndsWith" xml:space="preserve">
+    <value>Name parameter cannot be used with EndsWith.</value>
+  </data>
   <data name="Error_RemoveRecordTypeMismatch" xml:space="preserve">
     <value>Cannot remove a record of type {0} from a record set of type {1}. The types must match.</value>
   </data>

src/ResourceManager/Dns/Commands.Dns/Properties/Resources.resx

@@ -50,6 +50,10 @@ public class GetAzureDnsRecordSet : DnsBaseCmdlet
         [ValidateNotNullOrEmpty]
         public string RecordType { get; set; }
 
+        [Parameter(Mandatory = false, ValueFromPipelineByPropertyName = true, HelpMessage = "The relative name suffix to search.")]
+        [ValidateNotNullOrEmpty]
+        public string EndsWith { get; set; }
+
         public override void ExecuteCmdlet()
         {
             RecordType recordType = default(RecordType);
@@ -72,6 +76,11 @@ public override void ExecuteCmdlet()
                 resourceGroupName = this.Zone.ResourceGroupName;
             }
 
+            if (this.Name != null && this.EndsWith != null)
+            {
+                throw new PSArgumentException(ProjectResources.Error_NameAndEndsWith);
+            }
+
             if (this.Name != null)
             {
                 if (this.RecordType == null)
@@ -87,11 +96,11 @@ public override void ExecuteCmdlet()
                 List<DnsRecordSet> result = null;
                 if (this.RecordType == null)
                 {
-                    result = this.DnsClient.ListRecordSets(zoneName, resourceGroupName);
+                    result = this.DnsClient.ListRecordSets(zoneName, resourceGroupName, this.EndsWith);
                 }
                 else
                 {
-                    result = this.DnsClient.ListRecordSets(zoneName, resourceGroupName, recordType);
+                    result = this.DnsClient.ListRecordSets(zoneName, resourceGroupName, recordType, this.EndsWith);
                 }
 
                 this.WriteObject(result);

src/ResourceManager/Dns/Commands.Dns/Records/GetAzureDnsRescordSet.cs

@@ -16,6 +16,8 @@
 using System.Management.Automation;
 using Microsoft.Azure.Commands.Dns.Models;
 
+using ProjectResources = Microsoft.Azure.Commands.Dns.Properties.Resources;
+
 namespace Microsoft.Azure.Commands.Dns
 {
     /// <summary>
@@ -32,15 +34,24 @@ public class GetAzureDnsZone : DnsBaseCmdlet
         [ValidateNotNullOrEmpty]
         public string ResourceGroupName { get; set; }
 
+        [Parameter(Mandatory = false, ValueFromPipelineByPropertyName = true, HelpMessage = "The relative name suffix to search.")]
+        [ValidateNotNullOrEmpty]
+        public string EndsWith { get; set; }
+
         public override void ExecuteCmdlet()
         {
+            if (this.Name != null && this.EndsWith != null)
+            {
+                throw new PSArgumentException(ProjectResources.Error_NameAndEndsWith);
+            }
+
             if (this.Name != null)
             {
                 WriteObject(this.DnsClient.GetDnsZone(this.Name, this.ResourceGroupName));
             }
             else
             {
-                WriteObject(this.DnsClient.ListDnsZones(this.ResourceGroupName));
+                WriteObject(this.DnsClient.ListDnsZones(this.ResourceGroupName, this.EndsWith));
             }
         }
     }