Skip to content

Commit 53bf48c

Browse files
committed
keyspace: address review comments for group removal
Signed-off-by: bufferflies <1045931706@qq.com>
1 parent 911a71a commit 53bf48c

File tree

3 files changed

+13
-1
lines changed

3 files changed

+13
-1
lines changed

pkg/keyspace/tso_keyspace_group.go

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -445,6 +445,12 @@ func (m *GroupManager) RemoveKeyspacesFromGroup(groupID uint32, keyspaceIDs []ui
445445
if kg == nil {
446446
return errs.ErrKeyspaceGroupNotExists.FastGenByArgs(groupID)
447447
}
448+
if kg.IsSplitting() {
449+
return errs.ErrKeyspaceGroupInSplit.FastGenByArgs(groupID)
450+
}
451+
if kg.IsMerging() {
452+
return errs.ErrKeyspaceGroupInMerging.FastGenByArgs(groupID)
453+
}
448454

449455
// Build a set of keyspaces to remove (excluding default keyspace)
450456
toRemove := make(map[uint32]struct{})

server/apiv2/handlers/tso_keyspace_group.go

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -579,7 +579,7 @@ func RemoveKeyspacesFromGroup(c *gin.Context) {
579579
// Parse request body
580580
var params RemoveKeyspacesFromGroupParams
581581
if err := c.BindJSON(&params); err != nil {
582-
c.AbortWithStatusJSON(http.StatusBadRequest, errs.ErrBindJSON.Wrap(err).GenWithStackByCause())
582+
c.AbortWithStatusJSON(http.StatusBadRequest, errs.ErrBindJSON.Wrap(err).GenWithStackByCause().Error())
583583
return
584584
}
585585

@@ -629,6 +629,10 @@ func RemoveKeyspacesFromGroup(c *gin.Context) {
629629
c.AbortWithStatusJSON(http.StatusInternalServerError, err.Error())
630630
return
631631
}
632+
if kg == nil {
633+
c.AbortWithStatusJSON(http.StatusInternalServerError, errs.ErrKeyspaceGroupNotExists.FastGenByArgs(groupID).Error())
634+
return
635+
}
632636
c.IndentedJSON(http.StatusOK, kg)
633637
return
634638
}

tests/server/apiv2/handlers/tso_keyspace_group_test.go

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -274,7 +274,9 @@ func (suite *keyspaceGroupTestSuite) TestKeyspaceGroupErrorMessage() {
274274
func (suite *keyspaceGroupTestSuite) TestRemoveKeyspacesFromGroup() {
275275
re := suite.Require()
276276
re.NoError(failpoint.Enable("github.com/tikv/pd/server/delayStartServerLoop", `return(true)`))
277+
defer re.NoError(failpoint.Disable("github.com/tikv/pd/server/delayStartServerLoop"))
277278
re.NoError(failpoint.Enable("github.com/tikv/pd/pkg/keyspace/skipSplitRegion", "return(true)"))
279+
defer re.NoError(failpoint.Disable("github.com/tikv/pd/pkg/keyspace/skipSplitRegion"))
278280

279281
keyspaceManager := suite.server.GetKeyspaceManager()
280282
re.NotNil(keyspaceManager)

0 commit comments

Comments
 (0)