package org.opendaylight.controller.cluster.datastore.shardstrategy;
+import com.google.common.base.Optional;
import com.google.common.base.Preconditions;
-import org.opendaylight.yangtools.yang.data.api.InstanceIdentifier;
+import org.opendaylight.controller.cluster.datastore.Configuration;
+import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
public class ShardStrategyFactory {
- private static final Map<String, ShardStrategy> moduleNameToStrategyMap = new ConcurrentHashMap();
+ private static Map<String, ShardStrategy> moduleNameToStrategyMap =
+ new ConcurrentHashMap();
- private static final String UNKNOWN_MODULE_NAME = "unknown";
+ private static final String UNKNOWN_MODULE_NAME = "unknown";
+ private static Configuration configuration;
- public static ShardStrategy getStrategy(InstanceIdentifier path){
- Preconditions.checkNotNull(path, "path should not be null");
- String moduleName = getModuleName(path);
- ShardStrategy shardStrategy = moduleNameToStrategyMap.get(moduleName);
- if(shardStrategy == null){
- return new DefaultShardStrategy();
+ public static void setConfiguration(Configuration configuration){
+ ShardStrategyFactory.configuration = configuration;
+ moduleNameToStrategyMap = configuration.getModuleNameToShardStrategyMap();
}
- return shardStrategy;
- }
+ public static ShardStrategy getStrategy(YangInstanceIdentifier path) {
+ Preconditions.checkState(configuration != null, "configuration should not be missing");
+ Preconditions.checkNotNull(path, "path should not be null");
- private static String getModuleName(InstanceIdentifier path){
- return UNKNOWN_MODULE_NAME;
- }
+ String moduleName = getModuleName(path);
+ ShardStrategy shardStrategy = moduleNameToStrategyMap.get(moduleName);
+ if (shardStrategy == null) {
+ return new DefaultShardStrategy();
+ }
- /**
- * This is to be used in the future to register a custom shard strategy
- *
- * @param moduleName
- * @param shardStrategy
- */
- public static void registerShardStrategy(String moduleName, ShardStrategy shardStrategy){
- throw new UnsupportedOperationException("registering a custom shard strategy not supported yet");
- }
+ return shardStrategy;
+ }
+
+
+ private static String getModuleName(YangInstanceIdentifier path) {
+ String namespace = path.getPathArguments().iterator().next().getNodeType().getNamespace().toASCIIString();
+
+ Optional<String> optional =
+ configuration.getModuleNameFromNameSpace(namespace);
+
+ if(!optional.isPresent()){
+ return UNKNOWN_MODULE_NAME;
+ }
+
+ return optional.get();
+ }
}