api: supports GetRegion by hex key (#7160)

close #7159 api: supports GetRegion by hex key Signed-off-by: nolouch <[email protected]> Co-authored-by: ti-chi-bot[bot] <108142056+ti-chi-bot[bot]@users.noreply.github.com>
tikv · Sep 28, 2023 · 849d80d · 849d80d
1 parent 7a9e566
commit 849d80d
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/server/api/region.go b/server/api/region.go
@@ -277,6 +277,17 @@ func (h *regionHandler) GetRegion(w http.ResponseWriter, r *http.Request) {
 		h.rd.JSON(w, http.StatusBadRequest, err.Error())
 		return
 	}
+	// decode hex if query has params with hex format
+	formatStr := r.URL.Query().Get("format")
+	if formatStr == "hex" {
+		keyBytes, err := hex.DecodeString(key)
+		if err != nil {
+			h.rd.JSON(w, http.StatusBadRequest, err.Error())
+			return
+		}
+		key = string(keyBytes)
+	}
+
 	regionInfo := rc.GetRegionByKey([]byte(key))
 	h.rd.JSON(w, http.StatusOK, NewAPIRegionInfo(regionInfo))
 }

diff --git a/server/api/region_test.go b/server/api/region_test.go
@@ -147,6 +147,12 @@ func (suite *regionTestSuite) TestRegion() {
 	suite.NoError(tu.ReadGetJSON(re, testDialClient, url, r2))
 	r2.Adjust()
 	suite.Equal(NewAPIRegionInfo(r), r2)
+
+	url = fmt.Sprintf("%s/region/key/%s?format=hex", suite.urlPrefix, hex.EncodeToString([]byte("a")))
+	r2 = &RegionInfo{}
+	suite.NoError(tu.ReadGetJSON(re, testDialClient, url, r2))
+	r2.Adjust()
+	suite.Equal(NewAPIRegionInfo(r), r2)
 }
 
 func (suite *regionTestSuite) TestRegionCheck() {