package org.opendaylight.controller.cluster.datastore.shardstrategy;
import org.opendaylight.controller.cluster.datastore.Configuration;
-import org.opendaylight.yangtools.yang.data.api.InstanceIdentifier;
+import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
+
+import java.util.List;
public class ModuleShardStrategy implements ShardStrategy {
this.configuration = configuration;
}
- @Override public String findShard(InstanceIdentifier path) {
- return configuration.getShardNamesFromModuleName(moduleName).get(0);
+ @Override public String findShard(YangInstanceIdentifier path) {
+ List<String> shardNames =
+ configuration.getShardNamesFromModuleName(moduleName);
+ if(shardNames.size() == 0){
+ return DefaultShardStrategy.DEFAULT_SHARD;
+ }
+ return shardNames.get(0);
}
}