ElasticSearch RestClient ответ на ResponseHits API вместо Response API не работает - PullRequest
Новая
       9

ElasticSearch RestClient ответ на ResponseHits API вместо Response API не работает

0 голосов
/ 27 июня 2018

Я извлекаю документы изasticsearch через RestClient Java API и пытаюсь получить ответ в ResponseHits, чтобы иметь возможность обрабатывать несколько документов (в ответ), используя приведенный ниже код.

SearchHit[] contentSearchHits = contentSearchResponse.getHits().getHits();

Но я не могу установить RestClient ответ в ResponseHits API. От Затмения получаю это сообщение Type mismatch: cannot convert from Response to ResponseHits.

Если я использую RestHighLevelClient, могу использовать ResponseHits класс, и все идет хорошо. Но я не использую его из-за ограничения BUFFER_SIZE по умолчанию 100 МБ, где мне нужно больше BUFFER_SIZE.

Итак, есть ли способ обработать ответ RestClient, если при поиске возвращается несколько элементов.

Пожалуйста, найдите мой Java-код, который используется для извлечения документов из ElasticSearch. 

private final static String ATTACHMENT = "document_attachment";
    private final static String TYPE = "doc";
    static long BUFFER_SIZE = 520 * 1024 * 1024;   //  <---- set buffer to 520MB instead of 100MB


    public static void main(String args[])
    {
        RestClient restClient = null;
        Response contentSearchResponse=null;
        String responseBody = null;
        JSONObject source = null;
        String path = null;
        String filename = null;
        int id = 0;
        ResponseHits responseHits = null;

        RestClientBuilder builder =  null; 

        try {

        restClient = RestClient.builder(
                        new HttpHost("localhost", 9200, "http"),
                        new HttpHost("localhost", 9201, "http")).build();

        } catch (Exception e) {
            System.out.println(e.getMessage());
        }

        SearchRequest contentSearchRequest = new SearchRequest(ATTACHMENT); 
        SearchSourceBuilder contentSearchSourceBuilder = new SearchSourceBuilder();
        contentSearchRequest.types(TYPE);
        QueryBuilder attachmentQB = QueryBuilders.matchQuery("attachment.content", "activa");
        contentSearchSourceBuilder.query(attachmentQB);
        contentSearchSourceBuilder.size(50);
        contentSearchRequest.source(contentSearchSourceBuilder);

        Map<String, String> params = Collections.emptyMap();
        HttpEntity entity = new NStringEntity(contentSearchSourceBuilder.toString(), ContentType.APPLICATION_JSON);
        HttpAsyncResponseConsumerFactory.HeapBufferedResponseConsumerFactory consumerFactory =
                new HttpAsyncResponseConsumerFactory.HeapBufferedResponseConsumerFactory((int) BUFFER_SIZE);


        try {
            //contentSearchResponse = restClient.performRequest("GET", "/document_attachment/doc/_search", params, entity, consumerFactory); //  this is working with single document, not sure how to handle if request is returning multiple documents while searching.
            responseHits  = restClient.performRequest("GET", "/document_attachment/doc/_search", params, entity, consumerFactory); // am not able to set the response to `ResponseHits` class
        } catch (IOException e1) {
            e1.printStackTrace();
        } 

        try {
            responseBody = EntityUtils.toString(contentSearchResponse.getEntity());
        } catch (ParseException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }

        System.out.println("Converting to JSON");
        JSONObject jsonObject = new JSONObject(responseBody);
        JSONObject  hits = jsonObject.getJSONObject("hits");
        JSONArray hitsArray=hits.getJSONArray("hits");
        for(int i=0;i<hitsArray.length();i++) {
            JSONObject obj= hitsArray.getJSONObject(i);
            source = obj.getJSONObject("_source");
            id = Integer.parseInt(source.opt("id").toString());
            path = source.optString("path");
            filename = source.optString("filename");

        }

        JSONObject jsonBody = new JSONObject();
        jsonBody.put("id", id);
        jsonBody.put("path", path);
        jsonBody.put("filename", filename);
        System.out.println("Response --->"+jsonBody.toString());

        }

Использую ElasticSearch версии 6.2.3

1 Ответ

0 голосов
/ 27 июня 2018

Я нашел способ обработки нескольких документов из RestClient ответа 

             JSONArray hitsArray;
             JSONArray responseArray = new JSONArray();
             JSONObject responseObj;
             String fileRelativePath=null;

             JSONObject json = new JSONObject(responseBody);
             JSONObject hits = json.getJSONObject("hits");
             hitsArray = hits.getJSONArray("hits");
                for (int i=0; i<hitsArray.length(); i++) {
                JSONObject h = hitsArray.getJSONObject(i);
                JSONObject sourceJObj = h.getJSONObject("_source");
                        responseObj = new JSONObject();
                        fileRelativePath = sourceJObj.optString("path").toString().concat(sourceJObj.optString("filename").toString());
                        responseObj.put("id", Integer.parseInt(sourceJObj.opt("id").toString()));
                        responseObj.put("fileRelativePath", fileRelativePath);
                        responseArray.put(responseObj);

                }
                System.out.println("Response Length -->"+responseArray.length()+"Response --->"+responseArray.toString());
...