1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18 package org.apache.shardingsphere.globalclock.rule;
19
20 import lombok.Getter;
21 import org.apache.shardingsphere.globalclock.api.config.GlobalClockRuleConfiguration;
22 import org.apache.shardingsphere.globalclock.provider.GlobalClockProvider;
23 import org.apache.shardingsphere.infra.database.DatabaseTypeEngine;
24 import org.apache.shardingsphere.infra.database.core.type.DatabaseType;
25 import org.apache.shardingsphere.infra.metadata.database.ShardingSphereDatabase;
26 import org.apache.shardingsphere.infra.metadata.database.resource.unit.StorageUnit;
27 import org.apache.shardingsphere.infra.rule.scope.GlobalRule;
28 import org.apache.shardingsphere.infra.rule.attribute.RuleAttributes;
29 import org.apache.shardingsphere.infra.spi.type.typed.TypedSPILoader;
30 import org.apache.shardingsphere.transaction.spi.TransactionHook;
31
32 import java.util.Collection;
33 import java.util.Map;
34 import java.util.Optional;
35 import java.util.Properties;
36
37
38
39
40 public final class GlobalClockRule implements GlobalRule {
41
42 @Getter
43 private final GlobalClockRuleConfiguration configuration;
44
45 public GlobalClockRule(final GlobalClockRuleConfiguration ruleConfig, final Map<String, ShardingSphereDatabase> databases) {
46 configuration = ruleConfig;
47 if (ruleConfig.isEnabled()) {
48 TypedSPILoader.getService(GlobalClockProvider.class, getGlobalClockProviderType(), configuration.getProps());
49 TypedSPILoader.getService(TransactionHook.class, "GLOBAL_CLOCK", createProperties(databases));
50 }
51 }
52
53 private Properties createProperties(final Map<String, ShardingSphereDatabase> databases) {
54 Properties result = new Properties();
55 DatabaseType storageType = findStorageType(databases.values()).orElseGet(DatabaseTypeEngine::getDefaultStorageType);
56 result.setProperty("trunkType", storageType.getTrunkDatabaseType().orElse(storageType).getType());
57 result.setProperty("enabled", String.valueOf(configuration.isEnabled()));
58 result.setProperty("type", configuration.getType());
59 result.setProperty("provider", configuration.getProvider());
60 return result;
61 }
62
63 private Optional<DatabaseType> findStorageType(final Collection<ShardingSphereDatabase> databases) {
64 return databases.stream()
65 .flatMap(each -> each.getResourceMetaData().getStorageUnits().values().stream()).findFirst().map(StorageUnit::getStorageType);
66 }
67
68
69
70
71
72
73 public String getGlobalClockProviderType() {
74 return String.join(".", configuration.getType(), configuration.getProvider());
75 }
76
77 @Override
78 public RuleAttributes getAttributes() {
79 return new RuleAttributes();
80 }
81 }