fix: Remove the pkg/k8sutil/client.go and replace it with HAMi/pkg/ut…

…il/client in pkg/scheduler/scheduler.go Signed-off-by: Shouren Yang <yangshouren@gmail.com>
Project-HAMi · Dec 6, 2024 · 09448b9 · 09448b9
1 parent 94ba833
commit 09448b9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 52 deletions.
diff --git a/pkg/k8sutil/client.go b/pkg/k8sutil/client.go
diff --git a/pkg/scheduler/scheduler.go b/pkg/scheduler/scheduler.go
@@ -29,6 +29,7 @@ import (
 	"github.com/Project-HAMi/HAMi/pkg/scheduler/config"
 	"github.com/Project-HAMi/HAMi/pkg/scheduler/policy"
 	"github.com/Project-HAMi/HAMi/pkg/util"
+	"github.com/Project-HAMi/HAMi/pkg/util/client"
 
 	corev1 "k8s.io/api/core/v1"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
@@ -71,12 +72,6 @@ func NewScheduler() *Scheduler {
 	return s
 }
 
-func check(err error) {
-	if err != nil {
-		klog.Fatal(err)
-	}
-}
-
 func (s *Scheduler) onUpdateNode(_, newObj interface{}) {
 	s.nodeNotify <- struct{}{}
 }
@@ -125,9 +120,7 @@ func (s *Scheduler) onDelPod(obj interface{}) {
 }
 
 func (s *Scheduler) Start() {
-	kubeClient, err := k8sutil.NewClient()
-	check(err)
-	s.kubeClient = kubeClient
+	s.kubeClient = client.GetClient()
 	informerFactory := informers.NewSharedInformerFactoryWithOptions(s.kubeClient, time.Hour*1)
 	s.podLister = informerFactory.Core().V1().Pods().Lister()
 	s.nodeLister = informerFactory.Core().V1().Nodes().Lister()