use of org.opensearch.client.opensearch.core.BulkResponse 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.core.BulkResponse in project opensearch-java by opensearch-project.
the class RequestTest method testBulkRequest.
@Test
public void testBulkRequest() throws IOException {
AppData appData = new AppData();
appData.setIntValue(42);
appData.setMsg("Some message");
BulkResponse bulk = highLevelClient().bulk(_0 -> _0.operations(_1 -> _1.create(_2 -> _2.index("foo").id("abc").document(appData))).operations(_1 -> _1.create(_2 -> _2.index("foo").id("def").document(appData))));
assertFalse(bulk.errors());
assertEquals(2, bulk.items().size());
assertEquals(OperationType.Create, bulk.items().get(0).operationType());
assertEquals("foo", bulk.items().get(0).index());
assertEquals(1L, bulk.items().get(0).version().longValue());
assertEquals("foo", bulk.items().get(1).index());
assertEquals(1L, bulk.items().get(1).version().longValue());
assertEquals(42, highLevelClient().get(b -> b.index("foo").id("abc"), AppData.class).source().getIntValue());
}
Aggregations