Example usage for org.springframework.data.elasticsearch.core.query SearchQuery getQuery

List of usage examples for org.springframework.data.elasticsearch.core.query SearchQuery getQuery

  1. HOME
  2. Java
  3. org.springframework
  4. org.springframework.data.elasticsearch.core.query.*
  5. SearchQuery
  6. getQuery

Introduction

In this page you can find the example usage for org.springframework.data.elasticsearch.core.query SearchQuery getQuery.

Prototype

QueryBuilder getQuery();

Source Link

Usage

From source file:com.github.lothar.security.acl.elasticsearch.repository.AclElasticsearchRepository.java

@Override
public Page<T> search(SearchQuery query) {
    Assert.isInstanceOf(NativeSearchQuery.class, query, "NativeSearchQuery only are supported :(");
    NativeSearchQuery searchQuery = new NativeSearchQuery(and(query.getQuery(), aclFilter()));
    BeanUtils.copyProperties(query, searchQuery, "query");
    return elasticsearchOperations.queryForPage(searchQuery, getEntityClass());
}

From source file:org.apereo.openlrs.repositories.statements.ElasticSearchStatementRepository.java

@SuppressWarnings("unchecked")
@Override/*from   ww w.j  a  v  a 2  s. co  m*/
public List<Statement> get(Map<String, String> filters) {
    String actor = filters.get(StatementUtils.ACTOR_FILTER);
    String activity = filters.get(StatementUtils.ACTIVITY_FILTER);
    String since = filters.get(StatementUtils.SINCE_FILTER);
    String until = filters.get(StatementUtils.UNTIL_FILTER);
    int limit = getLimit(filters.get(StatementUtils.LIMIT_FILTER));
    ;

    XApiActor xApiActor = null;

    if (StringUtils.isNotBlank(actor)) {
        ObjectMapper objectMapper = new ObjectMapper();
        try {
            xApiActor = objectMapper.readValue(actor.getBytes(), XApiActor.class);
        } catch (Exception e) {
            log.error(e.getMessage(), e);
        }
    }

    SearchQuery searchQuery = null;

    if (StringUtils.isNotBlank(activity) && xApiActor != null) {
        QueryBuilder actorQuery = buildActorQuery(xApiActor);
        QueryBuilder activityQuery = nestedQuery("object", boolQuery().must(matchQuery("object.id", activity)));

        BoolQueryBuilder boolQuery = boolQuery().must(actorQuery).must(activityQuery);

        searchQuery = startQuery(limit, boolQuery).build();
    } else if (xApiActor != null) {

        QueryBuilder query = buildActorQuery(xApiActor);

        if (query != null) {
            searchQuery = startQuery(limit, query).build();
        }
    } else if (StringUtils.isNotBlank(activity)) {
        QueryBuilder query = nestedQuery("object", boolQuery().must(matchQuery("object.id", activity)));
        searchQuery = startQuery(limit, query).build();
    } else if (StringUtils.isNotBlank(since) || StringUtils.isNotBlank(until)) {
        QueryBuilder query = null;

        if (StringUtils.isNotBlank(since) && StringUtils.isNotBlank(until)) {
            query = new RangeQueryBuilder("stored").from(since).to(until);
        } else {
            if (StringUtils.isNotBlank(since)) {
                query = new RangeQueryBuilder("stored").from(since).to("now");
            }

            if (StringUtils.isNotBlank(until)) {
                try {

                    DateFormat formatter = new SimpleDateFormat("yyyy-MM-dd'T'HH:mm:ss'Z'");
                    TimeZone tz = TimeZone.getTimeZone("UTC");
                    formatter.setTimeZone(tz);
                    Date date = (Date) formatter.parse(until);
                    Calendar calendar = Calendar.getInstance();
                    calendar.setTime(date);
                    calendar.add(Calendar.YEAR, -1);

                    query = new RangeQueryBuilder("stored").from(formatter.format(calendar.getTime()))
                            .to(until);
                } catch (ParseException e) {
                    log.error(e.getMessage(), e);
                    return null;
                }
            }
        }

        NativeSearchQueryBuilder searchQueryBuilder = startQuery(limit, query);

        searchQuery = searchQueryBuilder.withSort(new FieldSortBuilder("stored").order(SortOrder.DESC)).build();
    } else if (limit > 0) {
        searchQuery = startQuery(limit, null).build();
    }

    if (searchQuery != null) {
        if (log.isDebugEnabled()) {
            if (searchQuery.getQuery() != null) {
                log.debug(String.format("Elasticsearch query %s", searchQuery.getQuery().toString()));
            }
        }

        Iterable<Statement> iterableStatements = esSpringDataRepository.search(searchQuery);
        if (iterableStatements != null) {
            return IteratorUtils.toList(iterableStatements.iterator());
        }
    }
    return null;
}

From source file:org.apereo.openlrs.storage.elasticsearch.XApiOnlyElasticsearchTierTwoStorage.java

@Override
public Page<OpenLRSEntity> findWithFilters(Map<String, String> filters, Pageable pageable) {
    String actor = filters.get(StatementUtils.ACTOR_FILTER);
    String activity = filters.get(StatementUtils.ACTIVITY_FILTER);
    String since = filters.get(StatementUtils.SINCE_FILTER);
    String until = filters.get(StatementUtils.UNTIL_FILTER);
    int limit = getLimit(filters.get(StatementUtils.LIMIT_FILTER));
    ;//w  ww  .ja  v a2s  . co m

    XApiActor xApiActor = null;

    if (StringUtils.isNotBlank(actor)) {
        ObjectMapper objectMapper = new ObjectMapper();
        try {
            xApiActor = objectMapper.readValue(actor.getBytes(), XApiActor.class);
        } catch (Exception e) {
            log.error(e.getMessage(), e);
        }
    }

    SearchQuery searchQuery = null;

    if (StringUtils.isNotBlank(activity) && xApiActor != null) {
        QueryBuilder actorQuery = buildActorQuery(xApiActor);
        QueryBuilder activityQuery = nestedQuery("object", boolQuery().must(matchQuery("object.id", activity)));

        BoolQueryBuilder boolQuery = boolQuery().must(actorQuery).must(activityQuery);

        searchQuery = startQuery(limit, boolQuery).build();
    } else if (xApiActor != null) {

        QueryBuilder query = buildActorQuery(xApiActor);

        if (query != null) {
            searchQuery = startQuery(limit, query).build();
        }
    } else if (StringUtils.isNotBlank(activity)) {
        QueryBuilder query = nestedQuery("object", boolQuery().must(matchQuery("object.id", activity)));
        searchQuery = startQuery(limit, query).build();
    } else if (StringUtils.isNotBlank(since) || StringUtils.isNotBlank(until)) {
        QueryBuilder query = null;

        if (StringUtils.isNotBlank(since) && StringUtils.isNotBlank(until)) {
            query = new RangeQueryBuilder("stored").from(since).to(until);
        } else {
            if (StringUtils.isNotBlank(since)) {
                query = new RangeQueryBuilder("stored").from(since).to("now");
            }

            if (StringUtils.isNotBlank(until)) {
                try {

                    DateFormat formatter = new SimpleDateFormat("yyyy-MM-dd'T'HH:mm:ss'Z'");
                    TimeZone tz = TimeZone.getTimeZone("UTC");
                    formatter.setTimeZone(tz);
                    Date date = (Date) formatter.parse(until);
                    Calendar calendar = Calendar.getInstance();
                    calendar.setTime(date);
                    calendar.add(Calendar.YEAR, -1);

                    query = new RangeQueryBuilder("stored").from(formatter.format(calendar.getTime()))
                            .to(until);
                } catch (ParseException e) {
                    log.error(e.getMessage(), e);
                    return null;
                }
            }
        }

        NativeSearchQueryBuilder searchQueryBuilder = startQuery(limit, query);

        searchQuery = searchQueryBuilder.withSort(new FieldSortBuilder("stored").order(SortOrder.DESC)).build();
    } else if (limit > 0) {
        searchQuery = startQuery(limit, null).build();
    }

    if (searchQuery != null) {
        if (log.isDebugEnabled()) {
            if (searchQuery.getQuery() != null) {
                log.debug(String.format("Elasticsearch query %s", searchQuery.getQuery().toString()));
            }
        }

        Iterable<Statement> iterableStatements = esSpringDataRepository.search(searchQuery);
        if (iterableStatements != null) {
            return new PageImpl<OpenLRSEntity>(IteratorUtils.toList(iterableStatements.iterator()));
        }
    }
    return null;
}

From source file:org.springframework.data.elasticsearch.core.ElasticsearchTemplate.java

@Override
public <T> List<String> queryForIds(SearchQuery query) {
    SearchRequestBuilder request = prepareSearch(query).setQuery(query.getQuery()).setNoFields();
    if (query.getFilter() != null) {
        request.setFilter(query.getFilter());
    }/*from   w w w .  j  a  v a2  s.  c o  m*/
    SearchResponse response = request.execute().actionGet();
    return extractIds(response);
}

From source file:org.springframework.data.elasticsearch.core.ElasticsearchTemplate.java

@Override
public <T> long count(SearchQuery query, Class<T> clazz) {
    ElasticsearchPersistentEntity<T> persistentEntity = getPersistentEntityFor(clazz);
    CountRequestBuilder countRequestBuilder = client.prepareCount(persistentEntity.getIndexName())
            .setTypes(persistentEntity.getIndexType());
    if (query.getQuery() != null) {
        countRequestBuilder.setQuery(query.getQuery());
    }/*w ww  .j  av  a2  s.  c  o  m*/
    return countRequestBuilder.execute().actionGet().getCount();
}

From source file:org.springframework.data.elasticsearch.core.ElasticsearchTemplate.java

@Override
public String scan(SearchQuery searchQuery, long scrollTimeInMillis, boolean noFields) {
    Assert.notNull(searchQuery.getIndices(), "No index defined for Query");
    Assert.notNull(searchQuery.getTypes(), "No type define for Query");
    Assert.notNull(searchQuery.getPageable(), "Query.pageable is required for scan & scroll");

    SearchRequestBuilder requestBuilder = client.prepareSearch(toArray(searchQuery.getIndices()))
            .setSearchType(SCAN).setQuery(searchQuery.getQuery()).setTypes(toArray(searchQuery.getTypes()))
            .setScroll(TimeValue.timeValueMillis(scrollTimeInMillis)).setFrom(0)
            .setSize(searchQuery.getPageable().getPageSize());

    if (searchQuery.getFilter() != null) {
        requestBuilder.setFilter(searchQuery.getFilter());
    }/*from  ww  w. j ava  2 s . c  o  m*/

    if (noFields) {
        requestBuilder.setNoFields();
    }
    return requestBuilder.execute().actionGet().getScrollId();
}

From source file:org.springframework.data.elasticsearch.core.ElasticsearchTemplate.java

private SearchResponse doSearch(SearchRequestBuilder searchRequest, SearchQuery searchQuery) {
    if (searchQuery.getFilter() != null) {
        searchRequest.setFilter(searchQuery.getFilter());
    }/*from   w ww .j  a va2s.c  o m*/

    if (searchQuery.getElasticsearchSort() != null) {
        searchRequest.addSort(searchQuery.getElasticsearchSort());
    }

    if (CollectionUtils.isNotEmpty(searchQuery.getFacets())) {
        for (FacetRequest facetRequest : searchQuery.getFacets()) {
            FacetBuilder facet = facetRequest.getFacet();
            if (facetRequest.applyQueryFilter() && searchQuery.getFilter() != null) {
                facet.facetFilter(searchQuery.getFilter());
            }
            searchRequest.addFacet(facet);
        }
    }

    if (searchQuery.getHighlightFields() != null) {
        for (HighlightBuilder.Field highlightField : searchQuery.getHighlightFields()) {
            searchRequest.addHighlightedField(highlightField);
        }
    }

    return searchRequest.setQuery(searchQuery.getQuery()).execute().actionGet();
}