Examples with BulkRequest org.opensearch.action.bulk.BulkRequest used on opensource projects

Example 11 with BulkRequest

use of org.opensearch.action.bulk.BulkRequest in project OpenSearch by opensearch-project.

the class CrudIT method testMultiGetWithIds.

public void testMultiGetWithIds() throws IOException {
    BulkRequest bulk = new BulkRequest();
    bulk.setRefreshPolicy(RefreshPolicy.IMMEDIATE);
    bulk.add(new IndexRequest("index").id("id1").source("{\"field\":\"value1\"}", XContentType.JSON));
    bulk.add(new IndexRequest("index").id("id2").source("{\"field\":\"value2\"}", XContentType.JSON));
    MultiGetRequest multiGetRequest = new MultiGetRequest();
    multiGetRequest.add("index", "id1");
    multiGetRequest.add("index", "id2");
}

Example 12 with BulkRequest

use of org.opensearch.action.bulk.BulkRequest in project OpenSearch by opensearch-project.

the class CrudIT method testMultiGet.

public void testMultiGet() throws IOException {
    {
        MultiGetRequest multiGetRequest = new MultiGetRequest();
        multiGetRequest.add("index", "id1");
        multiGetRequest.add("index", "id2");
        MultiGetResponse response = execute(multiGetRequest, highLevelClient()::mget, highLevelClient()::mgetAsync);
        assertEquals(2, response.getResponses().length);
        assertTrue(response.getResponses()[0].isFailed());
        assertNull(response.getResponses()[0].getResponse());
        assertEquals("id1", response.getResponses()[0].getFailure().getId());
        assertEquals("index", response.getResponses()[0].getFailure().getIndex());
        assertEquals("OpenSearch exception [type=index_not_found_exception, reason=no such index [index]]", response.getResponses()[0].getFailure().getFailure().getMessage());
        assertTrue(response.getResponses()[1].isFailed());
        assertNull(response.getResponses()[1].getResponse());
        assertEquals("id2", response.getResponses()[1].getId());
        assertEquals("index", response.getResponses()[1].getIndex());
        assertEquals("OpenSearch exception [type=index_not_found_exception, reason=no such index [index]]", response.getResponses()[1].getFailure().getFailure().getMessage());
    }
    BulkRequest bulk = new BulkRequest();
    bulk.setRefreshPolicy(RefreshPolicy.IMMEDIATE);
    IndexRequest index = new IndexRequest("index").id("id1");
    index.source("{\"field\":\"value1\"}", XContentType.JSON);
    bulk.add(index);
    index = new IndexRequest("index").id("id2");
    index.source("{\"field\":\"value2\"}", XContentType.JSON);
    bulk.add(index);
    highLevelClient().bulk(bulk, RequestOptions.DEFAULT);
    {
        MultiGetRequest multiGetRequest = new MultiGetRequest();
        multiGetRequest.add("index", "id1");
        multiGetRequest.add("index", "id2");
        MultiGetResponse response = execute(multiGetRequest, highLevelClient()::mget, highLevelClient()::mgetAsync);
        assertEquals(2, response.getResponses().length);
        assertFalse(response.getResponses()[0].isFailed());
        assertNull(response.getResponses()[0].getFailure());
        assertEquals("id1", response.getResponses()[0].getId());
        assertEquals("index", response.getResponses()[0].getIndex());
        assertEquals(Collections.singletonMap("field", "value1"), response.getResponses()[0].getResponse().getSource());
        assertFalse(response.getResponses()[1].isFailed());
        assertNull(response.getResponses()[1].getFailure());
        assertEquals("id2", response.getResponses()[1].getId());
        assertEquals("index", response.getResponses()[1].getIndex());
        assertEquals(Collections.singletonMap("field", "value2"), response.getResponses()[1].getResponse().getSource());
    }
}

Example 13 with BulkRequest

use of org.opensearch.action.bulk.BulkRequest in project OpenSearch by opensearch-project.

the class UpdateByQueryIT method testUpdateByQueryConflict.

public void testUpdateByQueryConflict() throws IOException {
    final String index = "testupdatebyqueryconflict";
    final Settings settings = Settings.builder().put("number_of_shards", 1).put("number_of_replicas", 0).build();
    createIndex(index, settings);
    final BulkRequest bulkRequest = new BulkRequest().add(new IndexRequest(index).id("1").source(Collections.singletonMap("foo", "bar"), XContentType.JSON)).add(new IndexRequest(index).id("2").source(Collections.singletonMap("foo", "bar"), XContentType.JSON)).setRefreshPolicy(WriteRequest.RefreshPolicy.IMMEDIATE);
    assertThat(highLevelClient().bulk(bulkRequest, RequestOptions.DEFAULT).status(), equalTo(RestStatus.OK));
    putConflictPipeline();
    final UpdateByQueryRequest updateByQueryRequest = new UpdateByQueryRequest();
    updateByQueryRequest.indices(index);
    updateByQueryRequest.setRefresh(true);
    updateByQueryRequest.setPipeline(CONFLICT_PIPELINE_ID);
    final BulkByScrollResponse response = highLevelClient().updateByQuery(updateByQueryRequest, RequestOptions.DEFAULT);
    assertThat(response.getVersionConflicts(), equalTo(1L));
    assertThat(response.getSearchFailures(), empty());
    assertThat(response.getBulkFailures(), hasSize(1));
    assertThat(response.getBulkFailures().stream().map(BulkItemResponse.Failure::getMessage).collect(Collectors.toSet()), everyItem(containsString("version conflict")));
    assertThat(response.getTotal(), equalTo(2L));
    assertThat(response.getCreated(), equalTo(0L));
    assertThat(response.getUpdated(), equalTo(1L));
    assertThat(response.getDeleted(), equalTo(0L));
    assertThat(response.getNoops(), equalTo(0L));
    assertThat(response.getBatches(), equalTo(1));
    assertTrue(response.getTook().getMillis() > 0);
}

Example 14 with BulkRequest

use of org.opensearch.action.bulk.BulkRequest in project OpenSearch by opensearch-project.

the class SearchDocumentationIT method indexSearchTestData.

private void indexSearchTestData() throws IOException {
    CreateIndexRequest authorsRequest = new CreateIndexRequest("authors").mapping(XContentFactory.jsonBuilder().startObject().startObject("properties").startObject("id").field("type", "keyword").endObject().startObject("user").field("type", "keyword").field("doc_values", "false").endObject().endObject().endObject());
    CreateIndexResponse authorsResponse = highLevelClient().indices().create(authorsRequest, RequestOptions.DEFAULT);
    assertTrue(authorsResponse.isAcknowledged());
    CreateIndexRequest reviewersRequest = new CreateIndexRequest("contributors").mapping(XContentFactory.jsonBuilder().startObject().startObject("properties").startObject("id").field("type", "keyword").endObject().startObject("user").field("type", "keyword").field("store", "true").endObject().endObject().endObject());
    CreateIndexResponse reviewersResponse = highLevelClient().indices().create(reviewersRequest, RequestOptions.DEFAULT);
    assertTrue(reviewersResponse.isAcknowledged());
    BulkRequest bulkRequest = new BulkRequest();
    bulkRequest.add(new IndexRequest("posts").id("1").source(XContentType.JSON, "id", 1, "title", "In which order are my OpenSearch queries executed?", "user", Arrays.asList("foobar", "quxx"), "innerObject", Collections.singletonMap("key", "value")));
    bulkRequest.add(new IndexRequest("posts").id("2").source(XContentType.JSON, "id", 2, "title", "Current status and upcoming changes in OpenSearch", "user", Arrays.asList("foobar", "grault"), "innerObject", Collections.singletonMap("key", "value")));
    bulkRequest.add(new IndexRequest("posts").id("3").source(XContentType.JSON, "id", 3, "title", "The Future of Federated Search in OpenSearch", "user", Arrays.asList("foobar", "quuz"), "innerObject", Collections.singletonMap("key", "value")));
    bulkRequest.add(new IndexRequest("authors").id("1").source(XContentType.JSON, "id", 1, "user", "foobar"));
    bulkRequest.add(new IndexRequest("contributors").id("1").source(XContentType.JSON, "id", 1, "user", "quuz"));
    bulkRequest.setRefreshPolicy(WriteRequest.RefreshPolicy.IMMEDIATE);
    BulkResponse bulkResponse = highLevelClient().bulk(bulkRequest, RequestOptions.DEFAULT);
    assertSame(RestStatus.OK, bulkResponse.status());
    assertFalse(bulkResponse.hasFailures());
}

Example 15 with BulkRequest

use of org.opensearch.action.bulk.BulkRequest in project OpenSearch by opensearch-project.

the class SearchDocumentationIT method testSearchRequestSuggestions.

@SuppressWarnings({ "unused", "rawtypes" })
public void testSearchRequestSuggestions() throws IOException {
    RestHighLevelClient client = highLevelClient();
    {
        BulkRequest request = new BulkRequest();
        request.add(new IndexRequest("posts").id("1").source(XContentType.JSON, "user", "foobar"));
        request.add(new IndexRequest("posts").id("2").source(XContentType.JSON, "user", "quxx"));
        request.add(new IndexRequest("posts").id("3").source(XContentType.JSON, "user", "quzz"));
        request.add(new IndexRequest("posts").id("4").source(XContentType.JSON, "user", "corge"));
        request.setRefreshPolicy(WriteRequest.RefreshPolicy.IMMEDIATE);
        BulkResponse bulkResponse = client.bulk(request, RequestOptions.DEFAULT);
        assertSame(RestStatus.OK, bulkResponse.status());
        assertFalse(bulkResponse.hasFailures());
    }
    {
        SearchRequest searchRequest = new SearchRequest();
        // tag::search-request-suggestion
        SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder();
        SuggestionBuilder termSuggestionBuilder = // <1>
        SuggestBuilders.termSuggestion("user").text("fooarb");
        SuggestBuilder suggestBuilder = new SuggestBuilder();
        // <2>
        suggestBuilder.addSuggestion("suggest_user", termSuggestionBuilder);
        searchSourceBuilder.suggest(suggestBuilder);
        // end::search-request-suggestion
        searchRequest.source(searchSourceBuilder);
        SearchResponse searchResponse = client.search(searchRequest, RequestOptions.DEFAULT);
        {
            // tag::search-request-suggestion-get
            // <1>
            Suggest suggest = searchResponse.getSuggest();
            // <2>
            TermSuggestion termSuggestion = suggest.getSuggestion("suggest_user");
            for (TermSuggestion.Entry entry : termSuggestion.getEntries()) {
                // <3>
                for (TermSuggestion.Entry.Option option : entry) {
                    // <4>
                    String suggestText = option.getText().string();
                }
            }
            // end::search-request-suggestion-get
            assertEquals(1, termSuggestion.getEntries().size());
            assertEquals(1, termSuggestion.getEntries().get(0).getOptions().size());
            assertEquals("foobar", termSuggestion.getEntries().get(0).getOptions().get(0).getText().string());
        }
    }
}