TairSearch: add tft.explaincost

tair-opensource · Apr 10, 2023 · 93a4ada · 93a4ada
1 parent cad6f5b
commit 93a4ada
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 0 deletions.
diff --git a/tair/taircommands.go b/tair/taircommands.go
@@ -90,6 +90,7 @@ type TairCmdable interface {
 	TftScanDocIdArgs(ctx context.Context, index string, cursor string, a *TftScanArgs) *redis.SliceCmd
 	TftAnalyzer(ctx context.Context, analyzerName string, text string) *redis.StringCmd
 	TftAnalyzerWithArgs(ctx context.Context, analyzerName string, text string, a *TftAnalyzerArgs) *redis.StringCmd
+	TftExplaincost(ctx context.Context, index string, request string) *redis.StringCmd
 	TftAddSug(ctx context.Context, index string, textWeight map[string]int64) *redis.IntCmd
 	TftDelSug(ctx context.Context, index string, text ...string) *redis.IntCmd
 	TftSugSum(ctx context.Context, index string) *redis.IntCmd

diff --git a/tair/tairsearch.go b/tair/tairsearch.go
@@ -410,6 +410,16 @@ func (tc tairCmdable) TftAnalyzerWithArgs(ctx context.Context, analyzerName stri
 	return cmd
 }
 
+func (tc tairCmdable) TftExplaincost(ctx context.Context, index string, request string) *redis.StringCmd {
+	a := make([]interface{}, 3)
+	a[0] = "TFT.EXPLAINCOST"
+	a[1] = index
+	a[2] = request
+	cmd := redis.NewStringCmd(ctx, a...)
+	_ = tc(ctx, cmd)
+	return cmd
+}
+
 func (tc tairCmdable) TftAddSug(ctx context.Context, index string, textWeight map[string]int64) *redis.IntCmd {
 	args := make([]interface{}, 2)
 	args[0] = "TFT.ADDSUG"

diff --git a/tair/tairsearch_test.go b/tair/tairsearch_test.go
@@ -1,6 +1,7 @@
 package tair_test
 
 import (
+	"encoding/json"
 	"github.com/alibaba/tair-go/tair"
 	"github.com/go-redis/redis/v8"
 	"github.com/stretchr/testify/assert"
@@ -92,6 +93,12 @@ func (suite *TairSearchTestSuite) TestTftAddDoc() {
 	assert.NoError(suite.T(), err4)
 	assert.Equal(suite.T(), result4, "{\"tftkey\":{\"mappings\":{\"_source\":{\"enabled\":true,\"excludes\":[],\"includes\":[]},\"dynamic\":\"false\",\"properties\":{\"f0\":{\"boost\":1.0,\"enabled\":true,\"ignore_above\":-1,\"index\":true,\"similarity\":\"classic\",\"type\":\"text\"},\"f1\":{\"boost\":1.0,\"enabled\":true,\"ignore_above\":-1,\"index\":true,\"similarity\":\"classic\",\"type\":\"text\"}}}}}")
 
+	result5, err5 := suite.tairClient.TftExplaincost(ctx, "tftkey", "{\"query\":{\"match\":{\"f1\":\"3\"}}}").Result()
+	assert.NoError(suite.T(), err5)
+	var results map[string]interface{}
+	json.Unmarshal([]byte(result5), &results)
+	_, ok := results["QUERY_COST"]
+	assert.Equal(suite.T(), ok, true)
 }
 
 func (suite *TairSearchTestSuite) TestTftMSearch() {