mirror of
https://github.com/open-cluster-management-io/ocm.git
synced 2026-05-25 10:32:59 +00:00
52 lines
1.7 KiB
Go
52 lines
1.7 KiB
Go
package hub
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/openshift/library-go/pkg/controller/controllercmd"
|
|
"k8s.io/client-go/kubernetes"
|
|
"k8s.io/client-go/tools/events"
|
|
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"
|
|
scheduling "open-cluster-management.io/placement/pkg/controllers/scheduling"
|
|
)
|
|
|
|
// 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)
|
|
|
|
broadcaster := events.NewBroadcaster(&events.EventSinkImpl{Interface: kubeClient.EventsV1()})
|
|
|
|
broadcaster.StartRecordingToSink(ctx.Done())
|
|
|
|
schedulingController := scheduling.NewSchedulingController(
|
|
clusterClient,
|
|
clusterInformers.Cluster().V1().ManagedClusters(),
|
|
clusterInformers.Cluster().V1alpha1().ManagedClusterSets(),
|
|
clusterInformers.Cluster().V1alpha1().ManagedClusterSetBindings(),
|
|
clusterInformers.Cluster().V1alpha1().Placements(),
|
|
clusterInformers.Cluster().V1alpha1().PlacementDecisions(),
|
|
controllerContext.EventRecorder,
|
|
broadcaster.NewRecorder(clusterscheme.Scheme, "placementController"),
|
|
)
|
|
|
|
go clusterInformers.Start(ctx.Done())
|
|
|
|
go schedulingController.Run(ctx, 1)
|
|
|
|
<-ctx.Done()
|
|
return nil
|
|
}
|