import com.google.common.base.Preconditions;
import org.opendaylight.controller.cluster.datastore.config.Configuration;
+import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
+import org.opendaylight.mdsal.dom.api.DOMDataTreeIdentifier;
import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
public class ShardStrategyFactory {
private static final String UNKNOWN_MODULE_NAME = "unknown";
private final Configuration configuration;
+ private final LogicalDatastoreType logicalStoreType;
- public ShardStrategyFactory(final Configuration configuration) {
+ public ShardStrategyFactory(final Configuration configuration, LogicalDatastoreType logicalStoreType) {
Preconditions.checkState(configuration != null, "configuration should not be missing");
this.configuration = configuration;
+ this.logicalStoreType = Preconditions.checkNotNull(logicalStoreType);
}
public ShardStrategy getStrategy(final YangInstanceIdentifier path) {
Preconditions.checkNotNull(path, "path should not be null");
- String moduleName = getModuleName(path);
- ShardStrategy shardStrategy = configuration.getStrategyForModule(moduleName);
+ // try with the legacy module based shard mapping
+ final String moduleName = getModuleName(path);
+ final ShardStrategy shardStrategy = configuration.getStrategyForModule(moduleName);
if (shardStrategy == null) {
- return DefaultShardStrategy.getInstance();
+ // retry with prefix based sharding
+ final ShardStrategy strategyForPrefix =
+ configuration.getStrategyForPrefix(new DOMDataTreeIdentifier(logicalStoreType, path));
+ if (strategyForPrefix == null) {
+ return DefaultShardStrategy.getInstance();
+ }
+ return strategyForPrefix;
}
return shardStrategy;
public static ShardStrategy newShardStrategyInstance(String moduleName, String strategyName,
Configuration configuration) {
- if(ModuleShardStrategy.NAME.equals(strategyName)){
+ if (ModuleShardStrategy.NAME.equals(strategyName)) {
return new ModuleShardStrategy(moduleName, configuration);
}
}
private String getModuleName(final YangInstanceIdentifier path) {
+ if (path.isEmpty()) {
+ return UNKNOWN_MODULE_NAME;
+ }
+
String namespace = path.getPathArguments().iterator().next().getNodeType().getNamespace().toASCIIString();
String moduleName = configuration.getModuleNameFromNameSpace(namespace);
return moduleName != null ? moduleName : UNKNOWN_MODULE_NAME;