Skip to content

Commit

Permalink
Rename nodeNameToProviderID to more specific nodeNameToInstanceID
Browse files Browse the repository at this point in the history
  • Loading branch information
wongma7 committed Mar 18, 2022
1 parent 4049c9c commit 21e3cd0
Show file tree
Hide file tree
Showing 2 changed files with 12 additions and 12 deletions.
14 changes: 7 additions & 7 deletions pkg/providers/v1/aws.go
Original file line number Diff line number Diff line change
Expand Up @@ -1515,11 +1515,11 @@ func (c *Cloud) HasClusterID() bool {

// NodeAddresses is an implementation of Instances.NodeAddresses.
func (c *Cloud) NodeAddresses(ctx context.Context, name types.NodeName) ([]v1.NodeAddress, error) {
providerID, err := c.nodeNameToProviderID(name)
instanceID, err := c.nodeNameToInstanceID(name)
if err != nil {
return nil, fmt.Errorf("could not look up provider ID for node %q: %v", name, err)
return nil, fmt.Errorf("could not look up instance ID for node %q: %v", name, err)
}
return c.NodeAddressesByProviderID(ctx, string(providerID))
return c.NodeAddressesByProviderID(ctx, string(instanceID))
}

// extractIPv4NodeAddresses maps the instance information from EC2 to an array of NodeAddresses.
Expand Down Expand Up @@ -5014,10 +5014,10 @@ func (c *Cloud) findInstanceByNodeName(nodeName types.NodeName) (*ec2.Instance,
func (c *Cloud) getInstanceByNodeName(nodeName types.NodeName) (*ec2.Instance, error) {
var instance *ec2.Instance

// we leverage node cache to try to retrieve node's provider id first, as
// get instance by provider id is way more efficient than by filters in
// we leverage node cache to try to retrieve node's instance id first, as
// get instance by instance id is way more efficient than by filters in
// aws context
awsID, err := c.nodeNameToProviderID(nodeName)
awsID, err := c.nodeNameToInstanceID(nodeName)
if err != nil {
klog.V(3).Infof("Unable to convert node name %q to aws instanceID, fall back to findInstanceByNodeName: %v", nodeName, err)
instance, err = c.findInstanceByNodeName(nodeName)
Expand Down Expand Up @@ -5048,7 +5048,7 @@ func isFargateNode(nodeName string) bool {
return strings.HasPrefix(nodeName, fargateNodeNamePrefix)
}

func (c *Cloud) nodeNameToProviderID(nodeName types.NodeName) (InstanceID, error) {
func (c *Cloud) nodeNameToInstanceID(nodeName types.NodeName) (InstanceID, error) {
if strings.HasPrefix(string(nodeName), rbnNamePrefix) {
return InstanceID(nodeName), nil
}
Expand Down
10 changes: 5 additions & 5 deletions pkg/providers/v1/aws_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -2597,7 +2597,7 @@ func TestRegionIsValid(t *testing.T) {
assert.False(t, isRegionValid("pl-fake-991a", fake.metadata), "expected region 'pl-fake-991' to be invalid but it was not")
}

func TestNodeNameToProviderID(t *testing.T) {
func TestNodeNameToInstanceID(t *testing.T) {
testNodeName := types.NodeName("ip-10-0-0-1.ec2.internal")
testProviderID := "aws:///us-east-1c/i-02bce90670bb0c7cd"
fakeAWS := newMockedFakeAWSServices(TestClusterID)
Expand All @@ -2609,17 +2609,17 @@ func TestNodeNameToProviderID(t *testing.T) {
c.SetInformers(fakeInformerFactory)

// no node name
_, err = c.nodeNameToProviderID("")
_, err = c.nodeNameToInstanceID("")
assert.Error(t, err)

// informer has not synced
c.nodeInformerHasSynced = informerNotSynced
_, err = c.nodeNameToProviderID(testNodeName)
_, err = c.nodeNameToInstanceID(testNodeName)
assert.Error(t, err)

// informer has synced but node not found
c.nodeInformerHasSynced = informerSynced
_, err = c.nodeNameToProviderID(testNodeName)
_, err = c.nodeNameToInstanceID(testNodeName)
assert.Error(t, err)

// we are able to find the node in cache
Expand All @@ -2632,7 +2632,7 @@ func TestNodeNameToProviderID(t *testing.T) {
},
})
assert.NoError(t, err)
_, err = c.nodeNameToProviderID(testNodeName)
_, err = c.nodeNameToInstanceID(testNodeName)
assert.NoError(t, err)
}

Expand Down

0 comments on commit 21e3cd0

Please sign in to comment.