@Test public void testPartitionedBy() throws Exception { CreateTableAnalyzedStatement analysis = (CreateTableAnalyzedStatement) analyze( "create table my_table (" + " id integer," + " no_index string index off," + " name string," + " date timestamp" + ") partitioned by (name)"); assertThat(analysis.partitionedBy().size(), is(1)); assertThat(analysis.partitionedBy().get(0), contains("name", "string")); // partitioned columns must be not indexed in mapping Map<String, Object> nameMapping = (Map<String, Object>) analysis.mappingProperties().get("name"); assertThat( mapToSortedString(nameMapping), is("doc_values=false, index=no, store=false, type=string")); Map<String, Object> metaMapping = (Map) analysis.mapping().get("_meta"); assertThat((Map<String, Object>) metaMapping.get("columns"), not(hasKey("name"))); List<List<String>> partitionedByMeta = (List<List<String>>) metaMapping.get("partitioned_by"); assertTrue(analysis.isPartitioned()); assertThat(partitionedByMeta.size(), is(1)); assertThat(partitionedByMeta.get(0).get(0), is("name")); assertThat(partitionedByMeta.get(0).get(1), is("string")); }
@Test public void testPartitionedByPartOfPrimaryKey() throws Exception { CreateTableAnalyzedStatement analysis = (CreateTableAnalyzedStatement) analyze( "create table my_table (" + " id1 integer," + " id2 integer," + " date timestamp," + " primary key (id1, id2)" + ") partitioned by (id1)"); assertThat(analysis.partitionedBy().size(), is(1)); assertThat(analysis.partitionedBy().get(0), contains("id1", "integer")); Map<String, Object> oMapping = (Map<String, Object>) analysis.mappingProperties().get("id1"); assertThat( mapToSortedString(oMapping), is("doc_values=false, index=no, store=false, type=integer")); assertThat( (Map<String, Object>) ((Map) analysis.mapping().get("_meta")).get("columns"), not(hasKey("id1"))); }
@Test public void testPartitionedByNestedColumns() throws Exception { CreateTableAnalyzedStatement analysis = (CreateTableAnalyzedStatement) analyze( "create table my_table (" + " id integer," + " no_index string index off," + " o object as (" + " name string" + " )," + " date timestamp" + ") partitioned by (date, o['name'])"); assertThat(analysis.partitionedBy().size(), is(2)); Map<String, Object> oMapping = (Map<String, Object>) analysis.mappingProperties().get("o"); assertThat( mapToSortedString(oMapping), is( "doc_values=false, dynamic=true, index=not_analyzed, properties={" + "name={doc_values=false, index=no, store=false, type=string}}, store=false, type=object")); assertThat( (Map<String, Object>) ((Map) analysis.mapping().get("_meta")).get("columns"), not(hasKey("date"))); Map metaColumns = (Map) ((Map) analysis.mapping().get("_meta")).get("columns"); assertNull(metaColumns); assertThat(analysis.partitionedBy().get(0), contains("date", "date")); assertThat(analysis.partitionedBy().get(1), contains("o.name", "string")); }
@Test public void testPartitionedByMultipleColumns() throws Exception { CreateTableAnalyzedStatement analysis = (CreateTableAnalyzedStatement) analyze( "create table my_table (" + " name string," + " date timestamp" + ") partitioned by (name, date)"); assertThat(analysis.partitionedBy().size(), is(2)); Map<String, Object> properties = analysis.mappingProperties(); assertThat( mapToSortedString(properties), is( "date={doc_values=false, index=no, store=false, type=date}, " + "name={doc_values=false, index=no, store=false, type=string}")); assertThat( (Map<String, Object>) ((Map) analysis.mapping().get("_meta")).get("columns"), allOf(not(hasKey("name")), not(hasKey("date")))); assertThat(analysis.partitionedBy().get(0), contains("name", "string")); assertThat(analysis.partitionedBy().get(1), contains("date", "date")); }