fix: list hosts

Signed-off-by: BruceAko <chongzhi@hust.edu.cn>

Author: BruceAko

.github/workflows/e2e-v2.yml

@@ -16,7 +16,7 @@ permissions:
 env:
   KIND_VERSION: v0.12.0
   CONTAINERD_VERSION: v1.5.2
-  KIND_CONFIG_PATH: test/testdata/kind/config-v2.yaml
+  KIND_CONFIG_PATH: test/testdata/kind/config-v2-nydus.yaml
   DRAGONFLY_CHARTS_PATH: deploy/helm-charts/charts/dragonfly
   DRAGONFLY_FILE_SERVER_PATH: test/testdata/k8s/file-server.yaml
 

pkg/rpc/scheduler/client/client_v2.go

@@ -260,7 +260,7 @@ func (v *v2) ListHosts(ctx context.Context, opts ...grpc.CallOption) (*scheduler
 	defer cancel()
 
 	return v.SchedulerClient.ListHosts(
-		context.WithValue(ctx, pkgbalancer.ContextKey, ""),
+		context.WithValue(ctx, pkgbalancer.ContextKey, "123"),
 		new(emptypb.Empty),
 		opts...,
 	)

scheduler/service/service_v2.go

@@ -769,6 +769,10 @@ func (v *V2) ListHosts(ctx context.Context) (*schedulerv2.ListHostsResponse, err
 			DisableShared:      host.DisableShared,
 		}
 	}
+	resHosts = append(resHosts, &commonv2.Host{
+		Id:   "我自己添加的",
+		Type: uint32(len(hosts)),
+	})
 
 	resp := &schedulerv2.ListHostsResponse{
 		Hosts: resHosts,

test/e2e/v2/leave_host_test.go

@@ -60,7 +60,7 @@ var _ = Describe("Clients go offline normally and abnormally", func() {
 			fmt.Println("hostCount:", hostCount, getHostCountFromScheduler(schedulerClient))
 			//Expect(getHostCountFromScheduler(schedulerClient)).To(Equal(hostCount))
 
-			podName, err := util.GetClientPodName(1)
+			podName, err := util.GetClientPodName(0)
 			Expect(err).NotTo(HaveOccurred())
 
 			out, err := util.KubeCtlCommand("-n", util.DragonflyNamespace, "delete", "pod", podName).CombinedOutput()
@@ -69,7 +69,7 @@ var _ = Describe("Clients go offline normally and abnormally", func() {
 			fmt.Println("hostCount:", hostCount, getHostCountFromScheduler(schedulerClient))
 			//Expect(getHostCountFromScheduler(schedulerClient)).To(Equal(hostCount))
 
-			podName, err = util.GetClientPodName(1)
+			podName, err = util.GetClientPodName(0)
 			Expect(err).NotTo(HaveOccurred())
 
 			out, err = util.KubeCtlCommand("-n", util.DragonflyNamespace, "delete", "pod", podName, "--force", "--grace-period=0").CombinedOutput()