Skip to content

Commit

Permalink
Merge pull request #6997 from morshielt/node_info_prov_iface
Browse files Browse the repository at this point in the history
Change GCE AnnotationNodeInfoProvider nested provider to interface instead of concrete type
  • Loading branch information
k8s-ci-robot committed Jul 2, 2024
2 parents 965c84a + 4260701 commit 4d05113
Showing 1 changed file with 11 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -29,19 +29,26 @@ import (

// AnnotationNodeInfoProvider is a wrapper for MixedTemplateNodeInfoProvider.
type AnnotationNodeInfoProvider struct {
mixedTemplateNodeInfoProvider *MixedTemplateNodeInfoProvider
templateNodeInfoProvider TemplateNodeInfoProvider
}

// NewAnnotationNodeInfoProvider returns AnnotationNodeInfoProvider.
// NewAnnotationNodeInfoProvider returns AnnotationNodeInfoProvider wrapping MixedTemplateNodeInfoProvider.
func NewAnnotationNodeInfoProvider(t *time.Duration, forceDaemonSets bool) *AnnotationNodeInfoProvider {
return &AnnotationNodeInfoProvider{
mixedTemplateNodeInfoProvider: NewMixedTemplateNodeInfoProvider(t, forceDaemonSets),
templateNodeInfoProvider: NewMixedTemplateNodeInfoProvider(t, forceDaemonSets),
}
}

// NewCustomAnnotationNodeInfoProvider returns AnnotationNodeInfoProvider wrapping TemplateNodeInfoProvider.
func NewCustomAnnotationNodeInfoProvider(templateNodeInfoProvider TemplateNodeInfoProvider) *AnnotationNodeInfoProvider {
return &AnnotationNodeInfoProvider{
templateNodeInfoProvider: templateNodeInfoProvider,
}
}

// Process returns the nodeInfos set for this cluster.
func (p *AnnotationNodeInfoProvider) Process(ctx *context.AutoscalingContext, nodes []*apiv1.Node, daemonsets []*appsv1.DaemonSet, taintConfig taints.TaintConfig, currentTime time.Time) (map[string]*schedulerframework.NodeInfo, errors.AutoscalerError) {
nodeInfos, err := p.mixedTemplateNodeInfoProvider.Process(ctx, nodes, daemonsets, taintConfig, currentTime)
nodeInfos, err := p.templateNodeInfoProvider.Process(ctx, nodes, daemonsets, taintConfig, currentTime)
if err != nil {
return nil, err
}
Expand Down

0 comments on commit 4d05113

Please sign in to comment.