package org.opendaylight.controller.cluster.datastore.shardstrategy;
-import java.util.List;
import org.opendaylight.controller.cluster.datastore.config.Configuration;
import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
private final String moduleName;
private final Configuration configuration;
- public ModuleShardStrategy(String moduleName, Configuration configuration){
+ public ModuleShardStrategy(String moduleName, Configuration configuration) {
this.moduleName = moduleName;
-
this.configuration = configuration;
}
@Override
public String findShard(YangInstanceIdentifier path) {
- List<String> shardNames =
- configuration.getShardNamesFromModuleName(moduleName);
- if (shardNames.isEmpty()) {
- return DefaultShardStrategy.DEFAULT_SHARD;
- }
- return shardNames.get(0);
+ String shardName = configuration.getShardNameForModule(moduleName);
+ return shardName != null ? shardName : DefaultShardStrategy.DEFAULT_SHARD;
}
}