opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/datastore/shardstrategy/ModuleShardStrategy.java

   1 /*
   2  * Copyright (c) 2014 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8
   9 package org.opendaylight.controller.cluster.datastore.shardstrategy;
  10
  11 import java.util.List;
  12 import org.opendaylight.controller.cluster.datastore.Configuration;
  13 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  14
  15 public class ModuleShardStrategy implements ShardStrategy {
  16
  17     public static final String NAME = "module";
  18
  19     private final String moduleName;
  20     private final Configuration configuration;
  21
  22     public ModuleShardStrategy(String moduleName, Configuration configuration){
  23         this.moduleName = moduleName;
  24
  25         this.configuration = configuration;
  26     }
  27
  28     @Override
  29     public String findShard(YangInstanceIdentifier path) {
  30         List<String> shardNames =
  31             configuration.getShardNamesFromModuleName(moduleName);
  32         if (shardNames.isEmpty()) {
  33             return DefaultShardStrategy.DEFAULT_SHARD;
  34         }
  35         return shardNames.get(0);
  36     }
  37 }