90
90
import org .springframework .beans .BeansException ;
91
91
import org .springframework .context .ApplicationContext ;
92
92
import org .springframework .context .ApplicationContextAware ;
93
+ import org .springframework .data .convert .EntityConverter ;
93
94
import org .springframework .data .domain .Page ;
94
95
import org .springframework .data .domain .Pageable ;
95
96
import org .springframework .data .domain .Sort ;
@@ -323,14 +324,14 @@ public <T> T queryForObject(GetQuery query, Class<T> clazz) {
323
324
}
324
325
325
326
@ Override
326
- public <T > T queryForObject (GetQuery query , Class <T > clazz , GetResultMapper mapper ) {
327
+ public <T > T queryForObject (GetQuery query , Class <T > clazz , ElasticsearchConverter converter ) {
327
328
ElasticsearchPersistentEntity <T > persistentEntity = getPersistentEntityFor (clazz );
328
329
GetRequest request = new GetRequest (persistentEntity .getIndexName (), persistentEntity .getIndexType (),
329
330
query .getId ());
330
331
GetResponse response ;
331
332
try {
332
333
response = client .get (request , RequestOptions .DEFAULT );
333
- return mapper .mapResult (response , clazz );
334
+ return converter .mapResult (response , clazz );
334
335
} catch (IOException e ) {
335
336
throw new ElasticsearchException ("Error while getting for request: " + request .toString (), e );
336
337
}
@@ -359,9 +360,9 @@ public <T> AggregatedPage<T> queryForPage(SearchQuery query, Class<T> clazz) {
359
360
}
360
361
361
362
@ Override
362
- public <T > AggregatedPage <T > queryForPage (SearchQuery query , Class <T > clazz , SearchResultMapper mapper ) {
363
+ public <T > AggregatedPage <T > queryForPage (SearchQuery query , Class <T > clazz , ElasticsearchConverter converter ) {
363
364
SearchResponse response = doSearch (prepareSearch (query , clazz ), query );
364
- return mapper .mapResults (response , clazz , query .getPageable ());
365
+ return converter .mapResults (response , clazz , query .getPageable ());
365
366
}
366
367
367
368
@ Override
@@ -370,24 +371,24 @@ public <T> List<Page<T>> queryForPage(List<SearchQuery> queries, Class<T> clazz)
370
371
}
371
372
372
373
private <T > List <Page <T >> doMultiSearch (List <SearchQuery > queries , Class <T > clazz , MultiSearchRequest request ,
373
- SearchResultMapper resultsMapper ) {
374
+ ElasticsearchConverter converter ) {
374
375
MultiSearchResponse .Item [] items = getMultiSearchResult (request );
375
376
List <Page <T >> res = new ArrayList <>(queries .size ());
376
377
int c = 0 ;
377
378
for (SearchQuery query : queries ) {
378
- res .add (resultsMapper .mapResults (items [c ++].getResponse (), clazz , query .getPageable ()));
379
+ res .add (converter .mapResults (items [c ++].getResponse (), clazz , query .getPageable ()));
379
380
}
380
381
return res ;
381
382
}
382
383
383
384
private List <Page <?>> doMultiSearch (List <SearchQuery > queries , List <Class <?>> classes , MultiSearchRequest request ,
384
- SearchResultMapper resultsMapper ) {
385
+ ElasticsearchConverter converter ) {
385
386
MultiSearchResponse .Item [] items = getMultiSearchResult (request );
386
387
List <Page <?>> res = new ArrayList <>(queries .size ());
387
388
int c = 0 ;
388
389
Iterator <Class <?>> it = classes .iterator ();
389
390
for (SearchQuery query : queries ) {
390
- res .add (resultsMapper .mapResults (items [c ++].getResponse (), it .next (), query .getPageable ()));
391
+ res .add (converter .mapResults (items [c ++].getResponse (), it .next (), query .getPageable ()));
391
392
}
392
393
return res ;
393
394
}
@@ -405,12 +406,12 @@ private MultiSearchResponse.Item[] getMultiSearchResult(MultiSearchRequest reque
405
406
}
406
407
407
408
@ Override
408
- public <T > List <Page <T >> queryForPage (List <SearchQuery > queries , Class <T > clazz , SearchResultMapper mapper ) {
409
+ public <T > List <Page <T >> queryForPage (List <SearchQuery > queries , Class <T > clazz , ElasticsearchConverter converter ) {
409
410
MultiSearchRequest request = new MultiSearchRequest ();
410
411
for (SearchQuery query : queries ) {
411
412
request .add (prepareSearch (prepareSearch (query , clazz ), query ));
412
413
}
413
- return doMultiSearch (queries , clazz , request , mapper );
414
+ return doMultiSearch (queries , clazz , request , converter );
414
415
}
415
416
416
417
@ Override
@@ -419,13 +420,13 @@ public List<Page<?>> queryForPage(List<SearchQuery> queries, List<Class<?>> clas
419
420
}
420
421
421
422
@ Override
422
- public List <Page <?>> queryForPage (List <SearchQuery > queries , List <Class <?>> classes , SearchResultMapper mapper ) {
423
+ public List <Page <?>> queryForPage (List <SearchQuery > queries , List <Class <?>> classes , ElasticsearchConverter converter ) {
423
424
MultiSearchRequest request = new MultiSearchRequest ();
424
425
Iterator <Class <?>> it = classes .iterator ();
425
426
for (SearchQuery query : queries ) {
426
427
request .add (prepareSearch (prepareSearch (query , it .next ()), query ));
427
428
}
428
- return doMultiSearch (queries , classes , request , mapper );
429
+ return doMultiSearch (queries , classes , request , converter );
429
430
}
430
431
431
432
@ Override
@@ -508,7 +509,7 @@ public <T> Page<T> queryForPage(StringQuery query, Class<T> clazz) {
508
509
}
509
510
510
511
@ Override
511
- public <T > Page <T > queryForPage (StringQuery query , Class <T > clazz , SearchResultMapper mapper ) {
512
+ public <T > Page <T > queryForPage (StringQuery query , Class <T > clazz , ElasticsearchConverter converter ) {
512
513
SearchRequest request = prepareSearch (query , clazz );
513
514
request .source ().query ((wrapperQuery (query .getSource ())));
514
515
SearchResponse response ;
@@ -517,7 +518,7 @@ public <T> Page<T> queryForPage(StringQuery query, Class<T> clazz, SearchResultM
517
518
} catch (IOException e ) {
518
519
throw new ElasticsearchException ("Error for search request: " + request .toString (), e );
519
520
}
520
- return mapper .mapResults (response , clazz , query .getPageable ());
521
+ return converter .mapResults (response , clazz , query .getPageable ());
521
522
}
522
523
523
524
@ Override
@@ -532,14 +533,14 @@ public <T> CloseableIterator<T> stream(SearchQuery query, Class<T> clazz) {
532
533
}
533
534
534
535
@ Override
535
- public <T > CloseableIterator <T > stream (SearchQuery query , Class <T > clazz , SearchResultMapper mapper ) {
536
+ public <T > CloseableIterator <T > stream (SearchQuery query , Class <T > clazz , ElasticsearchConverter converter ) {
536
537
long scrollTimeInMillis = TimeValue .timeValueMinutes (1 ).millis ();
537
- return doStream (scrollTimeInMillis , startScroll (scrollTimeInMillis , query , clazz , mapper ), clazz , mapper );
538
+ return doStream (scrollTimeInMillis , startScroll (scrollTimeInMillis , query , clazz , converter ), clazz , converter );
538
539
}
539
540
540
541
private <T > CloseableIterator <T > doStream (long scrollTimeInMillis , ScrolledPage <T > page , Class <T > clazz ,
541
- SearchResultMapper mapper ) {
542
- return StreamQueries .streamResults (page , scrollId -> continueScroll (scrollId , scrollTimeInMillis , clazz , mapper ),
542
+ ElasticsearchConverter converter ) {
543
+ return StreamQueries .streamResults (page , scrollId -> continueScroll (scrollId , scrollTimeInMillis , clazz , converter ),
543
544
this ::clearScroll );
544
545
}
545
546
@@ -669,8 +670,8 @@ private <T> MultiGetResponse getMultiResponse(Query searchQuery, Class<T> clazz)
669
670
}
670
671
671
672
@ Override
672
- public <T > List <T > multiGet (SearchQuery searchQuery , Class <T > clazz , MultiGetResultMapper getResultMapper ) {
673
- return getResultMapper .mapResults (getMultiResponse (searchQuery , clazz ), clazz );
673
+ public <T > List <T > multiGet (SearchQuery searchQuery , Class <T > clazz , ElasticsearchConverter converter ) {
674
+ return converter .mapResults (getMultiResponse (searchQuery , clazz ), clazz );
674
675
}
675
676
676
677
@ Override
@@ -1038,16 +1039,16 @@ public <T> ScrolledPage<T> startScroll(long scrollTimeInMillis, CriteriaQuery cr
1038
1039
1039
1040
@ Override
1040
1041
public <T > ScrolledPage <T > startScroll (long scrollTimeInMillis , SearchQuery searchQuery , Class <T > clazz ,
1041
- SearchResultMapper mapper ) {
1042
+ ElasticsearchConverter converter ) {
1042
1043
SearchResponse response = doScroll (prepareScroll (searchQuery , scrollTimeInMillis , clazz ), searchQuery );
1043
- return mapper .mapResults (response , clazz , null );
1044
+ return converter .mapResults (response , clazz , null );
1044
1045
}
1045
1046
1046
1047
@ Override
1047
1048
public <T > ScrolledPage <T > startScroll (long scrollTimeInMillis , CriteriaQuery criteriaQuery , Class <T > clazz ,
1048
- SearchResultMapper mapper ) {
1049
+ ElasticsearchConverter converter ) {
1049
1050
SearchResponse response = doScroll (prepareScroll (criteriaQuery , scrollTimeInMillis , clazz ), criteriaQuery );
1050
- return mapper .mapResults (response , clazz , null );
1051
+ return converter .mapResults (response , clazz , null );
1051
1052
}
1052
1053
1053
1054
@ Override
@@ -1065,7 +1066,7 @@ public <T> ScrolledPage<T> continueScroll(@Nullable String scrollId, long scroll
1065
1066
1066
1067
@ Override
1067
1068
public <T > ScrolledPage <T > continueScroll (@ Nullable String scrollId , long scrollTimeInMillis , Class <T > clazz ,
1068
- SearchResultMapper mapper ) {
1069
+ ElasticsearchConverter converter ) {
1069
1070
SearchScrollRequest request = new SearchScrollRequest (scrollId );
1070
1071
request .scroll (TimeValue .timeValueMillis (scrollTimeInMillis ));
1071
1072
SearchResponse response ;
@@ -1074,7 +1075,7 @@ public <T> ScrolledPage<T> continueScroll(@Nullable String scrollId, long scroll
1074
1075
} catch (IOException e ) {
1075
1076
throw new ElasticsearchException ("Error for search request with scroll: " + request .toString (), e );
1076
1077
}
1077
- return mapper .mapResults (response , clazz , Pageable .unpaged ());
1078
+ return converter .mapResults (response , clazz , Pageable .unpaged ());
1078
1079
}
1079
1080
1080
1081
@ Override
@@ -1592,10 +1593,6 @@ private static MoreLikeThisQueryBuilder.Item[] toArray(MoreLikeThisQueryBuilder.
1592
1593
return values ;
1593
1594
}
1594
1595
1595
- protected ResultsMapper getResultsMapper () {
1596
- return elasticsearchConverter ;
1597
- }
1598
-
1599
1596
@ Deprecated
1600
1597
public static String readFileFromClasspath (String url ) {
1601
1598
return ResourceUtil .readFileFromClasspath (url );
0 commit comments