* terms of the Eclipse Public License v1.0 which accompanies this distribution,
* and is available at http://www.eclipse.org/legal/epl-v10.html
*/
-
package org.opendaylight.controller.cluster.datastore.shardstrategy;
-import com.google.common.base.Optional;
-import com.google.common.base.Preconditions;
-import java.util.Map;
-import java.util.concurrent.ConcurrentHashMap;
-import org.opendaylight.controller.cluster.datastore.Configuration;
+import static com.google.common.base.Preconditions.checkState;
+import static java.util.Objects.requireNonNull;
+
+import org.opendaylight.controller.cluster.datastore.config.Configuration;
import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
public class ShardStrategyFactory {
- private static Map<String, ShardStrategy> moduleNameToStrategyMap =
- new ConcurrentHashMap<>();
-
private static final String UNKNOWN_MODULE_NAME = "unknown";
- private static Configuration configuration;
+ private final Configuration configuration;
- public static void setConfiguration(final Configuration configuration){
- ShardStrategyFactory.configuration = configuration;
- moduleNameToStrategyMap = configuration.getModuleNameToShardStrategyMap();
+ public ShardStrategyFactory(final Configuration configuration) {
+ checkState(configuration != null, "configuration should not be missing");
+ this.configuration = configuration;
}
- public static ShardStrategy getStrategy(final YangInstanceIdentifier path) {
- Preconditions.checkState(configuration != null, "configuration should not be missing");
- Preconditions.checkNotNull(path, "path should not be null");
-
-
- String moduleName = getModuleName(path);
- ShardStrategy shardStrategy = moduleNameToStrategyMap.get(moduleName);
+ public ShardStrategy getStrategy(final YangInstanceIdentifier path) {
+ final String moduleName = getModuleName(requireNonNull(path, "path should not be null"));
+ final ShardStrategy shardStrategy = configuration.getStrategyForModule(moduleName);
if (shardStrategy == null) {
return DefaultShardStrategy.getInstance();
}
return shardStrategy;
}
+ public static ShardStrategy newShardStrategyInstance(final String moduleName, final String strategyName,
+ final Configuration configuration) {
+ if (ModuleShardStrategy.NAME.equals(strategyName)) {
+ return new ModuleShardStrategy(moduleName, configuration);
+ }
- private static String getModuleName(final YangInstanceIdentifier path) {
- String namespace = path.getPathArguments().iterator().next().getNodeType().getNamespace().toASCIIString();
-
- Optional<String> optional =
- configuration.getModuleNameFromNameSpace(namespace);
+ return DefaultShardStrategy.getInstance();
+ }
- if(!optional.isPresent()){
+ private String getModuleName(final YangInstanceIdentifier path) {
+ if (path.isEmpty()) {
return UNKNOWN_MODULE_NAME;
}
- return optional.get();
+ String namespace = path.getPathArguments().get(0).getNodeType().getNamespace().toString();
+ String moduleName = configuration.getModuleNameFromNameSpace(namespace);
+ return moduleName != null ? moduleName : UNKNOWN_MODULE_NAME;
}
}