From 6a053460feeffae9fdb995e82d6064f42e86b8c3 Mon Sep 17 00:00:00 2001
From: Patrick D'Addona <daddonapatrick@gmail.com>
Date: Thu, 14 May 2020 21:29:04 +0200
Subject: [PATCH] Enable azure_private_dns to work with non "AzurePublicCloud"
 clouds, like "AzureUSGovernmentCloud", "AzureChinaCloud" or
 "AzureGermanCloud"

---
 provider/azure/azure_private_dns.go | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/provider/azure/azure_private_dns.go b/provider/azure/azure_private_dns.go
index d07279d81..8205c56dc 100644
--- a/provider/azure/azure_private_dns.go
+++ b/provider/azure/azure_private_dns.go
@@ -65,9 +65,15 @@ func NewAzurePrivateDNSProvider(domainFilter endpoint.DomainFilter, zoneIDFilter
 		return nil, err
 	}
 
-	zonesClient := privatedns.NewPrivateZonesClient(subscriptionID)
+	settings, err := auth.GetSettingsFromEnvironment()
+	if err != nil {
+		return nil, err
+	}
+
+
+	zonesClient := privatedns.NewPrivateZonesClientWithBaseURI(settings.Environment.ResourceManagerEndpoint, subscriptionID)
 	zonesClient.Authorizer = authorizer
-	recordSetsClient := privatedns.NewRecordSetsClient(subscriptionID)
+	recordSetsClient := privatedns.NewRecordSetsClientWithBaseURI(settings.Environment.ResourceManagerEndpoint, subscriptionID)
 	recordSetsClient.Authorizer = authorizer
 
 	provider := &AzurePrivateDNSProvider{