feat(next/api): phrase query

leancloud · Jan 5, 2024 · e4d4c23 · e4d4c23
1 parent d63b2d1
commit e4d4c23
Showing 1 changed file with 33 additions and 3 deletions.
diff --git a/next/api/src/service/search-ticket.ts b/next/api/src/service/search-ticket.ts
@@ -12,6 +12,11 @@ import { Ticket } from '@/model/Ticket';
 import { TicketFieldValue } from '@/model/TicketFieldValue';
 import { createQueue } from '@/queue';
 
+interface Term {
+  value: string;
+  isPhrase?: true;
+}
+
 export class SearchTicketService {
   private esClient?: Client;
   private indexName: string;
@@ -285,9 +290,16 @@ export class SearchTicketService {
       );
     }
     if (filters.keyword) {
-      boolQuery.filter(
-        esb.multiMatchQuery(['title', 'content', 'fields.value'], filters.keyword).operator('and')
-      );
+      const terms = this.parseTerms(filters.keyword);
+      terms.forEach((term) => {
+        const matchQuery = esb
+          .multiMatchQuery(['title', 'content', 'fields.value'], term.value)
+          .operator('and');
+        if (term.isPhrase) {
+          matchQuery.type('phrase');
+        }
+        boolQuery.filter(matchQuery);
+      });
     }
 
     const body = esb
@@ -308,6 +320,24 @@ export class SearchTicketService {
     const totalCount = res.body.hits.total.value as number;
     return { ids, totalCount };
   }
+
+  parseTerms(input: string) {
+    const regex = /"((?:\\.|[^"\\])*)"|(\S+)/g;
+    const terms: Term[] = [];
+
+    for (const [, phrase, term] of input.matchAll(regex)) {
+      if (phrase) {
+        const value = phrase.trim().replace(/\\(.)/g, '$1');
+        if (value) {
+          terms.push({ value, isPhrase: true });
+        }
+      } else if (term) {
+        terms.push({ value: term });
+      }
+    }
+
+    return terms;
+  }
 }
 
 export const searchTicketService = new SearchTicketService();