import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
+import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
+import org.opendaylight.controller.cluster.access.concepts.MemberName;
import org.opendaylight.controller.cluster.datastore.shardstrategy.ShardStrategy;
import org.opendaylight.controller.cluster.datastore.shardstrategy.ShardStrategyFactory;
}
public ConfigurationImpl(final ModuleShardConfigProvider provider) {
- this.moduleConfigMap = ImmutableMap.copyOf(provider.retrieveModuleConfigs(this));
+ ImmutableMap.Builder<String, ModuleConfig> mapBuilder = ImmutableMap.builder();
+ for(Map.Entry<String, ModuleConfig.Builder> e: provider.retrieveModuleConfigs(this).entrySet()) {
+ mapBuilder.put(e.getKey(), e.getValue().build());
+ }
+
+ this.moduleConfigMap = mapBuilder.build();
this.allShardNames = createAllShardNames(moduleConfigMap.values());
this.namespaceToModuleName = createNamespaceToModuleName(moduleConfigMap.values());
}
@Override
- public Collection<String> getMemberShardNames(final String memberName){
+ public Collection<String> getMemberShardNames(final MemberName memberName){
Preconditions.checkNotNull(memberName, "memberName should not be null");
List<String> shards = new ArrayList<>();
- for(ModuleConfig moduleConfig: moduleConfigMap.values()) {
- for(ShardConfig shardConfig: moduleConfig.getShardConfigs()) {
+ for (ModuleConfig moduleConfig: moduleConfigMap.values()) {
+ for (ShardConfig shardConfig: moduleConfig.getShardConfigs()) {
if(shardConfig.getReplicas().contains(memberName)) {
shards.add(shardConfig.getName());
}
}
@Override
- public Collection<String> getMembersFromShardName(final String shardName) {
+ public Collection<MemberName> getMembersFromShardName(final String shardName) {
Preconditions.checkNotNull(shardName, "shardName should not be null");
for(ModuleConfig moduleConfig: moduleConfigMap.values()) {
}
@Override
- public Collection<String> getUniqueMemberNamesForAllShards() {
- Set<String> allNames = new HashSet<>();
+ public Collection<MemberName> getUniqueMemberNamesForAllShards() {
+ Set<MemberName> allNames = new HashSet<>();
for(String shardName: getAllShardNames()) {
allNames.addAll(getMembersFromShardName(shardName));
}
public synchronized void addModuleShardConfiguration(ModuleShardConfiguration config) {
Preconditions.checkNotNull(config, "ModuleShardConfiguration should not be null");
- ModuleConfig moduleConfig = new ModuleConfig(config.getModuleName());
- moduleConfig.setNameSpace(config.getNamespace().toASCIIString());
- moduleConfig.setShardStrategy(createShardStrategy(config.getModuleName(), config.getShardStrategyName()));
-
- moduleConfig.addShardConfig(config.getShardName(), ImmutableSet.copyOf(config.getShardMemberNames()));
+ ModuleConfig moduleConfig = ModuleConfig.builder(config.getModuleName()).
+ nameSpace(config.getNamespace().toASCIIString()).
+ shardStrategy(createShardStrategy(config.getModuleName(), config.getShardStrategyName())).
+ shardConfig(config.getShardName(), config.getShardMemberNames()).build();
- moduleConfigMap = ImmutableMap.<String, ModuleConfig>builder().putAll(moduleConfigMap).
- put(config.getModuleName(), moduleConfig).build();
+ updateModuleConfigMap(moduleConfig);
namespaceToModuleName = ImmutableMap.<String, String>builder().putAll(namespaceToModuleName).
put(moduleConfig.getNameSpace(), moduleConfig.getName()).build();
Preconditions.checkNotNull(shardName, "shardName should not be null");
return allShardNames.contains(shardName);
}
+
+ @Override
+ public void addMemberReplicaForShard (String shardName, MemberName newMemberName) {
+ Preconditions.checkNotNull(shardName, "shardName should not be null");
+ Preconditions.checkNotNull(newMemberName, "MemberName should not be null");
+
+ for(ModuleConfig moduleConfig: moduleConfigMap.values()) {
+ ShardConfig shardConfig = moduleConfig.getShardConfig(shardName);
+ if(shardConfig != null) {
+ Set<MemberName> replicas = new HashSet<>(shardConfig.getReplicas());
+ replicas.add(newMemberName);
+ updateModuleConfigMap(ModuleConfig.builder(moduleConfig).shardConfig(shardName, replicas).build());
+ return;
+ }
+ }
+ }
+
+ @Override
+ public void removeMemberReplicaForShard (String shardName, MemberName newMemberName) {
+ Preconditions.checkNotNull(shardName, "shardName should not be null");
+ Preconditions.checkNotNull(newMemberName, "MemberName should not be null");
+
+ for(ModuleConfig moduleConfig: moduleConfigMap.values()) {
+ ShardConfig shardConfig = moduleConfig.getShardConfig(shardName);
+ if(shardConfig != null) {
+ Set<MemberName> replicas = new HashSet<>(shardConfig.getReplicas());
+ replicas.remove(newMemberName);
+ updateModuleConfigMap(ModuleConfig.builder(moduleConfig).shardConfig(shardName, replicas).build());
+ return;
+ }
+ }
+ }
+
+ private void updateModuleConfigMap(ModuleConfig moduleConfig) {
+ Map<String, ModuleConfig> newModuleConfigMap = new HashMap<>(moduleConfigMap);
+ newModuleConfigMap.put(moduleConfig.getName(), moduleConfig);
+ moduleConfigMap = ImmutableMap.copyOf(newModuleConfigMap);
+ }
}