Merge pull request vitessio#7426 from tinyspeck/sarabee-vtadmin-api-c…

…lusters Add /api/clusters endpoint to vtadmin-api
tinyspeck · Feb 1, 2021 · 0ffb8c1 · 0ffb8c1
2 parents 4c9a50b + e3088b4
commit 0ffb8c1
Show file tree

Hide file tree

Showing 5 changed files with 264 additions and 44 deletions.
diff --git a/go/vt/proto/vtadmin/vtadmin.pb.go b/go/vt/proto/vtadmin/vtadmin.pb.go
diff --git a/go/vt/vtadmin/api.go b/go/vt/vtadmin/api.go
@@ -77,6 +77,7 @@ func NewAPI(clusters []*cluster.Cluster, opts grpcserver.Options, httpOpts vtadm
 
 	httpAPI := vtadminhttp.NewAPI(api)
 
+	router.HandleFunc("/clusters", httpAPI.Adapt(vtadminhttp.GetClusters)).Name("API.GetClusters")
 	router.HandleFunc("/gates", httpAPI.Adapt(vtadminhttp.GetGates)).Name("API.GetGates")
 	router.HandleFunc("/keyspaces", httpAPI.Adapt(vtadminhttp.GetKeyspaces)).Name("API.GetKeyspaces")
 	router.HandleFunc("/tablets", httpAPI.Adapt(vtadminhttp.GetTablets)).Name("API.GetTablets")
@@ -113,6 +114,25 @@ func (api *API) ListenAndServe() error {
 	return api.serv.ListenAndServe()
 }
 
+// GetClusters is part of the vtadminpb.VTAdminServer interface.
+func (api *API) GetClusters(ctx context.Context, req *vtadminpb.GetClustersRequest) (*vtadminpb.GetClustersResponse, error) {
+	span, _ := trace.NewSpan(ctx, "API.GetClusters")
+	defer span.Finish()
+
+	vcs := make([]*vtadminpb.Cluster, 0, len(api.clusters))
+
+	for _, c := range api.clusters {
+		vcs = append(vcs, &vtadminpb.Cluster{
+			Id:   c.ID,
+			Name: c.Name,
+		})
+	}
+
+	return &vtadminpb.GetClustersResponse{
+		Clusters: vcs,
+	}, nil
+}
+
 // GetGates is part of the vtadminpb.VTAdminServer interface.
 func (api *API) GetGates(ctx context.Context, req *vtadminpb.GetGatesRequest) (*vtadminpb.GetGatesResponse, error) {
 	span, ctx := trace.NewSpan(ctx, "API.GetGates")