Examples with DeleteRequest org.opensearch.action.delete.DeleteRequest used on opensource projects

Example 1 with DeleteRequest

use of org.opensearch.action.delete.DeleteRequest in project veilarbportefolje by navikt.

the class OpensearchIndexerV2 method delete.

@SneakyThrows
private void delete(AktorId aktoerId) {
    DeleteRequest deleteRequest = new DeleteRequest();
    deleteRequest.index(indexName.getValue());
    deleteRequest.id(aktoerId.get());
    try {
        restHighLevelClient.delete(deleteRequest, RequestOptions.DEFAULT);
        log.info("Slettet dokument for {} ", aktoerId);
    } catch (OpenSearchException e) {
        if (e.status() == RestStatus.NOT_FOUND) {
            log.info("Kunne ikke finne dokument for bruker {} ved sletting av indeks", aktoerId.get());
        } else {
            final String message = format("Det skjedde en feil ved sletting i opensearch for bruker %s", aktoerId.get());
            log.error(message, e);
        }
    }
}

Example 2 with DeleteRequest

use of org.opensearch.action.delete.DeleteRequest in project OpenSearch by opensearch-project.

the class RequestConverters method delete.

static Request delete(DeleteRequest deleteRequest) {
    String endpoint = endpoint(deleteRequest.index(), deleteRequest.id());
    Request request = new Request(HttpDelete.METHOD_NAME, endpoint);
    Params parameters = new Params();
    parameters.withRouting(deleteRequest.routing());
    parameters.withTimeout(deleteRequest.timeout());
    parameters.withVersion(deleteRequest.version());
    parameters.withVersionType(deleteRequest.versionType());
    parameters.withIfSeqNo(deleteRequest.ifSeqNo());
    parameters.withIfPrimaryTerm(deleteRequest.ifPrimaryTerm());
    parameters.withRefreshPolicy(deleteRequest.getRefreshPolicy());
    parameters.withWaitForActiveShards(deleteRequest.waitForActiveShards());
    request.addParameters(parameters.asMap());
    return request;
}

Example 3 with DeleteRequest

use of org.opensearch.action.delete.DeleteRequest in project OpenSearch by opensearch-project.

the class RequestConvertersTests method testDelete.

public void testDelete() {
    String index = randomAlphaOfLengthBetween(3, 10);
    String id = randomAlphaOfLengthBetween(3, 10);
    DeleteRequest deleteRequest = new DeleteRequest(index, id);
    Map<String, String> expectedParams = new HashMap<>();
    setRandomTimeout(deleteRequest::timeout, ReplicationRequest.DEFAULT_TIMEOUT, expectedParams);
    setRandomRefreshPolicy(deleteRequest::setRefreshPolicy, expectedParams);
    setRandomVersion(deleteRequest, expectedParams);
    setRandomVersionType(deleteRequest::versionType, expectedParams);
    setRandomIfSeqNoAndTerm(deleteRequest, expectedParams);
    if (frequently()) {
        if (randomBoolean()) {
            String routing = randomAlphaOfLengthBetween(3, 10);
            deleteRequest.routing(routing);
            expectedParams.put("routing", routing);
        }
    }
    Request request = RequestConverters.delete(deleteRequest);
    assertEquals(HttpDelete.METHOD_NAME, request.getMethod());
    assertEquals("/" + index + "/_doc/" + id, request.getEndpoint());
    assertEquals(expectedParams, request.getParameters());
    assertNull(request.getEntity());
}

Example 4 with DeleteRequest

use of org.opensearch.action.delete.DeleteRequest in project OpenSearch by opensearch-project.

the class RequestConvertersTests method testBulk.

public void testBulk() throws IOException {
    Map<String, String> expectedParams = new HashMap<>();
    BulkRequest bulkRequest = new BulkRequest();
    if (randomBoolean()) {
        String timeout = randomTimeValue();
        bulkRequest.timeout(timeout);
        expectedParams.put("timeout", timeout);
    } else {
        expectedParams.put("timeout", BulkShardRequest.DEFAULT_TIMEOUT.getStringRep());
    }
    setRandomRefreshPolicy(bulkRequest::setRefreshPolicy, expectedParams);
    XContentType xContentType = randomFrom(XContentType.JSON, XContentType.SMILE);
    int nbItems = randomIntBetween(10, 100);
    DocWriteRequest<?>[] requests = new DocWriteRequest<?>[nbItems];
    for (int i = 0; i < nbItems; i++) {
        String index = randomAlphaOfLength(5);
        String id = randomAlphaOfLength(5);
        BytesReference source = RandomObjects.randomSource(random(), xContentType);
        DocWriteRequest.OpType opType = randomFrom(DocWriteRequest.OpType.values());
        DocWriteRequest<?> docWriteRequest;
        if (opType == DocWriteRequest.OpType.INDEX) {
            IndexRequest indexRequest = new IndexRequest(index).id(id).source(source, xContentType);
            docWriteRequest = indexRequest;
            if (randomBoolean()) {
                indexRequest.setPipeline(randomAlphaOfLength(5));
            }
        } else if (opType == DocWriteRequest.OpType.CREATE) {
            IndexRequest createRequest = new IndexRequest(index).id(id).source(source, xContentType).create(true);
            docWriteRequest = createRequest;
        } else if (opType == DocWriteRequest.OpType.UPDATE) {
            final UpdateRequest updateRequest = new UpdateRequest(index, id).doc(new IndexRequest().source(source, xContentType));
            docWriteRequest = updateRequest;
            if (randomBoolean()) {
                updateRequest.retryOnConflict(randomIntBetween(1, 5));
            }
            if (randomBoolean()) {
                randomizeFetchSourceContextParams(updateRequest::fetchSource, new HashMap<>());
            }
        } else if (opType == DocWriteRequest.OpType.DELETE) {
            docWriteRequest = new DeleteRequest(index, id);
        } else {
            throw new UnsupportedOperationException("optype [" + opType + "] not supported");
        }
        if (randomBoolean()) {
            docWriteRequest.routing(randomAlphaOfLength(10));
        }
        if (opType != DocWriteRequest.OpType.UPDATE && randomBoolean()) {
            docWriteRequest.setIfSeqNo(randomNonNegativeLong());
            docWriteRequest.setIfPrimaryTerm(randomLongBetween(1, 200));
        }
        requests[i] = docWriteRequest;
    }
    bulkRequest.add(requests);
    Request request = RequestConverters.bulk(bulkRequest);
    assertEquals("/_bulk", request.getEndpoint());
    assertEquals(expectedParams, request.getParameters());
    assertEquals(HttpPost.METHOD_NAME, request.getMethod());
    assertEquals(xContentType.mediaTypeWithoutParameters(), request.getEntity().getContentType().getValue());
    byte[] content = new byte[(int) request.getEntity().getContentLength()];
    try (InputStream inputStream = request.getEntity().getContent()) {
        Streams.readFully(inputStream, content);
    }
    BulkRequest parsedBulkRequest = new BulkRequest();
    parsedBulkRequest.add(content, 0, content.length, xContentType);
    assertEquals(bulkRequest.numberOfActions(), parsedBulkRequest.numberOfActions());
    for (int i = 0; i < bulkRequest.numberOfActions(); i++) {
        DocWriteRequest<?> originalRequest = bulkRequest.requests().get(i);
        DocWriteRequest<?> parsedRequest = parsedBulkRequest.requests().get(i);
        assertEquals(originalRequest.opType(), parsedRequest.opType());
        assertEquals(originalRequest.index(), parsedRequest.index());
        assertEquals(originalRequest.id(), parsedRequest.id());
        assertEquals(originalRequest.routing(), parsedRequest.routing());
        assertEquals(originalRequest.version(), parsedRequest.version());
        assertEquals(originalRequest.versionType(), parsedRequest.versionType());
        assertEquals(originalRequest.ifSeqNo(), parsedRequest.ifSeqNo());
        assertEquals(originalRequest.ifPrimaryTerm(), parsedRequest.ifPrimaryTerm());
        DocWriteRequest.OpType opType = originalRequest.opType();
        if (opType == DocWriteRequest.OpType.INDEX) {
            IndexRequest indexRequest = (IndexRequest) originalRequest;
            IndexRequest parsedIndexRequest = (IndexRequest) parsedRequest;
            assertEquals(indexRequest.getPipeline(), parsedIndexRequest.getPipeline());
            assertToXContentEquivalent(indexRequest.source(), parsedIndexRequest.source(), xContentType);
        } else if (opType == DocWriteRequest.OpType.UPDATE) {
            UpdateRequest updateRequest = (UpdateRequest) originalRequest;
            UpdateRequest parsedUpdateRequest = (UpdateRequest) parsedRequest;
            assertEquals(updateRequest.retryOnConflict(), parsedUpdateRequest.retryOnConflict());
            assertEquals(updateRequest.fetchSource(), parsedUpdateRequest.fetchSource());
            if (updateRequest.doc() != null) {
                assertToXContentEquivalent(updateRequest.doc().source(), parsedUpdateRequest.doc().source(), xContentType);
            } else {
                assertNull(parsedUpdateRequest.doc());
            }
        }
    }
}

Example 5 with DeleteRequest

use of org.opensearch.action.delete.DeleteRequest in project OpenSearch by opensearch-project.

the class CrudIT method testBulkProcessorIntegration.

public void testBulkProcessorIntegration() throws IOException {
    int nbItems = randomIntBetween(10, 100);
    boolean[] errors = new boolean[nbItems];
    XContentType xContentType = randomFrom(XContentType.JSON, XContentType.SMILE);
    AtomicReference<BulkResponse> responseRef = new AtomicReference<>();
    AtomicReference<BulkRequest> requestRef = new AtomicReference<>();
    AtomicReference<Throwable> error = new AtomicReference<>();
    BulkProcessor.Listener listener = new BulkProcessor.Listener() {

        @Override
        public void beforeBulk(long executionId, BulkRequest request) {
        }

        @Override
        public void afterBulk(long executionId, BulkRequest request, BulkResponse response) {
            responseRef.set(response);
            requestRef.set(request);
        }

        @Override
        public void afterBulk(long executionId, BulkRequest request, Throwable failure) {
            error.set(failure);
        }
    };
    try (BulkProcessor processor = BulkProcessor.builder((request, bulkListener) -> highLevelClient().bulkAsync(request, RequestOptions.DEFAULT, bulkListener), listener).setConcurrentRequests(0).setBulkSize(new ByteSizeValue(5, ByteSizeUnit.GB)).setBulkActions(nbItems + 1).build()) {
        for (int i = 0; i < nbItems; i++) {
            String id = String.valueOf(i);
            boolean erroneous = randomBoolean();
            errors[i] = erroneous;
            DocWriteRequest.OpType opType = randomFrom(DocWriteRequest.OpType.values());
            if (opType == DocWriteRequest.OpType.DELETE) {
                if (erroneous == false) {
                    assertEquals(RestStatus.CREATED, highLevelClient().index(new IndexRequest("index").id(id).source("field", -1), RequestOptions.DEFAULT).status());
                }
                DeleteRequest deleteRequest = new DeleteRequest("index", id);
                processor.add(deleteRequest);
            } else {
                if (opType == DocWriteRequest.OpType.INDEX) {
                    IndexRequest indexRequest = new IndexRequest("index").id(id).source(xContentType, "id", i);
                    if (erroneous) {
                        indexRequest.setIfSeqNo(12L);
                        indexRequest.setIfPrimaryTerm(12L);
                    }
                    processor.add(indexRequest);
                } else if (opType == DocWriteRequest.OpType.CREATE) {
                    IndexRequest createRequest = new IndexRequest("index").id(id).source(xContentType, "id", i).create(true);
                    if (erroneous) {
                        assertEquals(RestStatus.CREATED, highLevelClient().index(createRequest, RequestOptions.DEFAULT).status());
                    }
                    processor.add(createRequest);
                } else if (opType == DocWriteRequest.OpType.UPDATE) {
                    UpdateRequest updateRequest = new UpdateRequest("index", id).doc(new IndexRequest().source(xContentType, "id", i));
                    if (erroneous == false) {
                        assertEquals(RestStatus.CREATED, highLevelClient().index(new IndexRequest("index").id(id).source("field", -1), RequestOptions.DEFAULT).status());
                    }
                    processor.add(updateRequest);
                }
            }
        }
        assertNull(responseRef.get());
        assertNull(requestRef.get());
    }
    BulkResponse bulkResponse = responseRef.get();
    BulkRequest bulkRequest = requestRef.get();
    assertEquals(RestStatus.OK, bulkResponse.status());
    assertTrue(bulkResponse.getTook().getMillis() > 0);
    assertEquals(nbItems, bulkResponse.getItems().length);
    assertNull(error.get());
    validateBulkResponses(nbItems, errors, bulkResponse, bulkRequest);
}