package org.opendaylight.controller.cluster.datastore.shardstrategy;
import com.google.common.base.Preconditions;
-import org.opendaylight.yangtools.yang.data.api.InstanceIdentifier;
-
-import java.util.Map;
-import java.util.concurrent.ConcurrentHashMap;
+import org.opendaylight.controller.cluster.datastore.config.Configuration;
+import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
public class ShardStrategyFactory {
- private static final Map<String, ShardStrategy> moduleNameToStrategyMap = new ConcurrentHashMap();
+ private static final String UNKNOWN_MODULE_NAME = "unknown";
+
+ private final Configuration configuration;
+
+ public ShardStrategyFactory(final Configuration configuration) {
+ Preconditions.checkState(configuration != null, "configuration should not be missing");
+ this.configuration = configuration;
+ }
- private static final String UNKNOWN_MODULE_NAME = "unknown";
+ public ShardStrategy getStrategy(final YangInstanceIdentifier path) {
+ Preconditions.checkNotNull(path, "path should not be null");
- public static ShardStrategy getStrategy(InstanceIdentifier path){
- Preconditions.checkNotNull(path, "path should not be null");
+ String moduleName = getModuleName(path);
+ ShardStrategy shardStrategy = configuration.getStrategyForModule(moduleName);
+ if (shardStrategy == null) {
+ return DefaultShardStrategy.getInstance();
+ }
- String moduleName = getModuleName(path);
- ShardStrategy shardStrategy = moduleNameToStrategyMap.get(moduleName);
- if(shardStrategy == null){
- return new DefaultShardStrategy();
+ return shardStrategy;
}
- return shardStrategy;
- }
+ public static ShardStrategy newShardStrategyInstance(String moduleName, String strategyName,
+ Configuration configuration) {
+ if (ModuleShardStrategy.NAME.equals(strategyName)) {
+ return new ModuleShardStrategy(moduleName, configuration);
+ }
+ return DefaultShardStrategy.getInstance();
+ }
- private static String getModuleName(InstanceIdentifier path){
- return UNKNOWN_MODULE_NAME;
- }
+ private String getModuleName(final YangInstanceIdentifier path) {
+ if (path.isEmpty()) {
+ return UNKNOWN_MODULE_NAME;
+ }
- /**
- * 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");
- }
+ String namespace = path.getPathArguments().iterator().next().getNodeType().getNamespace().toASCIIString();
+ String moduleName = configuration.getModuleNameFromNameSpace(namespace);
+ return moduleName != null ? moduleName : UNKNOWN_MODULE_NAME;
+ }
}