Examples with GetRequest org.opensearch.action.get.GetRequest used on opensource projects

Example 26 with GetRequest

use of org.opensearch.action.get.GetRequest in project OpenSearch by opensearch-project.

the class TransportGetTaskAction method getFinishedTaskFromIndex.

/**
 * Send a {@link GetRequest} to the tasks index looking for a persisted copy of the task completed task. It'll only be found only if the
 * task's result was stored. Called on the node that once had the task if that node is still part of the cluster or on the
 * coordinating node if the node is no longer part of the cluster.
 */
void getFinishedTaskFromIndex(Task thisTask, GetTaskRequest request, ActionListener<GetTaskResponse> listener) {
    GetRequest get = new GetRequest(TaskResultsService.TASK_INDEX, request.getTaskId().toString());
    get.setParentTask(clusterService.localNode().getId(), thisTask.getId());
    client.get(get, ActionListener.wrap(r -> onGetFinishedTaskFromIndex(r, listener), e -> {
        if (ExceptionsHelper.unwrap(e, IndexNotFoundException.class) != null) {
            // We haven't yet created the index for the task results so it can't be found.
            listener.onFailure(new ResourceNotFoundException("task [{}] isn't running and hasn't stored its results", e, request.getTaskId()));
        } else {
            listener.onFailure(e);
        }
    }));
}

Example 27 with GetRequest

use of org.opensearch.action.get.GetRequest in project OpenSearch by opensearch-project.

the class TermsQueryBuilder method fetch.

private void fetch(TermsLookup termsLookup, Client client, ActionListener<List<Object>> actionListener) {
    GetRequest getRequest = new GetRequest(termsLookup.index(), termsLookup.id());
    getRequest.preference("_local").routing(termsLookup.routing());
    client.get(getRequest, ActionListener.delegateFailure(actionListener, (delegatedListener, getResponse) -> {
        List<Object> terms = new ArrayList<>();
        if (getResponse.isSourceEmpty() == false) {
            // extract terms only if the doc source exists
            List<Object> extractedValues = XContentMapValues.extractRawValues(termsLookup.path(), getResponse.getSourceAsMap());
            terms.addAll(extractedValues);
        }
        delegatedListener.onResponse(terms);
    }));
}

Example 28 with GetRequest

use of org.opensearch.action.get.GetRequest in project OpenSearch by opensearch-project.

the class RequestConvertersTests method doTestSourceExists.

private static void doTestSourceExists(BiFunction<String, String, GetSourceRequest> requestFunction) throws IOException {
    String index = randomAlphaOfLengthBetween(3, 10);
    String id = randomAlphaOfLengthBetween(3, 10);
    final GetSourceRequest getRequest = requestFunction.apply(index, id);
    Map<String, String> expectedParams = new HashMap<>();
    if (randomBoolean()) {
        String preference = randomAlphaOfLengthBetween(3, 10);
        getRequest.preference(preference);
        expectedParams.put("preference", preference);
    }
    if (randomBoolean()) {
        String routing = randomAlphaOfLengthBetween(3, 10);
        getRequest.routing(routing);
        expectedParams.put("routing", routing);
    }
    if (randomBoolean()) {
        boolean realtime = randomBoolean();
        getRequest.realtime(realtime);
        if (realtime == false) {
            expectedParams.put("realtime", "false");
        }
    }
    if (randomBoolean()) {
        boolean refresh = randomBoolean();
        getRequest.refresh(refresh);
        if (refresh) {
            expectedParams.put("refresh", "true");
        }
    }
    Request request = RequestConverters.sourceExists(getRequest);
    assertEquals(HttpHead.METHOD_NAME, request.getMethod());
    assertEquals("/" + index + "/_source/" + id, request.getEndpoint());
    assertEquals(expectedParams, request.getParameters());
    assertNull(request.getEntity());
}

Example 29 with GetRequest

use of org.opensearch.action.get.GetRequest in project OpenSearch by opensearch-project.

the class RequestConvertersTests method getAndExistsTest.

private static void getAndExistsTest(Function<GetRequest, Request> requestConverter, String method) {
    String index = randomAlphaOfLengthBetween(3, 10);
    String id = randomAlphaOfLengthBetween(3, 10);
    GetRequest getRequest = new GetRequest(index, id);
    Map<String, String> expectedParams = new HashMap<>();
    if (randomBoolean()) {
        if (randomBoolean()) {
            String preference = randomAlphaOfLengthBetween(3, 10);
            getRequest.preference(preference);
            expectedParams.put("preference", preference);
        }
        if (randomBoolean()) {
            String routing = randomAlphaOfLengthBetween(3, 10);
            getRequest.routing(routing);
            expectedParams.put("routing", routing);
        }
        if (randomBoolean()) {
            boolean realtime = randomBoolean();
            getRequest.realtime(realtime);
            if (realtime == false) {
                expectedParams.put("realtime", "false");
            }
        }
        if (randomBoolean()) {
            boolean refresh = randomBoolean();
            getRequest.refresh(refresh);
            if (refresh) {
                expectedParams.put("refresh", "true");
            }
        }
        if (randomBoolean()) {
            long version = randomLong();
            getRequest.version(version);
            if (version != Versions.MATCH_ANY) {
                expectedParams.put("version", Long.toString(version));
            }
        }
        setRandomVersionType(getRequest::versionType, expectedParams);
        if (randomBoolean()) {
            int numStoredFields = randomIntBetween(1, 10);
            String[] storedFields = new String[numStoredFields];
            String storedFieldsParam = randomFields(storedFields);
            getRequest.storedFields(storedFields);
            expectedParams.put("stored_fields", storedFieldsParam);
        }
        if (randomBoolean()) {
            randomizeFetchSourceContextParams(getRequest::fetchSourceContext, expectedParams);
        }
    }
    Request request = requestConverter.apply(getRequest);
    assertEquals("/" + index + "/_doc/" + id, request.getEndpoint());
    assertEquals(expectedParams, request.getParameters());
    assertNull(request.getEntity());
    assertEquals(method, request.getMethod());
}

Example 30 with GetRequest

use of org.opensearch.action.get.GetRequest in project OpenSearch by opensearch-project.

the class CrudIT method testExists.

public void testExists() throws IOException {
    {
        GetRequest getRequest = new GetRequest("index", "id");
        assertFalse(execute(getRequest, highLevelClient()::exists, highLevelClient()::existsAsync));
    }
    IndexRequest index = new IndexRequest("index").id("id");
    index.source("{\"field1\":\"value1\",\"field2\":\"value2\"}", XContentType.JSON);
    index.setRefreshPolicy(RefreshPolicy.IMMEDIATE);
    highLevelClient().index(index, RequestOptions.DEFAULT);
    {
        GetRequest getRequest = new GetRequest("index", "id");
        assertTrue(execute(getRequest, highLevelClient()::exists, highLevelClient()::existsAsync));
    }
    {
        GetRequest getRequest = new GetRequest("index", "does_not_exist");
        assertFalse(execute(getRequest, highLevelClient()::exists, highLevelClient()::existsAsync));
    }
    {
        GetRequest getRequest = new GetRequest("index", "does_not_exist").version(1);
        assertFalse(execute(getRequest, highLevelClient()::exists, highLevelClient()::existsAsync));
    }
}