mirror of
https://github.com/open-cluster-management-io/ocm.git
synced 2026-05-11 11:48:33 +00:00
241a3eaaa3
Some checks failed
Scorecard supply-chain security / Scorecard analysis (push) Failing after 1m1s
Post / coverage (push) Failing after 22m36s
Post / images (amd64) (push) Failing after 13m6s
Post / images (arm64) (push) Failing after 2m21s
Post / image manifest (push) Has been skipped
Post / trigger clusteradm e2e (push) Has been skipped
Close stale issues and PRs / stale (push) Successful in 13s
Signed-off-by: Jian Qiu <jqiu@redhat.com>
98 lines
3.1 KiB
Go
98 lines
3.1 KiB
Go
package hub
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/openshift/library-go/pkg/controller/controllercmd"
|
|
"k8s.io/apiserver/pkg/server/mux"
|
|
"k8s.io/client-go/kubernetes"
|
|
"k8s.io/utils/clock"
|
|
|
|
clusterclient "open-cluster-management.io/api/client/cluster/clientset/versioned"
|
|
clusterscheme "open-cluster-management.io/api/client/cluster/clientset/versioned/scheme"
|
|
clusterinformers "open-cluster-management.io/api/client/cluster/informers/externalversions"
|
|
|
|
"open-cluster-management.io/ocm/pkg/common/helpers"
|
|
"open-cluster-management.io/ocm/pkg/placement/controllers/metrics"
|
|
"open-cluster-management.io/ocm/pkg/placement/controllers/scheduling"
|
|
"open-cluster-management.io/ocm/pkg/placement/debugger"
|
|
)
|
|
|
|
// RunControllerManager starts the controllers on hub to make placement decisions.
|
|
func RunControllerManager(ctx context.Context, controllerContext *controllercmd.ControllerContext) error {
|
|
clusterClient, err := clusterclient.NewForConfig(controllerContext.KubeConfig)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
kubeClient, err := kubernetes.NewForConfig(controllerContext.KubeConfig)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
clusterInformers := clusterinformers.NewSharedInformerFactory(clusterClient, 10*time.Minute)
|
|
|
|
return RunControllerManagerWithInformers(ctx, controllerContext, kubeClient, clusterClient, clusterInformers)
|
|
}
|
|
|
|
func RunControllerManagerWithInformers(
|
|
ctx context.Context,
|
|
controllerContext *controllercmd.ControllerContext,
|
|
kubeClient kubernetes.Interface,
|
|
clusterClient clusterclient.Interface,
|
|
clusterInformers clusterinformers.SharedInformerFactory,
|
|
) error {
|
|
recorder, err := helpers.NewEventRecorder(ctx, clusterscheme.Scheme, kubeClient.EventsV1(), "placement-controller")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
metrics := metrics.NewScheduleMetrics(clock.RealClock{})
|
|
|
|
scheduler := scheduling.NewPluginScheduler(
|
|
scheduling.NewSchedulerHandler(
|
|
clusterClient,
|
|
clusterInformers.Cluster().V1beta1().PlacementDecisions().Lister(),
|
|
clusterInformers.Cluster().V1alpha1().AddOnPlacementScores().Lister(),
|
|
clusterInformers.Cluster().V1().ManagedClusters().Lister(),
|
|
recorder, metrics),
|
|
)
|
|
|
|
if controllerContext.Server != nil {
|
|
debug := debugger.NewDebugger(
|
|
scheduler,
|
|
clusterInformers.Cluster().V1beta1().Placements(),
|
|
clusterInformers.Cluster().V1().ManagedClusters(),
|
|
)
|
|
|
|
installDebugger(controllerContext.Server.Handler.NonGoRestfulMux, debug)
|
|
}
|
|
|
|
schedulingController := scheduling.NewSchedulingController(
|
|
ctx,
|
|
clusterClient,
|
|
clusterInformers.Cluster().V1().ManagedClusters(),
|
|
clusterInformers.Cluster().V1beta2().ManagedClusterSets(),
|
|
clusterInformers.Cluster().V1beta2().ManagedClusterSetBindings(),
|
|
clusterInformers.Cluster().V1beta1().Placements(),
|
|
clusterInformers.Cluster().V1beta1().PlacementDecisions(),
|
|
clusterInformers.Cluster().V1alpha1().AddOnPlacementScores(),
|
|
scheduler,
|
|
controllerContext.EventRecorder, recorder, metrics,
|
|
)
|
|
|
|
go clusterInformers.Start(ctx.Done())
|
|
|
|
go schedulingController.Run(ctx, 1)
|
|
|
|
<-ctx.Done()
|
|
|
|
return nil
|
|
}
|
|
|
|
func installDebugger(mux *mux.PathRecorderMux, d *debugger.Debugger) {
|
|
mux.HandlePrefix(debugger.DebugPath, http.HandlerFunc(d.Handler))
|
|
}
|