Improves DnsUpdateClient.

alanedwardes · Feb 9, 2024 · 2d19de0 · 2d19de0
1 parent e33666f
commit 2d19de0
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/Ae.Dns.Client/DnsUpdateClient.cs b/src/Ae.Dns.Client/DnsUpdateClient.cs
@@ -33,6 +33,7 @@ public async Task<DnsMessage> Query(DnsMessage query, CancellationToken token =
             query.EnsureOperationCode(DnsOperationCode.UPDATE);
 
             var hostnames = query.Nameservers.Select(x => x.Host).ToArray();
+            var addresses = query.Nameservers.Select(x => x.Resource).OfType<DnsIpAddressResource>().Select(x => x.IPAddress).ToArray();
 
             void ChangeRecords(ICollection<DnsResourceRecord> records)
             {
@@ -41,6 +42,11 @@ void ChangeRecords(ICollection<DnsResourceRecord> records)
                     records.Remove(recordToRemove);
                 }
 
+                foreach (var recordToRemove in records.Where(x => x.Resource is DnsIpAddressResource ipr && addresses.Contains(ipr.IPAddress)).ToArray())
+                {
+                    records.Remove(recordToRemove);
+                }
+
                 foreach (var nameserver in query.Nameservers)
                 {
                     records.Add(nameserver);