Fix max eni limit for multicard

aws · Feb 3, 2024 · 7e98171 · 7e98171
1 parent 9a23696
commit 7e98171
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/pkg/vpc/vpc.go b/pkg/vpc/vpc.go
@@ -59,6 +59,13 @@ func GetENILimit(instanceType string) (int, error) {
 		log.Errorf("%s: %s", instanceType, ErrInstanceTypeNotExist)
 		return -1, ErrInstanceTypeNotExist
 	}
+	if len(instance.NetworkCards) > 1 {
+		eniLimit := 0
+		for _, networkCards := range instance.NetworkCards {
+			eniLimit = eniLimit + int(networkCards.MaximumNetworkInterfaces)
+		}
+		return eniLimit, nil
+	}
 	return instance.ENILimit, nil
 }
 

diff --git a/pkg/vpc/vpc_test.go b/pkg/vpc/vpc_test.go
@@ -30,6 +30,10 @@ func TestGetENILimit(t *testing.T) {
 	eniLimit, err = GetENILimit("a1.4xlarge")
 	assert.Equal(t, eniLimit, 8)
 	assert.NoError(t, err)
+
+	eniLimit, err = GetENILimit("p5.48xlarge")
+	assert.Equal(t, eniLimit, 64)
+	assert.NoError(t, err)
 }
 
 func TestGetIPv4Limit(t *testing.T) {