Search code examples
elasticsearchspring-data-elasticsearchelasticsearch-java-api

Spring data elasticsearch: setting sort value to entity field


Finding nearest stores from elasticsearch, using geo distance sort using spring-data-elastic search.

Problem

StoreES getting populated with orgCode and storeCode , because they are in _source. How can I populate distance value from search response sort[0] value.

StoreElasticSearchRepository method

public default List<StoreES> nearStores(Double latitude, Double longitude, Integer page, Integer size) {
    GeoDistanceSortBuilder distanceSortbuilder = SortBuilders.geoDistanceSort("location").point(latitude, longitude)
                        .order(SortOrder.ASC).unit(DistanceUnit.KILOMETERS);
                SearchQuery query = new NativeSearchQueryBuilder().withQuery(QueryBuilders.matchAllQuery())
                        .withSort(distanceSortbuilder).withPageable(new PageRequest(page, size)).build();
                FacetedPage<StoreES> storesPage = this.search(query);
    return storesPage.getContent();
}

Rest request equivalent for above query:

{
  "sort": {
    "_geo_distance": {
      "location": {
        "lat": 12.9259,
        "lon": 77.6229
      },
      "order": "asc",
      "unit": "km"
    }
  }
}

Result for query

    {
  "took": 5,
  "timed_out": false,
  "_shards": {
    "total": 5,
    "successful": 5,
    "failed": 0
  },
  "hits": {
    "total": 1365,
    "max_score": null,
    "hits": [
      {
        "_index": "app",
        "_type": "store",
        "_id": "99991258",
        "_score": null,
        "_source": {
          "id": 1,
          "orgCode": "ABC",
          "storeCode": "12345",
       },
      "sort": [
          0.49933591591981075
        ]
}

Entity object

public class StoreES{

    private String orgCode;

    private String storeCode;
  
    private Double distance;

// setter gettter methods
}

Thanks In advance.


Solution

  • Work around

    public default List<StoreES> nearStores(Double latitude, Double longitude, String address, Integer page,
                Integer size) {
    DistanceSortBuilder distanceSortbuilder = SortBuilders.geoDistanceSort("location").point(latitude, longitude)
                        .order(SortOrder.ASC).unit(DistanceUnit.KILOMETERS);
                SearchQuery query = new NativeSearchQueryBuilder().withQuery(QueryBuilders.matchAllQuery())
                        .withSort(distanceSortbuilder).withPageable(new PageRequest(page, size)).build();
                FacetedPage<StoreES> storesPage = this.search(query);
    
        List<StoreES> stores = storesPage.getContent();
    
            // ##############calculate distance##################
            for (StoreES store : stores) {
                if (store.getLocation() != null) {
                    double distance = GeoDistance.DEFAULT.calculate(latitude, longitude, store.getLocation().getLat(),
                            store.getLocation().getLon(), DistanceUnit.KILOMETERS);
                    store.setDistance(distance);
                }
                store.setDistanceUnit(DistanceUnit.KILOMETERS.toString());
            }
            return stores;
        }