handling v1beta1.NodeClass -> EC2NodeClass

aws · Sep 15, 2023 · 3754d8e · 3754d8e
1 parent d033501
commit 3754d8e
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/pkg/controllers/nodeclass/controller.go b/pkg/controllers/nodeclass/controller.go
@@ -160,7 +160,7 @@ func (c *Controller) resolveAMIs(ctx context.Context, nodeClass *v1beta1.EC2Node
 	return nil
 }
 
-func (c *Controller) resolveLicenses(ctx context.Context, nodeClass *v1beta1.NodeClass) error {
+func (c *Controller) resolveLicenses(ctx context.Context, nodeClass *v1beta1.EC2NodeClass) error {
 	licenses, err := c.licenseProvider.List(ctx, nodeClass)
 	if err != nil {
 		return err
@@ -171,7 +171,7 @@ func (c *Controller) resolveLicenses(ctx context.Context, nodeClass *v1beta1.Nod
 
 }
 
-func (c *Controller) resolveHostResourceGroups(ctx context.Context, nodeClass *v1beta1.NodeClass) error {
+func (c *Controller) resolveHostResourceGroups(ctx context.Context, nodeClass *v1beta1.EC2NodeClass) error {
     result , err :=  c.hostResourceGroupProvider.Get(ctx, nodeClass)
     if err != nil {
         return err

diff --git a/pkg/providers/hostresourcegroup/hostresourcegroup.go b/pkg/providers/hostresourcegroup/hostresourcegroup.go
@@ -30,7 +30,7 @@ func NewProvider(rgapi resourcegroupsiface.ResourceGroupsAPI, cache *cache.Cache
 	}
 }
 
-func (p *Provider) Get(ctx context.Context, nodeClass *v1beta1.NodeClass) (*v1beta1.HostResourceGroup, error) {
+func (p *Provider) Get(ctx context.Context, nodeClass *v1beta1.EC2NodeClass) (*v1beta1.HostResourceGroup, error) {
 	p.Lock()
 	defer p.Unlock()
     var nextToken *string = nil

diff --git a/pkg/providers/license/license.go b/pkg/providers/license/license.go
@@ -48,7 +48,7 @@ func NewProvider(lmapi licensemanageriface.LicenseManagerAPI, cache *cache.Cache
 	}
 }
 
-func (p *Provider) List(ctx context.Context, nodeClass *v1beta1.NodeClass) ([]string, error) {
+func (p *Provider) List(ctx context.Context, nodeClass *v1beta1.EC2NodeClass) ([]string, error) {
 	p.Lock()
 	defer p.Unlock()