use of org.opensearch.client.opensearch._types.Result in project opensearch-java by opensearch-project.
the class CrudIT method testBulk.
public void testBulk() throws IOException {
AppData appData = new AppData();
appData.setIntValue(1337);
appData.setMsg("foo");
int nbItems = randomIntBetween(10, 100);
boolean[] errors = new boolean[nbItems];
List<BulkOperation> opsList = new ArrayList<>();
for (int i = 0; i < nbItems; i++) {
String id = String.valueOf(i);
boolean erroneous = randomBoolean();
errors[i] = erroneous;
BulkOperation.Kind opType = randomFrom(BulkOperation.Kind.Delete, BulkOperation.Kind.Index, BulkOperation.Kind.Create);
if (opType.equals(BulkOperation.Kind.Delete)) {
if (!erroneous) {
assertEquals(Result.Created, highLevelClient().index(b -> b.index("index").id(id).document(appData)).result());
}
BulkOperation op = new BulkOperation.Builder().delete(d -> d.index("index").id(id)).build();
opsList.add(op);
} else {
appData.setIntValue(i);
appData.setMsg("id");
if (opType.equals(BulkOperation.Kind.Index)) {
BulkOperation op = new BulkOperation.Builder().index(b -> b.index("index").id(id).document(appData).ifSeqNo(erroneous ? 12L : null).ifPrimaryTerm(erroneous ? 12L : null)).build();
opsList.add(op);
} else if (opType.equals(BulkOperation.Kind.Create)) {
BulkOperation op = new BulkOperation.Builder().create(o -> o.index("index").id(id).document(appData)).build();
opsList.add(op);
if (erroneous) {
assertEquals(Result.Created, highLevelClient().index(b -> b.index("index").id(id).document(appData)).result());
}
} else if (opType.equals(BulkOperation.Kind.Update)) {
BulkOperation op = new BulkOperation.Builder().update(o -> o.index("index").id(id).document(Collections.singletonMap("key", "value"))).build();
opsList.add(op);
if (!erroneous) {
assertEquals(Result.Created, highLevelClient().index(b -> b.index("index").id(id).document(appData)).result());
}
}
}
}
BulkRequest bulkRequest = new BulkRequest.Builder().operations(opsList).build();
BulkResponse bulkResponse = highLevelClient().bulk(bulkRequest);
assertTrue(bulkResponse.took() > 0);
assertEquals(nbItems, bulkResponse.items().size());
validateBulkResponses(nbItems, errors, bulkResponse, bulkRequest);
}
use of org.opensearch.client.opensearch._types.Result in project opensearch-java by opensearch-project.
the class RequestTest method testDataIngestion.
@Test
public void testDataIngestion() throws Exception {
String index = "ingest-test";
// Create an index
CreateIndexResponse createIndexResponse = highLevelClient().indices().create(b -> b.index(index));
assertEquals(index, createIndexResponse.index());
// Check that it actually exists. Example of a boolean response
BooleanResponse existsResponse = highLevelClient().indices().exists(b -> b.index(index));
assertTrue(existsResponse.value());
// Ingest some data
AppData appData = new AppData();
appData.setIntValue(1337);
appData.setMsg("foo");
String docId = highLevelClient().index(b -> b.index(index).id(// test with url-unsafe string
"my/Id").document(appData).refresh(// Make it visible for search
Refresh.True)).id();
assertEquals("my/Id", docId);
// Check auto-created mapping
GetMappingResponse mapping = highLevelClient().indices().getMapping(b -> b.index(index));
assertEquals(Property.Kind.Long, mapping.get("ingest-test").mappings().properties().get("intValue")._kind());
// Query by id
AppData esData = highLevelClient().get(b -> b.index(index).id(docId), AppData.class).source();
assertEquals(1337, esData.getIntValue());
assertEquals("foo", esData.getMsg());
// Query by id a non-existing document
final GetResponse<AppData> notExists = highLevelClient().get(b -> b.index(index).id("some-random-id"), AppData.class);
assertFalse(notExists.found());
assertNull(notExists.source());
// Search
SearchResponse<AppData> search = highLevelClient().search(b -> b.index(index), AppData.class);
long hits = search.hits().total().value();
assertEquals(1, hits);
esData = search.hits().hits().get(0).source();
assertEquals(1337, esData.getIntValue());
assertEquals("foo", esData.getMsg());
RequestItem item = RequestItem.of(_1 -> _1.header(_2 -> _2.index("test")).body(_2 -> _2.size(4)));
// MSearch: 1st search on an existing index, 2nd one on a non-existing index
final MsearchResponse<AppData> msearch = highLevelClient().msearch(_0 -> _0.searches(_1 -> _1.header(_3 -> _3.index(index)).body(_3 -> _3.query(_4 -> _4.matchAll(_5 -> _5)))).searches(_1 -> _1.header(_3 -> _3.index("non-existing")).body(_3 -> _3.query(_4 -> _4.matchAll(_5 -> _5)))), AppData.class);
assertEquals(2, msearch.responses().size());
assertTrue(msearch.responses().get(0).isResult());
assertEquals(1, msearch.responses().get(0).result().hits().hits().size());
assertTrue(msearch.responses().get(1).isFailure());
assertEquals(404, msearch.responses().get(1).failure().status());
}
Aggregations