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.DatabaseRuleCreateExecutor;
23 import org.apache.shardingsphere.distsql.statement.rdl.rule.database.DatabaseRuleDefinitionStatement;
24 import org.apache.shardingsphere.infra.config.rule.RuleConfiguration;
25 import org.apache.shardingsphere.infra.config.rule.decorator.RuleConfigurationDecorator;
26 import org.apache.shardingsphere.infra.instance.mode.ModeContextManager;
27 import org.apache.shardingsphere.infra.metadata.database.ShardingSphereDatabase;
28 import org.apache.shardingsphere.infra.metadata.version.MetaDataVersion;
29 import org.apache.shardingsphere.infra.spi.type.typed.TypedSPILoader;
30 import org.apache.shardingsphere.mode.manager.ContextManager;
31
32 import java.util.Collection;
33 import java.util.LinkedHashMap;
34 import java.util.Map.Entry;
35 import java.util.stream.Collectors;
36
37
38
39
40 @RequiredArgsConstructor
41 public final class CreateDatabaseRuleOperator implements DatabaseRuleOperator {
42
43 private final ContextManager contextManager;
44
45 @SuppressWarnings("rawtypes")
46 private final DatabaseRuleCreateExecutor executor;
47
48 @Override
49 @SuppressWarnings("unchecked")
50 public Collection<MetaDataVersion> operate(final DatabaseRuleDefinitionStatement sqlStatement, final ShardingSphereDatabase database, final RuleConfiguration currentRuleConfig) {
51 RuleConfiguration toBeCreatedRuleConfig = executor.buildToBeCreatedRuleConfiguration(sqlStatement);
52 ModeContextManager modeContextManager = contextManager.getInstanceContext().getModeContextManager();
53 return modeContextManager.alterRuleConfiguration(database.getName(), decorateRuleConfiguration(database, toBeCreatedRuleConfig));
54 }
55
56 @SuppressWarnings("unchecked")
57 private RuleConfiguration decorateRuleConfiguration(final ShardingSphereDatabase database, final RuleConfiguration ruleConfig) {
58 return TypedSPILoader.findService(RuleConfigurationDecorator.class, ruleConfig.getClass()).map(optional -> optional.decorate(database.getName(),
59 database.getResourceMetaData().getStorageUnits().entrySet().stream()
60 .collect(Collectors.toMap(Entry::getKey, entry -> entry.getValue().getDataSource(), (oldValue, currentValue) -> oldValue, LinkedHashMap::new)),
61 database.getRuleMetaData().getRules(), ruleConfig)).orElse(ruleConfig);
62 }
63 }