Context Extraktion für TextId zu Sentence Suche

thesaurus-linguae-aegyptiae · Aug 5, 2023 · e32140b · e32140b
1 parent b77446d
commit e32140b
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/src/main/java/tla/backend/es/query/SentenceSearchQueryBuilder.java b/src/main/java/tla/backend/es/query/SentenceSearchQueryBuilder.java
@@ -11,6 +11,7 @@
 import lombok.Getter;
 import lombok.extern.slf4j.Slf4j;
 import tla.backend.es.model.SentenceEntity;
+import tla.backend.es.model.SentenceEntity.Context;
 import tla.backend.service.ModelClass;
 import tla.domain.command.PassportSpec;
 
@@ -20,6 +21,23 @@
 public class SentenceSearchQueryBuilder extends ESQueryBuilder implements MultiLingQueryBuilder {
 
     public final static String AGG_ID_TEXT_IDS = "text_ids";
+
+    public void setContext(Context context) {
+   	 BoolQueryBuilder textQuery = boolQuery();
+   	String textId = context.getTextId();
+   	   if (textId != null) {
+
+              log.info("sentence query: receive {} text IDs", textId);
+              textQuery.must(
+                      QueryBuilders.termQuery(
+                      "context.textId",
+                      textId
+                  )
+              );
+              this.filter(textQuery);
+          }
+
+   }
 
     public void setTokens(Collection<TokenSearchQueryBuilder> tokenQueries) {
         BoolQueryBuilder tokenQuery = boolQuery();