1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18 package org.apache.shardingsphere.distsql.handler.engine.update.rdl.rule.engine.database.type;
19
20 import lombok.RequiredArgsConstructor;
21 import org.apache.shardingsphere.distsql.handler.engine.update.rdl.rule.engine.database.DatabaseRuleOperator;
22 import org.apache.shardingsphere.distsql.handler.engine.update.rdl.rule.spi.database.DatabaseRuleDropExecutor;
23 import org.apache.shardingsphere.distsql.statement.rdl.rule.aware.StaticDataSourceContainedRuleAwareStatement;
24 import org.apache.shardingsphere.distsql.statement.rdl.rule.database.DatabaseRuleDefinitionStatement;
25 import org.apache.shardingsphere.infra.config.rule.RuleConfiguration;
26 import org.apache.shardingsphere.infra.config.rule.scope.DatabaseRuleConfiguration;
27 import org.apache.shardingsphere.infra.instance.mode.ModeContextManager;
28 import org.apache.shardingsphere.infra.metadata.database.ShardingSphereDatabase;
29 import org.apache.shardingsphere.infra.metadata.version.MetaDataVersion;
30 import org.apache.shardingsphere.infra.rule.attribute.datasource.StaticDataSourceRuleAttribute;
31 import org.apache.shardingsphere.infra.yaml.config.pojo.rule.YamlRuleConfiguration;
32 import org.apache.shardingsphere.mode.tuple.annotation.RepositoryTupleEntity;
33 import org.apache.shardingsphere.infra.yaml.config.swapper.rule.YamlRuleConfigurationSwapperEngine;
34 import org.apache.shardingsphere.mode.manager.ContextManager;
35
36 import java.util.Collection;
37 import java.util.Collections;
38 import java.util.Objects;
39
40
41
42
43 @RequiredArgsConstructor
44 public final class DropDatabaseRuleOperator implements DatabaseRuleOperator {
45
46 private final ContextManager contextManager;
47
48 @SuppressWarnings("rawtypes")
49 private final DatabaseRuleDropExecutor executor;
50
51 @Override
52 @SuppressWarnings("unchecked")
53 public Collection<MetaDataVersion> operate(final DatabaseRuleDefinitionStatement sqlStatement, final ShardingSphereDatabase database, final RuleConfiguration currentRuleConfig) {
54 if (!executor.hasAnyOneToBeDropped(sqlStatement)) {
55 return Collections.emptyList();
56 }
57 ModeContextManager modeContextManager = contextManager.getInstanceContext().getModeContextManager();
58 RuleConfiguration toBeDroppedRuleConfig = executor.buildToBeDroppedRuleConfiguration(sqlStatement);
59 if (sqlStatement instanceof StaticDataSourceContainedRuleAwareStatement) {
60 for (StaticDataSourceRuleAttribute each : database.getRuleMetaData().getAttributes(StaticDataSourceRuleAttribute.class)) {
61 ((StaticDataSourceContainedRuleAwareStatement) sqlStatement).getNames().forEach(each::cleanStorageNodeDataSource);
62 }
63
64 }
65 modeContextManager.removeRuleConfigurationItem(database.getName(), toBeDroppedRuleConfig);
66 RuleConfiguration toBeAlteredRuleConfig = executor.buildToBeAlteredRuleConfiguration(sqlStatement);
67 if (null != toBeAlteredRuleConfig && ((DatabaseRuleConfiguration) toBeAlteredRuleConfig).isEmpty()) {
68 YamlRuleConfiguration yamlRuleConfig = new YamlRuleConfigurationSwapperEngine().swapToYamlRuleConfiguration(currentRuleConfig);
69 modeContextManager.removeRuleConfiguration(database.getName(), Objects.requireNonNull(yamlRuleConfig.getClass().getAnnotation(RepositoryTupleEntity.class)).value());
70 return Collections.emptyList();
71 }
72 return modeContextManager.alterRuleConfiguration(database.getName(), toBeAlteredRuleConfig);
73 }
74 }