Автозаполнение с помощью Hibernate Search - PullRequest
Новая
       5

Автозаполнение с помощью Hibernate Search

11 голосов
/ 19 марта 2011

Я пытаюсь создать лучшую функцию автозаполнения для моего сайта.Я хочу использовать Hibernate Search для этого, но, насколько я экспериментировал, он находит только полные слова для меня.

Итак, мой вопрос: можно ли искать только некоторые символы?

например.пользователь вводит 3 буквы и использует поиск в спящем режиме, чтобы показать ему все слова моих объектов БД, которые содержат эти 3 буквы?

PS.прямо сейчас я использую запрос "лайк" для этого ... но мой БД сильно вырос, и я хочу также расширить функциональность поиска по другим таблицам ...

Ответы [ 3 ]

12 голосов
/ 29 июня 2011

Основное редактирование Через год я смог улучшить исходный код, который я разместил, для создания этого:

Моя индексированная сущность: 

@Entity
@Indexed
@AnalyzerDef(name = "myanalyzer",
// Split input into tokens according to tokenizer
tokenizer = @TokenizerDef(factory = WhitespaceTokenizerFactory.class), //
filters = { //
// Normalize token text to lowercase, as the user is unlikely to care about casing when searching for matches
@TokenFilterDef(factory = LowerCaseFilterFactory.class),
// Index partial words starting at the front, so we can provide Autocomplete functionality
@TokenFilterDef(factory = NGramFilterFactory.class, params = { @Parameter(name = "maxGramSize", value = "1024") }),
// Close filters & Analyzerdef
})
@Analyzer(definition = "myanalyzer")
public class Compound extends DomainObject {
public static String[] getSearchFields(){...}
...
}

Все@Field с токены и хранятся в индексе;чтобы это работало:
@Field(index = Index.TOKENIZED, store = Store.YES)

@Transactional(readOnly = true)
public synchronized List<String> getSuggestions(final String searchTerm) {
    // Compose query for term over all fields in Compound
    String lowerCasedSearchTerm = searchTerm.toLowerCase();

    // Create a fullTextSession for the sessionFactory.getCurrentSession()
    FullTextSession fullTextSession = Search.getFullTextSession(getSession());

    // New DSL based query composition
    SearchFactory searchFactory = fullTextSession.getSearchFactory();
    QueryBuilder buildQuery = searchFactory.buildQueryBuilder().forEntity(Compound.class).get();
    TermContext keyword = buildQuery.keyword();
    WildcardContext wildcard = keyword.wildcard();
    String[] searchfields = Compound.getSearchfields();
    TermMatchingContext onFields = wildcard.onField(searchfields[0]);
    for (int i = 1; i < searchfields.length; i++)
        onFields.andField(searchfields[i]);
    TermTermination matching = onFields.matching(input.toLowerCase());
    Query query = matching.createQuery();

    // Convert the Search Query into something that provides results: Specify Compound again to be future proof
    FullTextQuery fullTextQuery = fullTextSession.createFullTextQuery(query, Compound.class);
    fullTextQuery.setMaxResults(20);

    // Projection does not work on collections or maps which are indexed via @IndexedEmbedded
    List<String> projectedFields = new ArrayList<String>();
    projectedFields.add(ProjectionConstants.DOCUMENT);
    List<String> embeddedFields = new ArrayList<String>();
    for (String fieldName : searchfields)
        if (fieldName.contains("."))
            embeddedFields.add(fieldName);
        else
            projectedFields.add(fieldName);

    @SuppressWarnings("unchecked")
    List<Object[]> results = fullTextQuery.setProjection(projectedFields.toArray(new String[projectedFields.size()])).list();

    // Keep a list of suggestions retrieved by search over all fields
    List<String> suggestions = new ArrayList<String>();
    for (Object[] projectedObjects : results) {
        // Retrieve the search suggestions for the simple projected field values
        for (int i = 1; i < projectedObjects.length; i++) {
            String fieldValue = projectedObjects[i].toString();
            if (fieldValue.toLowerCase().contains(lowerCasedSearchTerm))
                suggestions.add(fieldValue);
        }

        // Extract the search suggestions for the embedded fields from the document
        Document document = (Document) projectedObjects[0];
        for (String fieldName : embeddedFields)
            for (Field field : document.getFields(fieldName))
                if (field.stringValue().toLowerCase().contains(lowerCasedSearchTerm))
                    suggestions.add(field.stringValue());
    }

    // Return the composed list of suggestions, which might be empty
    return suggestions;
}

В конце я провожу некоторые споры по поводу обработки полей @IndexedEmbedded.Если у вас их нет, вы можете значительно упростить код, просто спроецировав searchFields и не обращая внимания на обработку документа и встроенного поля.

Как и прежде: Надеюсь, это будет полезно следующему человеку, который столкнется с этим вопросом.Если у кого-то есть какие-либо критические замечания или улучшения по поводу вышеприведенного кода, не стесняйтесь редактировать и сообщите мне.

Edit3 : проект, из которого был взят этот код, с тех порбыл с открытым исходным кодом;Вот соответствующие классы:

https://trac.nbic.nl/metidb/browser/trunk/metidb/metidb-core/src/main/java/org/metidb/domain/Compound.java
https://trac.nbic.nl/metidb/browser/trunk/metidb/metidb-core/src/main/java/org/metidb/dao/CompoundDAOImpl.java
https://trac.nbic.nl/metidb/browser/trunk/metidb/metidb-search/src/main/java/org/metidb/search/text/Autocompleter.java

7 голосов
/ 19 марта 2011

Вы можете индексировать поле, используя NGramFilter , как предложено здесь . Для достижения наилучших результатов вы должны использовать EdgeNgramFilter от Apache Solr, который создает нграммы от начального края термина и может также использоваться в поиске гибернации.

2 голосов
/ 08 мая 2013

Ответ Тима блестящий и помог мне преодолеть сложную часть. Это работало только для одного запроса слова для меня. На случай, если кто-то захочет, чтобы это работало для поиска по фразе. Просто замените все экземпляры Term соответствующими классами Phrase. Вот строки замены кода Тима, который помог мне. 

// New DSL based query composition
            //org.hibernate.search.query.dsl
            SearchFactory searchFactory = fullTextSession.getSearchFactory();
            QueryBuilder buildQuery = searchFactory.buildQueryBuilder().forEntity(MasterDiagnosis.class).get();
            PhraseContext keyword = buildQuery.phrase();
            keyword.withSlop(3);
            //WildcardContext wildcard = keyword.wildcard();
            String[] searchfields = MasterDiagnosis.getSearchfields();
            PhraseMatchingContext onFields = keyword.onField(searchfields[0]);
            for (int i = 1; i < searchfields.length; i++)
                onFields.andField(searchfields[i]);
            PhraseTermination matching = onFields.sentence(lowerCasedSearchTerm);
            Query query = matching.createQuery();
 // Convert the Search Query into something that provides results: Specify Compound again to be future proof
...