use of org.elasticsearch.client.RestHighLevelClient in project flink by apache.
the class ElasticsearchDynamicSinkBaseITCase method testWritingDocuments.
@Test
public void testWritingDocuments() throws Exception {
ResolvedSchema schema = new ResolvedSchema(Arrays.asList(Column.physical("a", DataTypes.BIGINT().notNull()), Column.physical("b", DataTypes.TIME()), Column.physical("c", DataTypes.STRING().notNull()), Column.physical("d", DataTypes.FLOAT()), Column.physical("e", DataTypes.TINYINT().notNull()), Column.physical("f", DataTypes.DATE()), Column.physical("g", DataTypes.TIMESTAMP().notNull())), Collections.emptyList(), UniqueConstraint.primaryKey("name", Arrays.asList("a", "g")));
GenericRowData rowData = GenericRowData.of(1L, 12345, StringData.fromString("ABCDE"), 12.12f, (byte) 2, 12345, TimestampData.fromLocalDateTime(LocalDateTime.parse("2012-12-12T12:12:12")));
String index = "writing-documents";
ElasticsearchDynamicSinkFactoryBase sinkFactory = getDynamicSinkFactory();
DynamicTableSink.SinkRuntimeProvider runtimeProvider = sinkFactory.createDynamicTableSink(getPrefilledTestContext(index).withSchema(schema).build()).getSinkRuntimeProvider(new ElasticsearchUtil.MockContext());
final SinkV2Provider sinkProvider = (SinkV2Provider) runtimeProvider;
final Sink<RowData> sink = sinkProvider.createSink();
StreamExecutionEnvironment environment = StreamExecutionEnvironment.getExecutionEnvironment();
environment.setParallelism(4);
rowData.setRowKind(RowKind.UPDATE_AFTER);
environment.<RowData>fromElements(rowData).sinkTo(sink);
environment.execute();
RestHighLevelClient client = getClient();
Map<String, Object> response = makeGetRequest(client, index, "1_2012-12-12T12:12:12");
Map<Object, Object> expectedMap = new HashMap<>();
expectedMap.put("a", 1);
expectedMap.put("b", "00:00:12");
expectedMap.put("c", "ABCDE");
expectedMap.put("d", 12.12d);
expectedMap.put("e", 2);
expectedMap.put("f", "2003-10-20");
expectedMap.put("g", "2012-12-12 12:12:12");
Assertions.assertEquals(response, expectedMap);
}
use of org.elasticsearch.client.RestHighLevelClient in project flink by apache.
the class ElasticsearchWriterITCase method setUp.
@BeforeEach
void setUp() {
metricListener = new MetricListener();
client = new RestHighLevelClient(RestClient.builder(HttpHost.create(ES_CONTAINER.getHttpHostAddress())));
context = new TestClient(client);
}
use of org.elasticsearch.client.RestHighLevelClient in project drill by apache.
the class ElasticComplexTypesTest method prepareData.
private static void prepareData() throws IOException {
restHighLevelClient = new RestHighLevelClient(RestClient.builder(HttpHost.create(HOST)));
String indexName = "arr";
indexNames.add(indexName);
CreateIndexRequest createIndexRequest = new CreateIndexRequest(indexName);
restHighLevelClient.indices().create(createIndexRequest, RequestOptions.DEFAULT);
XContentBuilder builder = XContentFactory.jsonBuilder();
builder.startObject();
builder.field("string_arr", Arrays.asList("a", "b", "c", "d"));
builder.field("int_arr", Arrays.asList(1, 2, 3, 4, 0));
builder.field("nest_int_arr", Arrays.asList(Arrays.asList(1, 2), Arrays.asList(3, 4, 0)));
builder.endObject();
IndexRequest indexRequest = new IndexRequest(indexName).source(builder);
restHighLevelClient.index(indexRequest, RequestOptions.DEFAULT);
restHighLevelClient.indices().refresh(new RefreshRequest(indexName), RequestOptions.DEFAULT);
indexName = "map";
indexNames.add(indexName);
createIndexRequest = new CreateIndexRequest(indexName);
restHighLevelClient.indices().create(createIndexRequest, RequestOptions.DEFAULT);
builder = XContentFactory.jsonBuilder();
builder.startObject();
builder.field("prim_field", 321);
builder.field("nest_field", ImmutableMap.of("a", 123, "b", "abc"));
builder.field("more_nest_field", ImmutableMap.of("a", 123, "b", ImmutableMap.of("c", "abc")));
builder.field("map_arr", Collections.singletonList(ImmutableMap.of("a", 123, "b", ImmutableMap.of("c", "abc"))));
builder.endObject();
indexRequest = new IndexRequest(indexName).source(builder);
restHighLevelClient.index(indexRequest, RequestOptions.DEFAULT);
restHighLevelClient.indices().refresh(new RefreshRequest(indexName), RequestOptions.DEFAULT);
}
use of org.elasticsearch.client.RestHighLevelClient in project drill by apache.
the class ElasticSearchPlanTest method prepareData.
private static void prepareData() throws IOException {
restHighLevelClient = new RestHighLevelClient(RestClient.builder(HttpHost.create(HOST)));
indexName = "nation";
CreateIndexRequest createIndexRequest = new CreateIndexRequest(indexName);
restHighLevelClient.indices().create(createIndexRequest, RequestOptions.DEFAULT);
XContentBuilder builder = XContentFactory.jsonBuilder();
builder.startObject();
builder.field("n_nationkey", 0);
builder.field("n_name", "ALGERIA");
builder.field("n_regionkey", 1);
builder.endObject();
IndexRequest indexRequest = new IndexRequest(indexName).source(builder);
restHighLevelClient.index(indexRequest, RequestOptions.DEFAULT);
restHighLevelClient.indices().refresh(new RefreshRequest(indexName), RequestOptions.DEFAULT);
}
use of org.elasticsearch.client.RestHighLevelClient in project drill by apache.
the class ElasticSearchQueryTest method prepareData.
private static void prepareData() throws IOException {
restHighLevelClient = new RestHighLevelClient(RestClient.builder(HttpHost.create(HOST)));
indexName = "employee";
CreateIndexRequest createIndexRequest = new CreateIndexRequest(indexName);
restHighLevelClient.indices().create(createIndexRequest, RequestOptions.DEFAULT);
XContentBuilder builder = XContentFactory.jsonBuilder();
builder.startObject();
builder.field("employee_id", 1);
builder.field("full_name", "Sheri Nowmer");
builder.field("first_name", "Sheri");
builder.field("last_name", "Nowmer");
builder.field("position_id", 1);
builder.field("position_title", "President");
builder.field("store_id", 0);
builder.field("department_id", 1);
builder.field("birth_date", "1961-08-26");
builder.field("hire_date", "1994-12-01 00:00:00.0");
builder.field("salary", 80000.0);
builder.field("supervisor_id", 0);
builder.field("education_level", "Graduate Degree");
builder.field("marital_status", "S");
builder.field("gender", "F");
builder.field("management_role", "Senior Management");
builder.endObject();
IndexRequest indexRequest = new IndexRequest(indexName).source(builder);
restHighLevelClient.index(indexRequest, RequestOptions.DEFAULT);
builder = XContentFactory.jsonBuilder();
builder.startObject();
builder.field("employee_id", 2);
builder.field("full_name", "Derrick Whelply");
builder.field("first_name", "Derrick");
builder.field("last_name", "Whelply");
builder.field("position_id", 2);
builder.field("position_title", "VP Country Manager");
builder.field("store_id", 0);
builder.field("department_id", 1);
builder.field("birth_date", "1915-07-03");
builder.field("hire_date", "1994-12-01 00:00:00.0");
builder.field("salary", 40000.0);
builder.field("supervisor_id", 1);
builder.field("education_level", "Graduate Degree");
builder.field("marital_status", "M");
builder.field("gender", "M");
builder.field("management_role", "Senior Management");
builder.endObject();
indexRequest = new IndexRequest(indexName).source(builder);
restHighLevelClient.index(indexRequest, RequestOptions.DEFAULT);
builder = XContentFactory.jsonBuilder();
builder.startObject();
builder.field("employee_id", 4);
builder.field("full_name", "Michael Spence");
builder.field("first_name", "Michael");
builder.field("last_name", "Spence");
builder.field("position_id", 2);
builder.field("position_title", "VP Country Manager");
builder.field("store_id", 0);
builder.field("department_id", 1);
builder.field("birth_date", "1969-06-20");
builder.field("hire_date", "1998-01-01 00:00:00.0");
builder.field("salary", 40000.0);
builder.field("supervisor_id", 1);
builder.field("education_level", "Graduate Degree");
builder.field("marital_status", "S");
builder.field("gender", "M");
builder.field("management_role", "Senior Management");
builder.endObject();
indexRequest = new IndexRequest(indexName).source(builder);
restHighLevelClient.index(indexRequest, RequestOptions.DEFAULT);
builder = XContentFactory.jsonBuilder();
builder.startObject();
builder.field("employee_id", 5);
builder.field("full_name", "Maya Gutierrez");
builder.field("first_name", "Maya");
builder.field("last_name", "Gutierrez");
builder.field("position_id", 2);
builder.field("position_title", "VP Country Manager");
builder.field("store_id", 0);
builder.field("department_id", 1);
builder.field("birth_date", "1951-05-10");
builder.field("hire_date", "1998-01-01 00:00:00.0");
builder.field("salary", 35000.0);
builder.field("supervisor_id", 1);
builder.field("education_level", "Bachelors Degree");
builder.field("marital_status", "M");
builder.field("gender", "F");
builder.field("management_role", "Senior Management");
builder.endObject();
indexRequest = new IndexRequest(indexName).source(builder);
restHighLevelClient.index(indexRequest, RequestOptions.DEFAULT);
builder = XContentFactory.jsonBuilder();
builder.startObject();
builder.field("employee_id", 6);
builder.field("full_name", "Roberta Damstra");
builder.field("first_name", "Roberta");
builder.field("last_name", "Damstra");
builder.field("position_id", 3);
builder.field("position_title", "VP Information Systems");
builder.field("store_id", 0);
builder.field("department_id", 2);
builder.field("birth_date", "1942-10-08");
builder.field("hire_date", "1994-12-01 00:00:00.0");
builder.field("salary", 25000.0);
builder.field("supervisor_id", 1);
builder.field("education_level", "Bachelors Degree");
builder.field("marital_status", "M");
builder.field("gender", "F");
builder.field("management_role", "Senior Management");
builder.endObject();
indexRequest = new IndexRequest(indexName).source(builder);
restHighLevelClient.index(indexRequest, RequestOptions.DEFAULT);
builder = XContentFactory.jsonBuilder();
builder.startObject();
builder.field("employee_id", 7);
builder.field("full_name", "Rebecca Kanagaki");
builder.field("first_name", "Rebecca");
builder.field("last_name", "Kanagaki");
builder.field("position_id", 4);
builder.field("position_title", "VP Human Resources");
builder.field("store_id", 0);
builder.field("department_id", 3);
builder.field("birth_date", "1949-03-27");
builder.field("hire_date", "1994-12-01 00:00:00.0");
builder.field("salary", 15000.0);
builder.field("supervisor_id", 1);
builder.field("education_level", "Bachelors Degree");
builder.field("marital_status", "M");
builder.field("gender", "F");
builder.field("management_role", "Senior Management");
builder.endObject();
indexRequest = new IndexRequest(indexName).source(builder);
restHighLevelClient.index(indexRequest, RequestOptions.DEFAULT);
builder = XContentFactory.jsonBuilder();
builder.startObject();
builder.field("employee_id", 8);
builder.field("full_name", "Kim Brunner");
builder.field("first_name", "Kim");
builder.field("last_name", "Brunner");
builder.field("position_id", 11);
builder.field("position_title", "Store Manager");
builder.field("store_id", 9);
builder.field("department_id", 11);
builder.field("birth_date", "1922-08-10");
builder.field("hire_date", "1998-01-01 00:00:00.0");
builder.field("salary", 10000.0);
builder.field("supervisor_id", 5);
builder.field("education_level", "Bachelors Degree");
builder.field("marital_status", "S");
builder.field("gender", "F");
builder.field("management_role", "Store Management");
builder.endObject();
indexRequest = new IndexRequest(indexName).source(builder);
restHighLevelClient.index(indexRequest, RequestOptions.DEFAULT);
builder = XContentFactory.jsonBuilder();
builder.startObject();
builder.field("employee_id", 9);
builder.field("full_name", "Brenda Blumberg");
builder.field("first_name", "Brenda");
builder.field("last_name", "Blumberg");
builder.field("position_id", 11);
builder.field("position_title", "Store Manager");
builder.field("store_id", 21);
builder.field("department_id", 11);
builder.field("birth_date", "1979-06-23");
builder.field("hire_date", "1998-01-01 00:00:00.0");
builder.field("salary", 17000.0);
builder.field("supervisor_id", 5);
builder.field("education_level", "Graduate Degree");
builder.field("marital_status", "M");
builder.field("gender", "F");
builder.field("management_role", "Store Management");
builder.endObject();
indexRequest = new IndexRequest(indexName).source(builder);
restHighLevelClient.index(indexRequest, RequestOptions.DEFAULT);
builder = XContentFactory.jsonBuilder();
builder.startObject();
builder.field("employee_id", 10);
builder.field("full_name", "Darren Stanz");
builder.field("first_name", "Darren");
builder.field("last_name", "Stanz");
builder.field("position_id", 5);
builder.field("position_title", "VP Finance");
builder.field("store_id", 0);
builder.field("department_id", 5);
builder.field("birth_date", "1949-08-26");
builder.field("hire_date", "1994-12-01 00:00:00.0");
builder.field("salary", 50000.0);
builder.field("supervisor_id", 1);
builder.field("education_level", "Partial College");
builder.field("marital_status", "M");
builder.field("gender", "M");
builder.field("management_role", "Senior Management");
builder.endObject();
indexRequest = new IndexRequest(indexName).source(builder);
restHighLevelClient.index(indexRequest, RequestOptions.DEFAULT);
builder = XContentFactory.jsonBuilder();
builder.startObject();
builder.field("employee_id", 11);
builder.field("full_name", "Jonathan Murraiin");
builder.field("first_name", "Jonathan");
builder.field("last_name", "Murraiin");
builder.field("position_id", 11);
builder.field("position_title", "Store Manager");
builder.field("store_id", 1);
builder.field("department_id", 11);
builder.field("birth_date", "1967-06-20");
builder.field("hire_date", "1998-01-01 00:00:00.0");
builder.field("salary", 15000.0);
builder.field("supervisor_id", 5);
builder.field("education_level", "Graduate Degree");
builder.field("marital_status", "S");
builder.field("gender", "M");
builder.field("management_role", "Store Management");
builder.endObject();
indexRequest = new IndexRequest(indexName).source(builder);
restHighLevelClient.index(indexRequest, RequestOptions.DEFAULT);
restHighLevelClient.indices().refresh(new RefreshRequest(indexName), RequestOptions.DEFAULT);
}
Aggregations