Add support for NS records in the Linode provider

kubernetes-sigs · Oct 13, 2023 · 30bf0d9 · 30bf0d9
1 parent d7cec32
commit 30bf0d9
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 4 deletions.
diff --git a/provider/linode/linode.go b/provider/linode/linode.go
@@ -241,8 +241,13 @@ func (p *LinodeProvider) submitChanges(ctx context.Context, changes LinodeChange
 	return nil
 }
 
-func getWeight() *int {
+func getWeight(recordType linodego.DomainRecordType) *int {
 	weight := 1
+
+	// NS records do not support having weight
+	if recordType == linodego.RecordTypeNS {
+		weight = 0
+	}
 	return &weight
 }
 
@@ -331,7 +336,7 @@ func (p *LinodeProvider) ApplyChanges(ctx context.Context, changes *plan.Changes
 						Target:   target,
 						Name:     getStrippedRecordName(zone, ep),
 						Type:     recordType,
-						Weight:   getWeight(),
+						Weight:   getWeight(recordType),
 						Port:     getPort(),
 						Priority: getPriority(),
 						TTLSec:   int(ep.RecordTTL),
@@ -395,7 +400,7 @@ func (p *LinodeProvider) ApplyChanges(ctx context.Context, changes *plan.Changes
 							Target:   target,
 							Name:     getStrippedRecordName(zone, ep),
 							Type:     recordType,
-							Weight:   getWeight(),
+							Weight:   getWeight(recordType),
 							Port:     getPort(),
 							Priority: getPriority(),
 							TTLSec:   int(ep.RecordTTL),
@@ -419,7 +424,7 @@ func (p *LinodeProvider) ApplyChanges(ctx context.Context, changes *plan.Changes
 							Target:   target,
 							Name:     getStrippedRecordName(zone, ep),
 							Type:     recordType,
-							Weight:   getWeight(),
+							Weight:   getWeight(recordType),
 							Port:     getPort(),
 							Priority: getPriority(),
 							TTLSec:   int(ep.RecordTTL),
@@ -515,6 +520,8 @@ func convertRecordType(recordType string) (linodego.DomainRecordType, error) {
 		return linodego.RecordTypeTXT, nil
 	case "SRV":
 		return linodego.RecordTypeSRV, nil
+	case "NS":
+		return linodego.RecordTypeNS, nil
 	default:
 		return "", fmt.Errorf("invalid Record Type: %s", recordType)
 	}

diff --git a/provider/linode/linode_test.go b/provider/linode/linode_test.go
@@ -135,6 +135,10 @@ func TestLinodeConvertRecordType(t *testing.T) {
 	require.NoError(t, err)
 	assert.Equal(t, linodego.RecordTypeSRV, record)
 
+	record, err = convertRecordType("NS")
+	require.NoError(t, err)
+	assert.Equal(t, linodego.RecordTypeNS, record)
+
 	_, err = convertRecordType("INVALID")
 	require.Error(t, err)
 }