opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/datastore/ConfigurationImpl.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;
  10
  11 import com.typesafe.config.Config;
  12 import com.typesafe.config.ConfigFactory;
  13 import com.typesafe.config.ConfigObject;
  14
  15 import java.util.ArrayList;
  16 import java.util.List;
  17
  18 public class ConfigurationImpl implements Configuration {
  19     private final List<ModuleShard> moduleShards = new ArrayList<>();
  20     private final List<Module> modules = new ArrayList<>();
  21
  22
  23     public ConfigurationImpl(String moduleShardsConfigPath,
  24         String modulesConfigPath){
  25         Config moduleShardsConfig = ConfigFactory.load(moduleShardsConfigPath);
  26         Config modulesConfig = ConfigFactory.load(modulesConfigPath);
  27
  28         readModuleShards(moduleShardsConfig);
  29
  30         readModules(modulesConfig);
  31     }
  32
  33     public List<String> getMemberShardNames(String memberName){
  34
  35         List<String> shards = new ArrayList();
  36         for(ModuleShard ms : moduleShards){
  37             for(Shard s : ms.getShards()){
  38                 for(String m : s.getReplicas()){
  39                     if(memberName.equals(m)){
  40                         shards.add(s.getName());
  41                     }
  42                 }
  43             }
  44         }
  45         return shards;
  46
  47     }
  48
  49
  50     private void readModules(Config modulesConfig) {
  51         List<? extends ConfigObject> modulesConfigObjectList =
  52             modulesConfig.getObjectList("modules");
  53
  54         for(ConfigObject o : modulesConfigObjectList){
  55             ConfigObjectWrapper w = new ConfigObjectWrapper(o);
  56             modules.add(new Module(w.stringValue("name"), w.stringValue(
  57                 "namespace"), w.stringValue("sharding-strategy")));
  58         }
  59     }
  60
  61     private void readModuleShards(Config moduleShardsConfig) {
  62         List<? extends ConfigObject> moduleShardsConfigObjectList =
  63             moduleShardsConfig.getObjectList("module-shards");
  64
  65         for(ConfigObject moduleShardConfigObject : moduleShardsConfigObjectList){
  66
  67             String moduleName = moduleShardConfigObject.get("name").unwrapped().toString();
  68
  69             List<? extends ConfigObject> shardsConfigObjectList =
  70                 moduleShardConfigObject.toConfig().getObjectList("shards");
  71
  72             List<Shard> shards = new ArrayList<>();
  73
  74             for(ConfigObject shard : shardsConfigObjectList){
  75                 String shardName = shard.get("name").unwrapped().toString();
  76                 List<String> replicas = shard.toConfig().getStringList("replicas");
  77                 shards.add(new Shard(shardName, replicas));
  78             }
  79
  80             this.moduleShards.add(new ModuleShard(moduleName, shards));
  81         }
  82     }
  83
  84
  85     public static class ModuleShard {
  86         private final String moduleName;
  87         private final List<Shard> shards;
  88
  89         public ModuleShard(String moduleName, List<Shard> shards) {
  90             this.moduleName = moduleName;
  91             this.shards = shards;
  92         }
  93
  94         public String getModuleName() {
  95             return moduleName;
  96         }
  97
  98         public List<Shard> getShards() {
  99             return shards;
 100         }
 101     }
 102
 103     public static class Shard {
 104         private final String name;
 105         private final List<String> replicas;
 106
 107         Shard(String name, List<String> replicas) {
 108             this.name = name;
 109             this.replicas = replicas;
 110         }
 111
 112         public String getName() {
 113             return name;
 114         }
 115
 116         public List<String> getReplicas() {
 117             return replicas;
 118         }
 119     }
 120
 121     public static class Module {
 122
 123         private final String name;
 124         private final String nameSpace;
 125         private final String shardingStrategy;
 126
 127         Module(String name, String nameSpace, String shardingStrategy) {
 128             this.name = name;
 129             this.nameSpace = nameSpace;
 130             this.shardingStrategy = shardingStrategy;
 131         }
 132
 133         public String getName() {
 134             return name;
 135         }
 136
 137         public String getNameSpace() {
 138             return nameSpace;
 139         }
 140
 141         public String getShardingStrategy() {
 142             return shardingStrategy;
 143         }
 144     }
 145
 146
 147     private static class ConfigObjectWrapper{
 148
 149         private final ConfigObject configObject;
 150
 151         ConfigObjectWrapper(ConfigObject configObject){
 152             this.configObject = configObject;
 153         }
 154
 155         public String stringValue(String name){
 156             return configObject.get(name).unwrapped().toString();
 157         }
 158     }
 159 }