pdctl: support show keyspace group primary

pkg/keyspace/tso_keyspace_group.go

@@ -1015,6 +1015,15 @@
 
 // GetKeyspaceGroupPrimaryByID returns the primary node of the keyspace group by ID.
 func (m *GroupManager) GetKeyspaceGroupPrimaryByID(id uint32) (string, error) {
+	// check if the keyspace group exists
+	kg, err := m.GetKeyspaceGroupByID(id)
+	if err != nil {
+		return "", err
+	}
+	if kg == nil {
+		return "", ErrKeyspaceGroupNotExists(id)
+	}
+
 	// default keyspace group: "/ms/{cluster_id}/tso/00000/primary".
 	// non-default keyspace group: "/ms/{cluster_id}/tso/keyspace_groups/election/{group}/primary".
 	path := fmt.Sprintf("/ms/%d/tso/00000/primary", m.clusterID)
@@ -1024,10 +1033,10 @@
 	leader := &tsopb.Participant{}
 	ok, _, err := etcdutil.GetProtoMsgWithModRev(m.client, path, leader)
 	if err != nil {
 		return "", err
 	}
 	if !ok {
 		return "", ErrKeyspaceGroupPrimaryNotFound
 	}
 	// The format of leader name is address-groupID.
 	contents := strings.Split(leader.GetName(), "-")

server/apiv2/handlers/tso_keyspace_group.go

@@ -47,7 +47,6 @@
 	router.DELETE("/:id/split", FinishSplitKeyspaceByID)
 	router.POST("/:id/merge", MergeKeyspaceGroups)
 	router.DELETE("/:id/merge", FinishMergeKeyspaceByID)
-	router.GET("/:id/primary", GetKeyspaceGroupPrimaryByID)
 }
 
 // CreateKeyspaceGroupParams defines the params for creating keyspace groups.
@@ -150,12 +149,22 @@
 		c.AbortWithStatusJSON(http.StatusInternalServerError, groupManagerUninitializedErr)
 		return
 	}
+
+	if c.Query("get_param") == "primary" {
+		primary, err := manager.GetKeyspaceGroupPrimaryByID(id)
+		if err != nil {
+			c.AbortWithStatusJSON(http.StatusInternalServerError, err.Error())
+			return
+		}
+		c.JSON(http.StatusOK, primary)
+		return
+	}
+
 	kg, err := manager.GetKeyspaceGroupByID(id)
 	if err != nil {
 		c.AbortWithStatusJSON(http.StatusInternalServerError, err.Error())
 		return
 	}
-
 	c.IndentedJSON(http.StatusOK, kg)
 }
 
@@ -486,34 +495,6 @@
 	c.JSON(http.StatusOK, nil)
 }
 
-// GetKeyspaceGroupPrimaryByID gets primary of keyspace group by ID.
-func GetKeyspaceGroupPrimaryByID(c *gin.Context) {
-	id, err := validateKeyspaceGroupID(c)
-	if err != nil {
-		c.AbortWithStatusJSON(http.StatusBadRequest, "invalid keyspace group id")
-		return
-	}
-	svr := c.MustGet(middlewares.ServerContextKey).(*server.Server)
-	manager := svr.GetKeyspaceGroupManager()
-	if manager == nil {
-		c.AbortWithStatusJSON(http.StatusInternalServerError, groupManagerUninitializedErr)
-		return
-	}
-	// check if keyspace group exists
-	kg, err := manager.GetKeyspaceGroupByID(id)
-	if err != nil || kg == nil {
-		c.AbortWithStatusJSON(http.StatusBadRequest, "keyspace group does not exist")
-		return
-	}
-	// get primary
-	primary, err := manager.GetKeyspaceGroupPrimaryByID(id)
-	if err != nil {
-		c.AbortWithStatusJSON(http.StatusInternalServerError, err.Error())
-		return
-	}
-	c.JSON(http.StatusOK, primary)
-}
-
 func validateKeyspaceGroupID(c *gin.Context) (uint32, error) {
 	id, err := strconv.ParseUint(c.Param("id"), 10, 64)
 	if err != nil {

tools/pd-ctl/pdctl/command/keyspace_group_command.go

@@ -349,18 +349,18 @@
 
 func showKeyspaceGroupPrimaryCommandFunc(cmd *cobra.Command, args []string) {
 	if len(args) < 1 {
 		cmd.Usage()
 		return
 	}
 	_, err := strconv.ParseUint(args[0], 10, 32)
 	if err != nil {
 		cmd.Printf("Failed to parse the keyspace group ID: %s\n", err)
 		return
 	}
-	r, err := doRequest(cmd, fmt.Sprintf("%s/%s/primary", keyspaceGroupsPrefix, args[0]), http.MethodGet, http.Header{})
+	r, err := doRequest(cmd, fmt.Sprintf("%s/%s?get_param=primary", keyspaceGroupsPrefix, args[0]), http.MethodGet, http.Header{})
 	if err != nil {
 		cmd.Printf("Failed to get the keyspace group primary information: %s\n", err)
 		return
 	}
 	cmd.Println(r)
 }