From 2bf30bd75e685cca240465118908f103b5808b42 Mon Sep 17 00:00:00 2001
From: LXW <1264174498@qq.com>
Date: Thu, 27 Jun 2024 11:25:27 +0800
Subject: [PATCH] (improvement)(Headless) Filter out SchemaElement mapped whose
 detect word length is 1 (#1237)

Co-authored-by: lxwcodemonkey
---
 .../supersonic/headless/chat/mapper/BaseMapper.java  | 12 ++++--------
 1 file changed, 4 insertions(+), 8 deletions(-)

diff --git a/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/mapper/BaseMapper.java b/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/mapper/BaseMapper.java
index 58f801a33..52a0fdb35 100644
--- a/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/mapper/BaseMapper.java
+++ b/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/mapper/BaseMapper.java
@@ -47,7 +47,7 @@ public abstract class BaseMapper implements SchemaMapper {
 
     private void filter(QueryContext queryContext) {
         filterByDataSetId(queryContext);
-        filterTermByDetectWordLen(queryContext);
+        filterByDetectWordLenLessThanOne(queryContext);
         switch (queryContext.getQueryDataType()) {
             case TAG:
                 filterByQueryDataType(queryContext, element -> !(element.getIsTag() > 0));
@@ -81,18 +81,14 @@ public abstract class BaseMapper implements SchemaMapper {
         }
     }
 
-    private static void filterTermByDetectWordLen(QueryContext queryContext) {
+    private static void filterByDetectWordLenLessThanOne(QueryContext queryContext) {
         Map<Long, List<SchemaElementMatch>> dataSetElementMatches =
                 queryContext.getMapInfo().getDataSetElementMatches();
         for (Map.Entry<Long, List<SchemaElementMatch>> entry : dataSetElementMatches.entrySet()) {
             List<SchemaElementMatch> value = entry.getValue();
             if (!CollectionUtils.isEmpty(value)) {
-                value.removeIf(schemaElementMatch -> {
-                    if (!SchemaElementType.TERM.equals(schemaElementMatch.getElement().getType())) {
-                        return false;
-                    }
-                    return StringUtils.length(schemaElementMatch.getDetectWord()) <= 1;
-                });
+                value.removeIf(schemaElementMatch ->
+                        StringUtils.length(schemaElementMatch.getDetectWord()) <= 1);
             }
         }
     }