完全匹配用户昵称

search/views/user.py

@@ -106,6 +106,25 @@ def search_user(query="", offset=0, size=10):
 
     """
     try:
+        es_cli_obj = ESPerform.get_cli()
+        q = {}
+        # 先获取完全匹配的用户ID
+        sheer_user_id = []
+        q["query"] = {
+            "term": {
+                "nick_name_pre": query
+            }
+        }
+        # que = {"query": {"term": {"nick_name_pre": query}}}
+        result_dict = ESPerform.get_search_results(es_cli_obj, "user", q, 0, 10)
+        res = result_dict["hits"]
+
+        if len(res) > 0:
+            user_id = [item["_source"]["user_id"] for item in res]
+            sheer_user_id.append(user_id)
+
+        logging.info("get res:%s" % res)
+        # 获取模糊匹配的用户ID
 
         multi_match = {
             "fields": ["nick_name"],
@@ -138,16 +157,17 @@ def search_user(query="", offset=0, size=10):
         }
 
         logging.info("get q:%s" % q)
-        es_cli_obj = ESPerform.get_cli()
         result_dict = ESPerform.get_search_results(es_cli_obj, "user", q, offset, size)
 
         logging.info("get result_dict:%s" % result_dict)
 
+        search_user_id = []
         res = result_dict["hits"]
 
         if len(res) > 0:
-            user_id = [item["_source"]["user_id"] for item in res]
-        return {"search_user_id": user_id}
+            search_user_id = [item["_source"]["user_id"] for item in res]
+
+        return {"sheer_user_id": sheer_user_id, "search_user_id": search_user_id}
 
     except:
         logging.error("catch exception,err_msg:%s" % traceback.format_exc())

trans2es/utils/user_transfer.py

@@ -41,6 +41,7 @@ class UserTransfer(object):
             res["id"] = instance.id
             res["user_id"] = instance.user_id
             res["nick_name"] = instance.nick_name
+            res["nick_name_pre"] = instance.nick_name
             res["profile_pic"] = instance.profile_pic
             res["gender"] = instance.gender
             res["city_id"] = instance.city_id