Example usage for org.springframework.data.elasticsearch.core.query NativeSearchQueryBuilder withSort

Introduction

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

Prototype

public NativeSearchQueryBuilder withSort(SortBuilder sortBuilder) 

Source Link

Usage

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

@SuppressWarnings("unchecked")
@Override//from  w  ww.  j  a  v a 2s. c o 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));
    ;/*from ww w .  j a v a  2s.  c o 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;
}

---