Specify DNS record types in Azure

1 files changed, 9 insertions, 6 deletions

diff --git a/node-repository/src/main/java/com/yahoo/vespa/hosted/provision/node/Dns.java b/node-repository/src/main/java/com/yahoo/vespa/hosted/provision/node/Dns.java
index 75ff0d29bd3..9f5aaf4b327 100644
--- a/node-repository/src/main/java/com/yahoo/vespa/hosted/provision/node/Dns.java
+++ b/node-repository/src/main/java/com/yahoo/vespa/hosted/provision/node/Dns.java
@@ -25,12 +25,7 @@ public class Dns {
 
     /** Returns the set of DNS record types for a host and its children and the given version (ipv6), host type, etc. */
     public static Set<RecordType> recordTypesFor(IP.Version ipVersion, NodeType hostType, CloudName cloudName, boolean enclave) {
-        if (cloudName == CloudName.AWS)
-            return enclave ?
-                   EnumSet.of(RecordType.FORWARD, RecordType.PUBLIC_FORWARD) :
-                   EnumSet.of(RecordType.FORWARD, RecordType.PUBLIC_FORWARD, RecordType.REVERSE);
-
-        if (cloudName == CloudName.GCP) {
+        if (cloudName == CloudName.AWS || cloudName == CloudName.GCP) {
             if (enclave) {
                 return ipVersion.is6() ?
                        EnumSet.of(RecordType.FORWARD, RecordType.PUBLIC_FORWARD) :
@@ -42,6 +37,14 @@ public class Dns {
             }
         }
 
+        if (cloudName == CloudName.AZURE) {
+            return ipVersion.is6() ?
+                    EnumSet.noneOf(RecordType.class) :
+                    enclave || hostType == confighost ?
+                            EnumSet.of(RecordType.FORWARD, RecordType.PUBLIC_FORWARD) :
+                            EnumSet.of(RecordType.FORWARD);
+        }
+
         throw new IllegalArgumentException("Does not manage DNS for cloud " + cloudName);
     }