You are viewing a plain text version of this content. The canonical link for it is here.
Posted to notifications@shardingsphere.apache.org by pa...@apache.org on 2021/06/11 05:59:10 UTC
[shardingsphere] branch master updated: Create
AlterTableStatementFederateRefresher #10696 (#10757)
This is an automated email from the ASF dual-hosted git repository.
panjuan pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/shardingsphere.git
The following commit(s) were added to refs/heads/master by this push:
new f6fda78 Create AlterTableStatementFederateRefresher #10696 (#10757)
f6fda78 is described below
commit f6fda78a6ec6caa30e5fb318602535030cf5c8e2
Author: MingxingLAI <80...@users.noreply.github.com>
AuthorDate: Fri Jun 11 13:58:30 2021 +0800
Create AlterTableStatementFederateRefresher #10696 (#10757)
* Create AlterTableStatementFederateRefresher #10696
* Fix failed unit test for MetadataRefresherFactory
* add final modifier for AlterTableStatementFederateRefresher
---
.../refresher/MetadataRefresherFactory.java | 2 +
.../refresher/MetadataRefresherFactoryTest.java | 2 +-
.../core/metadata/FederateSchemaMetadata.java | 9 +
.../type/AlterTableStatementFederateRefresher.java | 87 ++++++++
.../AlterTableStatementFederateRefresherTest.java | 218 +++++++++++++++++++++
5 files changed, 317 insertions(+), 1 deletion(-)
diff --git a/shardingsphere-infra/shardingsphere-infra-context/src/main/java/org/apache/shardingsphere/infra/context/metadata/refresher/MetadataRefresherFactory.java b/shardingsphere-infra/shardingsphere-infra-context/src/main/java/org/apache/shardingsphere/infra/context/metadata/refresher/MetadataRefresherFactory.java
index 4fc2c0d..d179946 100644
--- a/shardingsphere-infra/shardingsphere-infra-context/src/main/java/org/apache/shardingsphere/infra/context/metadata/refresher/MetadataRefresherFactory.java
+++ b/shardingsphere-infra/shardingsphere-infra-context/src/main/java/org/apache/shardingsphere/infra/context/metadata/refresher/MetadataRefresherFactory.java
@@ -28,6 +28,7 @@ import org.apache.shardingsphere.infra.metadata.schema.refresher.type.CreateView
import org.apache.shardingsphere.infra.metadata.schema.refresher.type.DropIndexStatementSchemaRefresher;
import org.apache.shardingsphere.infra.metadata.schema.refresher.type.DropTableStatementSchemaRefresher;
import org.apache.shardingsphere.infra.metadata.schema.refresher.type.DropViewStatementSchemaRefresher;
+import org.apache.shardingsphere.infra.optimize.core.metadata.refresher.type.AlterTableStatementFederateRefresher;
import org.apache.shardingsphere.infra.optimize.core.metadata.refresher.type.CreateTableStatementFederateRefresher;
import org.apache.shardingsphere.sql.parser.sql.common.statement.SQLStatement;
import org.apache.shardingsphere.sql.parser.sql.common.statement.ddl.AlterIndexStatement;
@@ -66,6 +67,7 @@ public final class MetadataRefresherFactory {
REGISTRY.get(CreateTableStatement.class).add(new CreateTableStatementSchemaRefresher());
REGISTRY.get(CreateTableStatement.class).add(new CreateTableStatementFederateRefresher());
REGISTRY.get(AlterTableStatement.class).add(new AlterTableStatementSchemaRefresher());
+ REGISTRY.get(AlterTableStatement.class).add(new AlterTableStatementFederateRefresher());
REGISTRY.get(DropTableStatement.class).add(new DropTableStatementSchemaRefresher());
REGISTRY.get(CreateIndexStatement.class).add(new CreateIndexStatementSchemaRefresher());
REGISTRY.get(AlterIndexStatement.class).add(new AlterIndexStatementSchemaRefresher());
diff --git a/shardingsphere-infra/shardingsphere-infra-context/src/test/java/org/apache/shardingsphere/infra/context/metadata/refresher/MetadataRefresherFactoryTest.java b/shardingsphere-infra/shardingsphere-infra-context/src/test/java/org/apache/shardingsphere/infra/context/metadata/refresher/MetadataRefresherFactoryTest.java
index 3eccaf4..431f071 100644
--- a/shardingsphere-infra/shardingsphere-infra-context/src/test/java/org/apache/shardingsphere/infra/context/metadata/refresher/MetadataRefresherFactoryTest.java
+++ b/shardingsphere-infra/shardingsphere-infra-context/src/test/java/org/apache/shardingsphere/infra/context/metadata/refresher/MetadataRefresherFactoryTest.java
@@ -41,7 +41,7 @@ public final class MetadataRefresherFactoryTest {
@Test
public void assertNewInstanceWithAlterTableStatement() {
- assertThat(MetadataRefresherFactory.newInstance(mock(AlterTableStatement.class)).size(), is(1));
+ assertThat(MetadataRefresherFactory.newInstance(mock(AlterTableStatement.class)).size(), is(2));
}
@Test
diff --git a/shardingsphere-infra/shardingsphere-infra-optimize/src/main/java/org/apache/shardingsphere/infra/optimize/core/metadata/FederateSchemaMetadata.java b/shardingsphere-infra/shardingsphere-infra-optimize/src/main/java/org/apache/shardingsphere/infra/optimize/core/metadata/FederateSchemaMetadata.java
index 987286e..cedfaf7 100644
--- a/shardingsphere-infra/shardingsphere-infra-optimize/src/main/java/org/apache/shardingsphere/infra/optimize/core/metadata/FederateSchemaMetadata.java
+++ b/shardingsphere-infra/shardingsphere-infra-optimize/src/main/java/org/apache/shardingsphere/infra/optimize/core/metadata/FederateSchemaMetadata.java
@@ -109,4 +109,13 @@ public final class FederateSchemaMetadata {
public void renew(final String tableName, final TableMetaData metaData) {
tables.put(tableName, new FederateTableMetadata(tableName, metaData));
}
+
+ /**
+ * Remove.
+ * @param tableName table name
+ */
+ @Synchronized
+ public void remove(final String tableName) {
+ tables.remove(tableName);
+ }
}
diff --git a/shardingsphere-infra/shardingsphere-infra-optimize/src/main/java/org/apache/shardingsphere/infra/optimize/core/metadata/refresher/type/AlterTableStatementFederateRefresher.java b/shardingsphere-infra/shardingsphere-infra-optimize/src/main/java/org/apache/shardingsphere/infra/optimize/core/metadata/refresher/type/AlterTableStatementFederateRefresher.java
new file mode 100644
index 0000000..47008a5
--- /dev/null
+++ b/shardingsphere-infra/shardingsphere-infra-optimize/src/main/java/org/apache/shardingsphere/infra/optimize/core/metadata/refresher/type/AlterTableStatementFederateRefresher.java
@@ -0,0 +1,87 @@
+/*
+ * 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.shardingsphere.infra.optimize.core.metadata.refresher.type;
+
+import org.apache.shardingsphere.infra.metadata.schema.builder.SchemaBuilderMaterials;
+import org.apache.shardingsphere.infra.metadata.schema.builder.TableMetaDataBuilder;
+import org.apache.shardingsphere.infra.metadata.schema.builder.loader.TableMetaDataLoader;
+import org.apache.shardingsphere.infra.metadata.schema.model.TableMetaData;
+import org.apache.shardingsphere.infra.optimize.core.metadata.FederateSchemaMetadata;
+import org.apache.shardingsphere.infra.optimize.core.metadata.refresher.FederateRefresher;
+import org.apache.shardingsphere.infra.rule.ShardingSphereRule;
+import org.apache.shardingsphere.infra.rule.type.TableContainedRule;
+import org.apache.shardingsphere.sql.parser.sql.common.statement.ddl.AlterTableStatement;
+
+import java.sql.SQLException;
+import java.util.Collection;
+import java.util.Objects;
+import java.util.Optional;
+import javax.sql.DataSource;
+
+/**
+ * ShardingSphere federate refresher for alter table statement.
+ */
+public final class AlterTableStatementFederateRefresher implements FederateRefresher<AlterTableStatement> {
+
+ @Override
+ public void refresh(final FederateSchemaMetadata schema, final Collection<String> routeDataSourceNames,
+ final AlterTableStatement sqlStatement, final SchemaBuilderMaterials materials) throws SQLException {
+ String tableName = sqlStatement.getTable().getTableName().getIdentifier().getValue();
+ if (sqlStatement.getRenameTable().isPresent()) {
+ String renameTableName = sqlStatement.getRenameTable().get().getTableName().getIdentifier().getValue();
+ TableMetaData tableMetaData = buildTableMetaData(routeDataSourceNames, materials, renameTableName);
+ schema.renew(renameTableName, tableMetaData);
+ schema.remove(tableName);
+ } else {
+ TableMetaData tableMetaData = buildTableMetaData(routeDataSourceNames, materials, tableName);
+ schema.renew(tableName, tableMetaData);
+ }
+ }
+
+ private TableMetaData buildTableMetaData(final Collection<String> routeDataSourceNames,
+ final SchemaBuilderMaterials materials, final String tableName) throws SQLException {
+ if (!containsInTableContainedRule(tableName, materials)) {
+ return loadTableMetaData(tableName, routeDataSourceNames, materials);
+ } else {
+ return TableMetaDataBuilder.build(tableName, materials).orElse(new TableMetaData());
+ }
+ }
+
+ private boolean containsInTableContainedRule(final String tableName, final SchemaBuilderMaterials materials) {
+ for (ShardingSphereRule each : materials.getRules()) {
+ if (each instanceof TableContainedRule && ((TableContainedRule) each).getTables().contains(tableName)) {
+ return true;
+ }
+ }
+ return false;
+ }
+
+ private TableMetaData loadTableMetaData(final String tableName, final Collection<String> routeDataSourceNames,
+ final SchemaBuilderMaterials materials) throws SQLException {
+ for (String routeDataSourceName : routeDataSourceNames) {
+ DataSource dataSource = materials.getDataSourceMap().get(routeDataSourceName);
+ Optional<TableMetaData> tableMetaDataOptional = Objects.isNull(dataSource) ? Optional.empty()
+ : TableMetaDataLoader.load(dataSource, tableName, materials.getDatabaseType());
+ if (!tableMetaDataOptional.isPresent()) {
+ continue;
+ }
+ return tableMetaDataOptional.get();
+ }
+ return new TableMetaData();
+ }
+}
diff --git a/shardingsphere-infra/shardingsphere-infra-optimize/src/test/java/org/apache/shardingsphere/infra/optimize/core/metadata/refresher/AlterTableStatementFederateRefresherTest.java b/shardingsphere-infra/shardingsphere-infra-optimize/src/test/java/org/apache/shardingsphere/infra/optimize/core/metadata/refresher/AlterTableStatementFederateRefresherTest.java
new file mode 100644
index 0000000..3baf655
--- /dev/null
+++ b/shardingsphere-infra/shardingsphere-infra-optimize/src/test/java/org/apache/shardingsphere/infra/optimize/core/metadata/refresher/AlterTableStatementFederateRefresherTest.java
@@ -0,0 +1,218 @@
+/*
+ * 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.shardingsphere.infra.optimize.core.metadata.refresher;
+
+import com.google.common.collect.ImmutableMap;
+import org.apache.shardingsphere.infra.database.type.dialect.MySQLDatabaseType;
+import org.apache.shardingsphere.infra.database.type.dialect.OracleDatabaseType;
+import org.apache.shardingsphere.infra.database.type.dialect.PostgreSQLDatabaseType;
+import org.apache.shardingsphere.infra.database.type.dialect.SQL92DatabaseType;
+import org.apache.shardingsphere.infra.database.type.dialect.SQLServerDatabaseType;
+import org.apache.shardingsphere.infra.metadata.schema.builder.SchemaBuilderMaterials;
+import org.apache.shardingsphere.infra.metadata.schema.model.ColumnMetaData;
+import org.apache.shardingsphere.infra.metadata.schema.model.IndexMetaData;
+import org.apache.shardingsphere.infra.metadata.schema.model.TableMetaData;
+import org.apache.shardingsphere.infra.optimize.core.metadata.FederateSchemaMetadata;
+import org.apache.shardingsphere.infra.optimize.core.metadata.refresher.type.AlterTableStatementFederateRefresher;
+import org.apache.shardingsphere.infra.rule.ShardingSphereRule;
+import org.apache.shardingsphere.infra.rule.type.TableContainedRule;
+import org.apache.shardingsphere.sql.parser.sql.common.segment.generic.table.SimpleTableSegment;
+import org.apache.shardingsphere.sql.parser.sql.common.segment.generic.table.TableNameSegment;
+import org.apache.shardingsphere.sql.parser.sql.common.statement.ddl.AlterTableStatement;
+import org.apache.shardingsphere.sql.parser.sql.common.value.identifier.IdentifierValue;
+import org.apache.shardingsphere.sql.parser.sql.dialect.statement.mysql.ddl.MySQLAlterTableStatement;
+import org.apache.shardingsphere.sql.parser.sql.dialect.statement.oracle.ddl.OracleAlterTableStatement;
+import org.apache.shardingsphere.sql.parser.sql.dialect.statement.postgresql.ddl.PostgreSQLAlterTableStatement;
+import org.apache.shardingsphere.sql.parser.sql.dialect.statement.sql92.ddl.SQL92AlterTableStatement;
+import org.apache.shardingsphere.sql.parser.sql.dialect.statement.sqlserver.ddl.SQLServerAlterTableStatement;
+import org.junit.Test;
+
+import java.sql.Connection;
+import java.sql.DatabaseMetaData;
+import javax.sql.DataSource;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+import java.util.Arrays;
+import java.util.Collection;
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.Map;
+
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertTrue;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.ArgumentMatchers.eq;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
+
+public final class AlterTableStatementFederateRefresherTest {
+
+ private final SchemaBuilderMaterials materials = mock(SchemaBuilderMaterials.class);
+
+ @Test
+ public void refreshTableWithRule() throws SQLException {
+ when(materials.getDatabaseType()).thenReturn(new MySQLDatabaseType());
+ refreshTableWithRule(new MySQLAlterTableStatement());
+ when(materials.getDatabaseType()).thenReturn(new OracleDatabaseType());
+ refreshTableWithRule(new OracleAlterTableStatement());
+ when(materials.getDatabaseType()).thenReturn(new PostgreSQLDatabaseType());
+ refreshTableWithRule(new PostgreSQLAlterTableStatement());
+ when(materials.getDatabaseType()).thenReturn(new SQLServerDatabaseType());
+ refreshTableWithRule(new SQLServerAlterTableStatement());
+ when(materials.getDatabaseType()).thenReturn(new SQL92DatabaseType());
+ refreshTableWithRule(new SQL92AlterTableStatement());
+ }
+
+ private void refreshTableWithRule(final AlterTableStatement alterTableStatement) throws SQLException {
+ alterTableStatement.setTable(new SimpleTableSegment(new TableNameSegment(1, 3, new IdentifierValue("t_order"))));
+ Map<String, DataSource> dataSourceMap = mock(HashMap.class);
+ TableContainedRule rule = mock(TableContainedRule.class);
+ Collection<ShardingSphereRule> rules = Collections.singletonList(rule);
+ when(materials.getRules()).thenReturn(rules);
+ when(rule.getTables()).thenReturn(Collections.singletonList("t_order"));
+ when(materials.getDataSourceMap()).thenReturn(dataSourceMap);
+ DataSource dataSource = mock(DataSource.class);
+ when(dataSourceMap.get(eq("ds"))).thenReturn(dataSource);
+ Connection connection = mock(Connection.class);
+ when(dataSource.getConnection()).thenReturn(connection);
+ DatabaseMetaData metaData = mock(DatabaseMetaData.class);
+ when(connection.getMetaData()).thenReturn(metaData);
+ ResultSet resultSet = mock(ResultSet.class);
+ when(metaData.getTables(any(), any(), any(), any())).thenReturn(resultSet);
+ when(resultSet.next()).thenReturn(false);
+ FederateRefresher<AlterTableStatement> federateRefresher = new AlterTableStatementFederateRefresher();
+ FederateSchemaMetadata schema = buildSchema();
+ federateRefresher.refresh(schema, Collections.singletonList("ds"), alterTableStatement, materials);
+ assertTrue(schema.getTables().containsKey("t_order"));
+ }
+
+ @Test
+ public void refreshTableWithoutRule() throws SQLException {
+ when(materials.getDatabaseType()).thenReturn(new MySQLDatabaseType());
+ refreshTableWithoutRule(new MySQLAlterTableStatement());
+ when(materials.getDatabaseType()).thenReturn(new OracleDatabaseType());
+ refreshTableWithoutRule(new OracleAlterTableStatement());
+ when(materials.getDatabaseType()).thenReturn(new PostgreSQLDatabaseType());
+ refreshTableWithoutRule(new PostgreSQLAlterTableStatement());
+ when(materials.getDatabaseType()).thenReturn(new SQLServerDatabaseType());
+ refreshTableWithoutRule(new SQLServerAlterTableStatement());
+ when(materials.getDatabaseType()).thenReturn(new SQL92DatabaseType());
+ refreshTableWithoutRule(new SQL92AlterTableStatement());
+ }
+
+ private void refreshTableWithoutRule(final AlterTableStatement alterTableStatement) throws SQLException {
+ alterTableStatement.setTable(
+ new SimpleTableSegment(new TableNameSegment(1, 3, new IdentifierValue("t_order"))));
+ Map<String, DataSource> dataSourceMap = mock(HashMap.class);
+ when(materials.getDataSourceMap()).thenReturn(dataSourceMap);
+ DataSource dataSource = mock(DataSource.class);
+ when(dataSourceMap.get(eq("ds"))).thenReturn(dataSource);
+ Connection connection = mock(Connection.class);
+ when(dataSource.getConnection()).thenReturn(connection);
+ DatabaseMetaData metaData = mock(DatabaseMetaData.class);
+ when(connection.getMetaData()).thenReturn(metaData);
+ ResultSet resultSet = mock(ResultSet.class);
+ when(metaData.getTables(any(), any(), any(), any())).thenReturn(resultSet);
+ when(resultSet.next()).thenReturn(false);
+ FederateRefresher<AlterTableStatement> federateRefresher = new AlterTableStatementFederateRefresher();
+ FederateSchemaMetadata schema = buildSchema();
+ federateRefresher.refresh(schema, Collections.singletonList("ds"), alterTableStatement, materials);
+ assertTrue(schema.getTables().containsKey("t_order"));
+ }
+
+ @Test
+ public void renameTableWithRule() throws SQLException {
+ when(materials.getDatabaseType()).thenReturn(new MySQLDatabaseType());
+ renameTableWithRule(new MySQLAlterTableStatement());
+ when(materials.getDatabaseType()).thenReturn(new OracleDatabaseType());
+ renameTableWithRule(new OracleAlterTableStatement());
+ when(materials.getDatabaseType()).thenReturn(new PostgreSQLDatabaseType());
+ renameTableWithRule(new PostgreSQLAlterTableStatement());
+ when(materials.getDatabaseType()).thenReturn(new SQLServerDatabaseType());
+ renameTableWithRule(new SQLServerAlterTableStatement());
+ when(materials.getDatabaseType()).thenReturn(new SQL92DatabaseType());
+ renameTableWithRule(new SQL92AlterTableStatement());
+ }
+
+ private void renameTableWithRule(final AlterTableStatement alterTableStatement) throws SQLException {
+ alterTableStatement.setTable(new SimpleTableSegment(new TableNameSegment(1, 3, new IdentifierValue("t_order"))));
+ alterTableStatement.setRenameTable(new SimpleTableSegment(new TableNameSegment(1, 3, new IdentifierValue("t_order_new"))));
+ Map<String, DataSource> dataSourceMap = mock(HashMap.class);
+ TableContainedRule rule = mock(TableContainedRule.class);
+ Collection<ShardingSphereRule> rules = Collections.singletonList(rule);
+ when(materials.getRules()).thenReturn(rules);
+ when(rule.getTables()).thenReturn(Arrays.asList("t_order", "t_order_new"));
+ when(materials.getDataSourceMap()).thenReturn(dataSourceMap);
+ DataSource dataSource = mock(DataSource.class);
+ when(dataSourceMap.get(eq("ds"))).thenReturn(dataSource);
+ Connection connection = mock(Connection.class);
+ when(dataSource.getConnection()).thenReturn(connection);
+ DatabaseMetaData metaData = mock(DatabaseMetaData.class);
+ when(connection.getMetaData()).thenReturn(metaData);
+ ResultSet resultSet = mock(ResultSet.class);
+ when(metaData.getTables(any(), any(), any(), any())).thenReturn(resultSet);
+ when(resultSet.next()).thenReturn(false);
+ FederateRefresher<AlterTableStatement> federateRefresher = new AlterTableStatementFederateRefresher();
+ FederateSchemaMetadata schema = buildSchema();
+ federateRefresher.refresh(schema, Collections.singletonList("ds"), alterTableStatement, materials);
+ assertFalse(schema.getTables().containsKey("t_order"));
+ assertTrue(schema.getTables().containsKey("t_order_new"));
+ }
+
+ @Test
+ public void renameTableWithoutRule() throws SQLException {
+ when(materials.getDatabaseType()).thenReturn(new MySQLDatabaseType());
+ renameTableWithoutRule(new MySQLAlterTableStatement());
+ when(materials.getDatabaseType()).thenReturn(new OracleDatabaseType());
+ renameTableWithoutRule(new OracleAlterTableStatement());
+ when(materials.getDatabaseType()).thenReturn(new PostgreSQLDatabaseType());
+ renameTableWithoutRule(new PostgreSQLAlterTableStatement());
+ when(materials.getDatabaseType()).thenReturn(new SQLServerDatabaseType());
+ renameTableWithoutRule(new SQLServerAlterTableStatement());
+ when(materials.getDatabaseType()).thenReturn(new SQL92DatabaseType());
+ renameTableWithoutRule(new SQL92AlterTableStatement());
+ }
+
+ private void renameTableWithoutRule(final AlterTableStatement alterTableStatement) throws SQLException {
+ alterTableStatement.setTable(new SimpleTableSegment(new TableNameSegment(1, 3, new IdentifierValue("t_order"))));
+ alterTableStatement.setRenameTable(new SimpleTableSegment(new TableNameSegment(1, 3, new IdentifierValue("t_order_new"))));
+ Map<String, DataSource> dataSourceMap = mock(HashMap.class);
+ when(materials.getDataSourceMap()).thenReturn(dataSourceMap);
+ DataSource dataSource = mock(DataSource.class);
+ when(dataSourceMap.get(eq("ds"))).thenReturn(dataSource);
+ Connection connection = mock(Connection.class);
+ when(dataSource.getConnection()).thenReturn(connection);
+ DatabaseMetaData metaData = mock(DatabaseMetaData.class);
+ when(connection.getMetaData()).thenReturn(metaData);
+ ResultSet resultSet = mock(ResultSet.class);
+ when(metaData.getTables(any(), any(), any(), any())).thenReturn(resultSet);
+ when(resultSet.next()).thenReturn(false);
+ FederateRefresher<AlterTableStatement> federateRefresher = new AlterTableStatementFederateRefresher();
+ FederateSchemaMetadata schema = buildSchema();
+ federateRefresher.refresh(schema, Collections.singletonList("ds"), alterTableStatement, materials);
+ assertFalse(schema.getTables().containsKey("t_order"));
+ assertTrue(schema.getTables().containsKey("t_order_new"));
+ }
+
+ private FederateSchemaMetadata buildSchema() {
+ Map<String, TableMetaData> metaData = ImmutableMap.of("t_order", new TableMetaData(Collections.singletonList(new ColumnMetaData("order_id", 1, false, false, false)),
+ Collections.singletonList(new IndexMetaData("index"))));
+ return new FederateSchemaMetadata("t_order", metaData);
+ }
+}
+