From 655bf3f2355998ab37d381ff72013de3c2f4ca1f Mon Sep 17 00:00:00 2001
From: "lvyanqi.lyq" <lvyanqi.lyq@alibaba-inc.com>
Date: Fri, 17 Feb 2023 17:51:20 +0800
Subject: [PATCH] tairsearch support msearch

---
 tair/tairsearch.py       |  3 ++
 tests/test_tairsearch.py | 65 ++++++++++++++++++++++++++++++++++++++++
 2 files changed, 68 insertions(+)

diff --git a/tair/tairsearch.py b/tair/tairsearch.py
index f9dee7e..1216f97 100644
--- a/tair/tairsearch.py
+++ b/tair/tairsearch.py
@@ -143,6 +143,9 @@ def tft_delall(self, index: KeyT) -> ResponseT:
     def tft_search(self, index: KeyT, query: str) -> ResponseT:
         return self.execute_command("TFT.SEARCH", index, query)
 
+    def tft_msearch(self, index_count: int, index: Iterable[KeyT], query: str) -> ResponseT:
+        return self.execute_command("TFT.MSEARCH", index_count, *index, query)
+
     def tft_addsug(self, index: KeyT, mapping: Dict[str, int]) -> ResponseT:
         pieces: List[EncodableT] = [index]
 
diff --git a/tests/test_tairsearch.py b/tests/test_tairsearch.py
index d5bd20b..b8ee5ab 100644
--- a/tests/test_tairsearch.py
+++ b/tests/test_tairsearch.py
@@ -374,6 +374,71 @@ def test_tft_search(self, t: Tair):
         result = t.tft_search(index, '{"sort":[{"price":{"order":"desc"}}]}')
         assert json.loads(want) == json.loads(result)
 
+    def test_tft_msearch(self, t: Tair):
+        index1 = "idx_" + str(uuid.uuid4())
+        index2 = "idx_" + str(uuid.uuid4())
+        mappings = """
+{
+  "mappings": {
+    "_source": { "enabled": true },
+    "properties": {
+      "product_id": { "type": "keyword", "ignore_above": 128 },
+      "product_name": { "type": "text" },
+      "product_title": { "type": "text", "analyzer": "jieba" },
+      "price": { "type": "double" }
+    }
+  }
+}"""
+        document1 = '{"product_id":"test1"}'
+        document2 = '{"product_id":"test2"}'
+        document3 = '{"product_id":"test3"}'
+        document4 = '{"product_id":"test4"}'
+
+        assert t.tft_createindex(index1, mappings)
+        assert t.tft_createindex(index2, mappings)
+        assert t.tft_madddoc(
+            index1, {document1: "00001", document2: "00002"}
+        )
+        assert t.tft_madddoc(
+            index2, {document3: "00003", document4: "00004"}
+        )
+
+        want = f"""{{
+    "aux_info": {{"index_crc64": 5843875291690071373}},
+    "hits": {{
+        "hits": [
+          {{
+            "_id": "00001",
+            "_index": "{index1}",
+            "_score": 1.0,
+            "_source": {{ "product_id": "test1" }}
+          }},
+          {{
+            "_id": "00002",
+            "_index": "{index1}",
+            "_score": 1.0,
+            "_source": {{ "product_id": "test2" }}
+          }},
+          {{
+            "_id": "00003",
+            "_index": "{index2}",
+            "_score": 1.0,
+            "_source": {{ "product_id": "test3" }}
+          }},
+          {{
+            "_id": "00004",
+            "_index": "{index2}",
+            "_score": 1.0,
+            "_source": {{ "product_id": "test4" }}
+          }}
+        ],
+        "max_score": 1.0,
+        "total": {{ "relation": "eq", "value": 4 }}
+      }}
+    }}"""
+        result = t.tft_msearch(2, {index1, index2}, '{"sort":[{"price":{"order":"desc"}}]}')
+        assert json.loads(want) == json.loads(result)
+
     def test_tft_addsug(self, t: Tair):
         index = "idx_" + str(uuid.uuid4())