| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| package org.apache.ignite.internal.schema.configuration; |
| |
| import java.util.Collection; |
| import java.util.Collections; |
| import java.util.List; |
| import java.util.Map; |
| import java.util.Set; |
| import java.util.concurrent.ExecutionException; |
| import java.util.stream.Collectors; |
| import org.apache.ignite.configuration.schemas.table.TableConfiguration; |
| import org.apache.ignite.configuration.schemas.table.TableValidator; |
| import org.apache.ignite.configuration.schemas.table.TablesConfiguration; |
| import org.apache.ignite.internal.configuration.ConfigurationRegistry; |
| import org.apache.ignite.internal.configuration.storage.TestConfigurationStorage; |
| import org.apache.ignite.schema.SchemaBuilders; |
| import org.apache.ignite.schema.definition.ColumnType; |
| import org.apache.ignite.schema.definition.TableDefinition; |
| import org.apache.ignite.schema.definition.builder.HashIndexDefinitionBuilder; |
| import org.apache.ignite.schema.definition.builder.PartialIndexDefinitionBuilder; |
| import org.apache.ignite.schema.definition.builder.SortedIndexDefinitionBuilder; |
| import org.apache.ignite.schema.definition.builder.TableSchemaBuilder; |
| import org.apache.ignite.schema.definition.index.HashIndexDefinition; |
| import org.apache.ignite.schema.definition.index.IndexColumnDefinition; |
| import org.apache.ignite.schema.definition.index.IndexDefinition; |
| import org.apache.ignite.schema.definition.index.PartialIndexDefinition; |
| import org.apache.ignite.schema.definition.index.SortOrder; |
| import org.apache.ignite.schema.definition.index.SortedIndexDefinition; |
| import org.junit.jupiter.api.AfterEach; |
| import org.junit.jupiter.api.BeforeEach; |
| import org.junit.jupiter.api.Disabled; |
| import org.junit.jupiter.api.Test; |
| |
| import static org.apache.ignite.configuration.annotation.ConfigurationType.DISTRIBUTED; |
| import static org.junit.jupiter.api.Assertions.assertEquals; |
| import static org.junit.jupiter.api.Assertions.assertNotNull; |
| import static org.junit.jupiter.api.Assertions.assertTrue; |
| |
| /** |
| * SchemaConfigurationConverter tests. |
| */ |
| @SuppressWarnings("InstanceVariableMayNotBeInitialized") |
| public class SchemaConfigurationConverterTest { |
| /** Table builder. */ |
| private TableSchemaBuilder tblBuilder; |
| |
| /** Configuration registry with one table for each test. */ |
| private ConfigurationRegistry confRegistry; |
| |
| /** |
| * Prepare configuration registry for test. |
| * |
| * @throws ExecutionException If failed. |
| * @throws InterruptedException If failed. |
| */ |
| @BeforeEach |
| public void createRegistry() throws ExecutionException, InterruptedException { |
| confRegistry = new ConfigurationRegistry( |
| List.of(TablesConfiguration.KEY), |
| Map.of(TableValidator.class, Set.of(TableValidatorImpl.INSTANCE)), |
| new TestConfigurationStorage(DISTRIBUTED), |
| List.of() |
| ); |
| |
| confRegistry.start(); |
| |
| tblBuilder = SchemaBuilders.tableBuilder("SNAME", "TNAME") |
| .columns( |
| SchemaBuilders.column("COL1", ColumnType.DOUBLE).build(), |
| SchemaBuilders.column("COL2", ColumnType.DOUBLE).build(), |
| SchemaBuilders.column("A", ColumnType.INT8).build(), |
| SchemaBuilders.column("B", ColumnType.INT8).build(), |
| SchemaBuilders.column("C", ColumnType.INT8).build() |
| ).withPrimaryKey("COL1"); |
| |
| TableDefinition tbl = tblBuilder.build(); |
| |
| confRegistry.getConfiguration(TablesConfiguration.KEY).change( |
| ch -> SchemaConfigurationConverter.createTable(tbl, ch) |
| .changeTables( |
| tblsCh -> tblsCh.createOrUpdate(tbl.canonicalName(), tblCh -> tblCh.changeReplicas(1)) |
| ) |
| ).get(); |
| } |
| |
| @AfterEach |
| void tearDown() { |
| confRegistry.stop(); |
| } |
| |
| /** |
| * Add/remove HashIndex into configuration and read it back. |
| */ |
| @Test |
| public void testConvertHashIndex() throws ExecutionException, InterruptedException { |
| HashIndexDefinitionBuilder builder = SchemaBuilders.hashIndex("testHI") |
| .withColumns("A", "B", "C") |
| .withHints(Collections.singletonMap("param", "value")); |
| HashIndexDefinition idx = builder.build(); |
| |
| getTbl().change(ch -> SchemaConfigurationConverter.addIndex(idx, ch)).get(); |
| |
| TableDefinition tbl = SchemaConfigurationConverter.convert(getTbl().value()); |
| |
| HashIndexDefinition idx2 = (HashIndexDefinition)getIdx(idx.name(), tbl.indices()); |
| |
| assertNotNull(idx2); |
| assertEquals("HASH", idx2.type()); |
| assertEquals(3, idx2.columns().size()); |
| } |
| |
| /** |
| * Add/remove SortedIndex into configuration and read it back. |
| */ |
| @Test |
| public void testConvertSortedIndex() throws ExecutionException, InterruptedException { |
| SortedIndexDefinitionBuilder builder = SchemaBuilders.sortedIndex("SIDX"); |
| |
| builder.addIndexColumn("A").asc().done(); |
| builder.addIndexColumn("B").desc().done(); |
| |
| SortedIndexDefinition idx = builder.build(); |
| |
| getTbl().change(ch -> SchemaConfigurationConverter.addIndex(idx, ch)).get(); |
| |
| TableDefinition tbl = SchemaConfigurationConverter.convert(getTbl().value()); |
| |
| SortedIndexDefinition idx2 = (SortedIndexDefinition)getIdx(idx.name(), tbl.indices()); |
| |
| assertNotNull(idx2); |
| assertEquals("SORTED", idx2.type()); |
| assertEquals(2, idx2.columns().size()); |
| assertEquals("A", idx2.columns().get(0).name()); |
| assertEquals("B", idx2.columns().get(1).name()); |
| assertEquals(SortOrder.ASC, idx2.columns().get(0).sortOrder()); |
| assertEquals(SortOrder.DESC, idx2.columns().get(1).sortOrder()); |
| } |
| |
| /** |
| * Add/remove index on primary key into configuration and read it back. |
| */ |
| @Test |
| public void testUniqIndex() throws ExecutionException, InterruptedException { |
| SortedIndexDefinition idx = SchemaBuilders.sortedIndex("pk_sorted") |
| .addIndexColumn("COL1").desc().done() |
| .unique(true) |
| .build(); |
| |
| getTbl().change(ch -> SchemaConfigurationConverter.addIndex(idx, ch)).get(); |
| |
| TableDefinition tbl = SchemaConfigurationConverter.convert(getTbl().value()); |
| |
| SortedIndexDefinition idx2 = (SortedIndexDefinition)getIdx(idx.name(), tbl.indices()); |
| |
| assertNotNull(idx2); |
| assertEquals("pk_sorted", idx2.name()); |
| assertEquals("SORTED", idx2.type()); |
| assertEquals(idx.columns().stream().map(IndexColumnDefinition::name).collect(Collectors.toList()), |
| idx2.columns().stream().map(IndexColumnDefinition::name).collect(Collectors.toList())); |
| assertTrue(idx2.unique()); |
| } |
| |
| /** |
| * Detect an index containing affinity key as unique one. |
| */ |
| @Disabled("https://issues.apache.org/jira/browse/IGNITE-15483") |
| @Test |
| public void testUniqueIndexDetection() throws ExecutionException, InterruptedException { |
| SortedIndexDefinition idx = SchemaBuilders.sortedIndex("uniq_sorted") |
| .addIndexColumn("A").done() |
| .addIndexColumn("COL1").desc().done() |
| .build(); |
| |
| getTbl().change(ch -> SchemaConfigurationConverter.addIndex(idx, ch)).get(); |
| |
| TableDefinition tbl = SchemaConfigurationConverter.convert(getTbl().value()); |
| |
| SortedIndexDefinition idx2 = (SortedIndexDefinition)getIdx(idx.name(), tbl.indices()); |
| |
| assertNotNull(idx2); |
| assertEquals("uniq_sorted", idx2.name()); |
| assertEquals("SORTED", idx2.type()); |
| |
| assertTrue(idx2.unique()); |
| |
| assertEquals(2, idx2.columns().size()); |
| assertEquals("A", idx2.columns().get(0).name()); |
| assertEquals("COL1", idx2.columns().get(1).name()); |
| assertEquals(SortOrder.ASC, idx2.columns().get(0).sortOrder()); |
| assertEquals(SortOrder.DESC, idx2.columns().get(1).sortOrder()); |
| } |
| |
| /** |
| * Add/remove PartialIndex into configuration and read it back. |
| */ |
| @Test |
| public void testPartialIndex() throws ExecutionException, InterruptedException { |
| PartialIndexDefinitionBuilder builder = SchemaBuilders.partialIndex("TEST"); |
| |
| builder.addIndexColumn("A").done(); |
| builder.withExpression("WHERE A > 0"); |
| |
| PartialIndexDefinition idx = builder.build(); |
| |
| getTbl().change(ch -> SchemaConfigurationConverter.addIndex(idx, ch)).get(); |
| |
| TableDefinition tbl = SchemaConfigurationConverter.convert(getTbl().value()); |
| |
| PartialIndexDefinition idx2 = (PartialIndexDefinition)getIdx(idx.name(), tbl.indices()); |
| |
| assertNotNull(idx2); |
| assertEquals("PARTIAL", idx2.type()); |
| assertEquals(idx.columns().size(), idx2.columns().size()); |
| } |
| |
| /** |
| * Add/remove table and read it back. |
| */ |
| @Test |
| public void testConvertTable() { |
| TableDefinition tbl = tblBuilder.build(); |
| |
| TableConfiguration tblCfg = confRegistry.getConfiguration(TablesConfiguration.KEY).tables() |
| .get(tbl.canonicalName()); |
| |
| TableDefinition tbl2 = SchemaConfigurationConverter.convert(tblCfg); |
| |
| assertEquals(tbl.canonicalName(), tbl2.canonicalName()); |
| assertEquals(tbl.indices().size(), tbl2.indices().size()); |
| assertEquals(tbl.keyColumns().size(), tbl2.keyColumns().size()); |
| assertEquals(tbl.affinityColumns().size(), tbl2.affinityColumns().size()); |
| assertEquals(tbl.valueColumns().size(), tbl2.valueColumns().size()); |
| } |
| |
| /** |
| * Get tests default table configuration. |
| * |
| * @return Configuration of default table. |
| */ |
| private TableConfiguration getTbl() { |
| return confRegistry.getConfiguration(TablesConfiguration.KEY).tables().get(tblBuilder.build().canonicalName()); |
| } |
| |
| /** |
| * Get table index by name. |
| * |
| * @param name Index name to find. |
| * @param idxs Table indexes. |
| * @return Index or {@code null} if there are no index with such name. |
| */ |
| private IndexDefinition getIdx(String name, Collection<IndexDefinition> idxs) { |
| return idxs.stream().filter(idx -> name.equals(idx.name())).findAny().orElse(null); |
| } |
| } |